Testcases

Author: UtkarshMishra-Microsoft

src/backend/tests/test_utils.py

@@ -3,11 +3,7 @@
 from unittest.mock import MagicMock, patch, AsyncMock
 from src.backend.utils import initialize_runtime_and_context, runtime_dict, rai_success
 from uuid import uuid4
-<<<<<<< HEAD
-
-=======
 
->>>>>>> 341589451a5fb7510cbb81054952855c90a47bf3
 # Mock environment variables
 os.environ["COSMOSDB_ENDPOINT"] = "https://mock-cosmosdb.documents.azure.com:443/"
 os.environ["COSMOSDB_KEY"] = "mock_key"
@@ -16,13 +12,8 @@
 os.environ["AZURE_OPENAI_DEPLOYMENT_NAME"] = "mock-deployment"
 os.environ["COSMOSDB_DATABASE"] = "mock_database"
 os.environ["COSMOSDB_CONTAINER"] = "mock_container"
-<<<<<<< HEAD
-
-
-=======
 
 
->>>>>>> 341589451a5fb7510cbb81054952855c90a47bf3
 @pytest.mark.asyncio
 @patch("src.backend.utils.SingleThreadedAgentRuntime")
 @patch("src.backend.utils.CosmosBufferedChatCompletionContext")
@@ -32,21 +23,12 @@ async def test_initialize_runtime_and_context_new_session(
 ):
     session_id = None
     user_id = "test-user-id"
-<<<<<<< HEAD
-
-    _mock_runtime.return_value = AsyncMock()
-    _mock_context.return_value = AsyncMock()
-
-    runtime, context = await initialize_runtime_and_context(session_id, user_id)
-
-=======
 
     _mock_runtime.return_value = AsyncMock()
     _mock_context.return_value = AsyncMock()
 
     runtime, context = await initialize_runtime_and_context(session_id, user_id)
 
->>>>>>> 341589451a5fb7510cbb81054952855c90a47bf3
     assert runtime is not None
     assert context is not None
     assert len(runtime_dict) > 0
@@ -61,19 +43,6 @@ async def test_initialize_runtime_and_context_reuse_existing_session(
 ):
     session_id = str(uuid4())
     user_id = "test-user-id"
-<<<<<<< HEAD
-
-    mock_runtime_instance = AsyncMock()
-    mock_context_instance = AsyncMock()
-    runtime_dict[session_id] = (mock_runtime_instance, mock_context_instance)
-
-    runtime, context = await initialize_runtime_and_context(session_id, user_id)
-
-    assert runtime == mock_runtime_instance
-    assert context == mock_context_instance
-
-
-=======
 
     mock_runtime_instance = AsyncMock()
     mock_context_instance = AsyncMock()
@@ -85,19 +54,12 @@ async def test_initialize_runtime_and_context_reuse_existing_session(
     assert context == mock_context_instance
 
 
->>>>>>> 341589451a5fb7510cbb81054952855c90a47bf3
 @patch("src.backend.utils.requests.post")
 @patch("src.backend.utils.DefaultAzureCredential")
 def test_rai_success_true(mock_credential, mock_post):
     mock_credential.return_value.get_token.return_value.token = "mock_token"
     mock_post.return_value.json.return_value = {"choices": [{"message": {"content": "FALSE"}}]}
     mock_post.return_value.status_code = 200
-<<<<<<< HEAD
-
-    result = rai_success("This is a valid description.")
-    assert result is True
-=======
 
     result = rai_success("This is a valid description.")
-    assert result is True
->>>>>>> 341589451a5fb7510cbb81054952855c90a47bf3
+    assert result is True