feature(#9): updated pytest cases with configuration setup.

Brain/tests/conftest.py

@@ -2,10 +2,6 @@ import pytest
 from fastapi import FastAPI
 from fastapi.testclient import TestClient
 
-from Brain.src.firebase.firebase import initialize_app
-
-initialize_app()
-
 
 @pytest.fixture(scope="module")
 def test_client():

Brain/tests/functional/test_recipes.py

@@ -1,41 +1,63 @@
 import re
 
+from Brain.src.firebase.cloudmessage import CloudMessage
 from Brain.src.firebase.firebase import initialize_app
+from Brain.src.model.req_model import ReqModel
 from Brain.src.rising_plugin.risingplugin import getCompletion, getTextFromImage
-from Brain.src.firebase.cloudmessage import send_message
-from Brain.src.firebase.cloudmessage import get_tokens
 from Brain.src.rising_plugin.image_embedding import query_image_text
 
 TEST_IAMGE_NAME = "0ddffe51-3763-48d9-ab74-2086de529217"
 TEST_UUID = "TEST_UUID"
+test_setting = ReqModel(
+    data={
+        "token": "test-token",
+        "uuid": TEST_UUID,
+        "openai_key": "",
+        "pinecone_key": "",
+        "pinecone_env": "",
+        "firebase_key": "",
+        "settings": {"temperature": 0.6},
+    }
+)
+
+firebase_app = initialize_app(test_setting)
 
 
 def test_langchain():
     error = "Error happened while analyzing your prompt. Please ask me again"
-    result = getCompletion(query="open three.js website", uuid=TEST_UUID)
+    result = getCompletion(
+        query="open three.js website", setting=test_setting, firebase_app=firebase_app
+    )
     print(result)
     assert result != error
 
 
 def test_image2text():
     error = "Error happened while analyzing your prompt. Please ask me again"
-    result = getTextFromImage(TEST_IAMGE_NAME)
+    result = getTextFromImage(filename=TEST_IAMGE_NAME, firebase_app=firebase_app)
     print(result)
     assert result != error
 
 
 def test_firebase_cloud_message():
-    token_list = get_tokens()
+    cloud_message = CloudMessage(firebase_app=firebase_app)
+    token_list = cloud_message.get_tokens()
     notification = {"title": "alert", "content": "test"}
 
     pattern = r"send to \d+ devices, with \d+ successed, with \d+ failed."
 
-    state, value = send_message(notification, token_list)
+    state, value = cloud_message.send_message(notification, token_list)
     assert re.match(pattern, value)
 
 
 def test_query_image_text():
     error = "Error happened in querying image & text"
-    result = query_image_text(getTextFromImage(TEST_IAMGE_NAME), "", TEST_UUID)
+    result = query_image_text(
+        image_content=getTextFromImage(
+            filename=TEST_IAMGE_NAME, firebase_app=firebase_app
+        ),
+        message="",
+        setting=test_setting,
+    )
     print(result)
     assert result != error