feature(#85): Implement FastApi Unit Test For Api Endpoints

Brain/tests/functional/test_api.py

@@ -0,0 +1,211 @@
+from fastapi.testclient import TestClient
+import pytest
+
+from app import app
+
+client = TestClient(app)
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "confs": {
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                    "token": "eSyP3i7ITZuq8hWn2qutTl:APA91bH1FtWkaTSJwuX4WKWAl3Q-ZFyrOw4UtMP4IfwuvNrHOThH7EvEGIhtguilLRyQNlLiXatEN0xntHAc8bbKobSGjge3wxIHlspbIWY_855CzONqaVdl3y3zOmgKZNnuhYi4gwbh",
+                    "uuid": "c40a09075d11940f",
+                },
+                "message": "Please search an image that shows Brown Teddy Bear",
+            }
+        )
+    ],
+)
+def test_send_notificatoin(body):
+    response = client.post("/sendNotification", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "image_name": "0ddffe51-3763-48d9-ab74-2086de529217",
+                "confs": {
+                    "token": "eSyP3i7ITZuq8hWn2qutTl:APA91bH1FtWkaTSJwuX4WKWAl3Q-ZFyrOw4UtMP4IfwuvNrHOThH7EvEGIhtguilLRyQNlLiXatEN0xntHAc8bbKobSGjge3wxIHlspbIWY_855CzONqaVdl3y3zOmgKZNnuhYi4gwbh",
+                    "uuid": "test-uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+                "status": "updated",
+            }
+        )
+    ],
+)
+def test_upload_image(body):
+    response = client.post("/uploadImage", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "image_name": "0ddffe51-3763-48d9-ab74-2086de529217",
+                "message": "This is the text about the image",
+                "confs": {
+                    "token": "eSyP3i7ITZuq8hWn2qutTl:APA91bH1FtWkaTSJwuX4WKWAl3Q-ZFyrOw4UtMP4IfwuvNrHOThH7EvEGIhtguilLRyQNlLiXatEN0xntHAc8bbKobSGjge3wxIHlspbIWY_855CzONqaVdl3y3zOmgKZNnuhYi4gwbh",
+                    "uuid": "test-uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+            }
+        )
+    ],
+)
+def test_image_relatedness(body):
+    response = client.post("/image_relatedness", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "prompt": {"image_name": "test_image", "message": "test_message"},
+                "completion": {"image_name": "test_image", "message": "test_message"},
+                "rating": 1,
+                "confs": {
+                    "token": "eSyP3i7ITZuq8hWn2qutTl:APA91bH1FtWkaTSJwuX4WKWAl3Q-ZFyrOw4UtMP4IfwuvNrHOThH7EvEGIhtguilLRyQNlLiXatEN0xntHAc8bbKobSGjge3wxIHlspbIWY_855CzONqaVdl3y3zOmgKZNnuhYi4gwbh",
+                    "uuid": "test-uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+            }
+        )
+    ],
+)
+def test_feedback(body):
+    response = client.post("/feedback", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "confs": {
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                }
+            }
+        )
+    ],
+)
+def test_feedback(body):
+    response = client.post("/feedback/test/1", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "confs": {
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                }
+            }
+        )
+    ],
+)
+def test_commands(body):
+    response = client.post("/commands", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "history": [
+                    {"role": "system", "content": "You are a helpful assistant."},
+                    {"role": "user", "content": "Who won the world series in 2020?"},
+                    {
+                        "role": "assistant",
+                        "content": "The Los Angeles Dodgers won the World Series in 2020.",
+                    },
+                ],
+                "user_input": "Where was it played?",
+                "confs": {
+                    "token": "test_token",
+                    "uuid": "test_uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+            }
+        )
+    ],
+)
+def test_chat_rising(body):
+    response = client.post("/chat_rising", json=body)
+    assert response.status_code == 200
+
+
+@pytest.mark.parametrize(
+    "body",
+    [
+        (
+            {
+                "contacts": [
+                    {
+                        "contactId": "1",
+                        "displayName": "Thomas",
+                        "phoneNumbers": ["217 374 8105"],
+                        "status": "updated",
+                    }
+                ],
+                "confs": {
+                    "token": "test_token",
+                    "uuid": "test-uuid",
+                    "openai_key": "",
+                    "pinecone_key": "",
+                    "pinecone_env": "",
+                    "firebase_key": "",
+                    "settings": {"temperature": 0.6},
+                },
+            }
+        )
+    ],
+)
+def test_train_contacts(body):
+    response = client.post("/train/contacts", json=body)
+    assert response.status_code == 200