Merge pull request #33 from MLOps-essi-upc/api_test

add API pytest

.github/workflows/backend.yml

@@ -6,7 +6,6 @@
    push:
      branches:
        - main
-       - github-actions
    pull_request:
      branches:
        - main
@@ -38,13 +37,21 @@
          echo "MLFLOW_TRACKING_USERNAME=${{ secrets.MLFLOW_TRACKING_USERNAME }}" >> .env
          echo "MLFLOW_TRACKING_PASSWORD=${{ secrets.MLFLOW_TRACKING_PASSWORD }}" >> .env
 
-     - name: Install packages # Install dependencies
-       run: pip install pytest mlflow python-dotenv
+     - name: Install packages # Install dependencies needed for tests (not for whole project)
+       run: pip install pytest mlflow python-dotenv fastapi httpx torch torchvision pydantic python-multipart dvc
+
+     - name: Pull model
+       run: |
+         dvc remote modify origin --local auth basic
+         dvc remote modify origin --local user ${{ secrets.DAGSHUB_USERNAME }}
+         dvc remote modify origin --local password ${{ secrets.DAGSHUB_TOKEN }}
+         dvc pull -r origin models/trained_model.pt
 
-     - name: Run training tests
+     - name: Run tests
        run: |
          pytest tests/test_mlflow.py
          pytest tests/test_train_model.py
+         pytest tests/test_api.py
        continue-on-error: false
 
 
@@ -74,7 +81,7 @@
      - name: Install packages # Install dependencies
        run: pip install dvc docker
 
-     - name: Pull data
+     - name: Pull model
        run: |
          dvc remote modify origin --local auth basic
          dvc remote modify origin --local user ${{ secrets.DAGSHUB_USERNAME }}

.github/workflows/frontend.yml

@@ -8,7 +8,6 @@
        - 'src/web/**'
      branches:
        - main
-       - github-actions
    pull_request:
      paths:
        - 'src/web/**'

src/app/api.py

@@ -2,7 +2,6 @@
 from typing import List
 from PIL import Image
 from io import BytesIO
-import uvicorn
 from fastapi import FastAPI, Request, File, UploadFile, HTTPException
 from torchvision import transforms
 import torch

tests/test_api.py

@@ -0,0 +1,29 @@
+import pytest
+from fastapi.testclient import TestClient
+import io
+import sys
+import os
+
+sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
+
+from src.app.api import app
+from pathlib import Path
+
+ROOT_DIR = Path(Path(__file__).resolve().parent)
+
+
+def test_make_prediction_endpoint():
+    with TestClient(app) as client:
+        # Replace 'path/to/your/image.jpg' with the actual path to your test image
+        image_path = ROOT_DIR / 'test_image_1.jpg'
+
+        # Open the image file in binary mode
+        with open(image_path, 'rb') as img_file:
+            # Create a file-like object from the image file
+            img_file_object = io.BytesIO(img_file.read())
+
+        files = {'beans_img': ('test_image_1.jpg', img_file_object, 'image/jpeg')}
+        response = client.post("/make_prediction", files=files)
+
+        assert response.status_code == 200
+        assert set(response.json().keys()) == {'prediction', 'probs'}