chore: Add integration tests files

.dockerignore

@@ -0,0 +1 @@
+uploads

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+*.jpeg filter=lfs diff=lfs merge=lfs -text

compose.yml

@@ -0,0 +1,78 @@
+version: "3.2"
+services:
+  api:
+    image: flask_api
+    container_name: ml_api
+    build:
+      context: ./api
+      target: build
+    ports:
+      - "8000:5000"
+    depends_on:
+      - redis
+      - model
+    volumes:
+      - ./uploads:/src/uploads
+    environment:
+      POSTGRES_DB: $POSTGRES_DB
+      POSTGRES_USER: $POSTGRES_USER
+      POSTGRES_PASSWORD: $POSTGRES_PASSWORD
+      DATABASE_HOST: $DATABASE_HOST
+      SECRET_KEY: $SECRET_KEY
+    networks:
+      - shared_network
+
+  redis:
+    image: redis:6.2.6
+    networks:
+      - shared_network
+
+  db:
+    image: postgres:latest
+    container_name: postgres_db
+    environment:
+      POSTGRES_DB: $POSTGRES_DB
+      POSTGRES_USER: $POSTGRES_USER
+      POSTGRES_PASSWORD: $POSTGRES_PASSWORD
+    volumes:
+      - postgres_data:/var/lib/postgresql/data
+    ports:
+      - "5432:5432"
+    networks:
+      - shared_network
+
+  model:
+    image: ml_service
+    container_name: ml_service
+    build:
+      context: ./model
+      dockerfile: ./Dockerfile
+    depends_on:
+      - redis
+    volumes:
+      - ./uploads:/src/uploads
+    networks:
+      - shared_network
+
+  ui:
+    image: ml_ui
+    container_name: ml_ui
+    build:
+      context: ./ui
+      target: build
+    ports:
+      - "9090:9090"
+    depends_on:
+      - api
+    environment:
+      - API_HOST=api
+      - API_PORT=5000
+    networks:
+      - shared_network
+
+networks:
+  shared_network:
+    external: true
+
+volumes:
+  postgres_data:

tests/pytest.ini

@@ -0,0 +1,3 @@
+[pytest]
+filterwarnings =
+    ignore::DeprecationWarning

tests/requirements.txt

@@ -0,0 +1,2 @@
+pytest==7.1.2
+requests==2.28.1

tests/test_integration.py

@@ -0,0 +1,54 @@
+import requests
+
+# 💡 NOTE Run tests with pytest test_integration.py -W ignore::DeprecationWarning
+
+
+def login(username, password):
+    """
+    Test the login function with valid credentials
+    """
+    url = "http://0.0.0.0:8000/login"
+    headers = {
+        "accept": "application/json",
+        "Content-Type": "application/x-www-form-urlencoded",
+    }
+    data = {
+        "grant_type": "",
+        "username": username,
+        "password": password,
+        "scope": "",
+        "client_id": "",
+        "client_secret": "",
+    }
+    response = requests.post(url, headers=headers, data=data)
+    if response.status_code == 200:
+        return response.json()["access_token"]
+    else:
+        return None
+
+
+def test_login():
+    """
+    Test the login function with valid credentials
+    """
+    token = login("admin@example.com", "admin")
+    assert token is not None
+
+
+def test_predict():
+    """
+    Test the predict function with valid credentials
+    """
+    token = login("admin@example.com", "admin")
+    files = [("file", ("dog.jpeg", open("dog.jpeg", "rb"), "image/jpeg"))]
+    headers = {"Authorization": f"Bearer {token}"}
+    response = requests.post(
+        "http://localhost:8000/model/predict",
+        headers=headers,
+        files=files,
+    )
+    assert response.status_code == 200
+    data = response.json()
+    assert data["success"] == True
+    assert data["prediction"] == "Eskimo_dog"
+    assert data["score"] == 0.9346