Add unit tests

microsoft · May 25, 2024 · a31674d · a31674d
1 parent 370edb1
commit a31674d
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 2 deletions.
diff --git a/...it_tests/dotenv_data/dotenv_no_datasource → ..._tests/dotenv_data/dotenv_no_datasource_1 b/...it_tests/dotenv_data/dotenv_no_datasource → ..._tests/dotenv_data/dotenv_no_datasource_1
@@ -10,4 +10,4 @@ AZURE_OPENAI_STREAM=False
 AZURE_OPENAI_ENDPOINT=https://dummy.openai.azure.com/
 AZURE_OPENAI_EMBEDDING_NAME=
 AZURE_OPENAI_EMBEDDING_ENDPOINT=
-AZURE_OPENAI_EMBEDDING_KEY=
+AZURE_OPENAI_EMBEDDING_KEY=
diff --git a/tests/unit_tests/dotenv_no_datasource_2 b/tests/unit_tests/dotenv_no_datasource_2
@@ -0,0 +1,14 @@
+AZURE_OPENAI_MODEL=my_model
+AZURE_OPENAI_KEY=dummy
+AZURE_OPENAI_TEMPERATURE=0
+AZURE_OPENAI_TOP_P=1.0
+AZURE_OPENAI_MAX_TOKENS=1000
+AZURE_OPENAI_STOP_SEQUENCE=
+AZURE_OPENAI_SYSTEM_MESSAGE=You are an AI assistant that helps people find information.
+AZURE_OPENAI_PREVIEW_API_VERSION=2024-05-01-preview
+AZURE_OPENAI_STREAM=False
+AZURE_OPENAI_ENDPOINT=https://dummy.openai.azure.com/
+AZURE_OPENAI_EMBEDDING_NAME=
+AZURE_OPENAI_EMBEDDING_ENDPOINT=
+AZURE_OPENAI_EMBEDDING_KEY=
+DATASOURCE_TYPE=AzureCognitiveSearch
diff --git a/tests/unit_tests/test_settings.py b/tests/unit_tests/test_settings.py
@@ -23,11 +23,17 @@ def app_settings(dotenv_path):
     yield getattr(settings_module, "app_settings")
 
 
-def test_dotenv_no_datasource(app_settings):    
+def test_dotenv_no_datasource_1(app_settings):    
     # Validate model object
     assert app_settings.base_settings.datasource_type is None
     assert app_settings.datasource is None
     assert app_settings.azure_openai is not None
+
+
+def test_dotenv_no_datasource_2(app_settings):    
+    # Validate model object
+    assert app_settings.datasource is None
+    assert app_settings.azure_openai is not None
 
 
 def test_dotenv_with_azure_search_success(app_settings):