editv1

Author: Harmanpreet Kaur

src/backend/tests/agents/test_agentutils.py

@@ -30,76 +30,6 @@ async def reset_event_loop():
     if not loop.is_closed():
         loop.close()
 
-@pytest.mark.asyncio
-async def test_extract_and_update_transition_states_invalid_response(reset_event_loop):
-    """Test handling of invalid JSON response from model client."""
-    session_id = "test_session"
-    user_id = "test_user"
-    step = Step(
-        data_type="step",
-        plan_id="test_plan",
-        action="test_action",
-        agent="HumanAgent",
-        session_id=session_id,
-        user_id=user_id,
-        agent_reply="test_reply",
-    )
-    model_client = AsyncMock()
-    cosmos_mock = MagicMock()
-
-    model_client.create.return_value = MagicMock(content="invalid_json")
-
-    with patch(
-        "src.backend.context.cosmos_memory.CosmosBufferedChatCompletionContext",
-        cosmos_mock,
-    ):
-        with pytest.raises(json.JSONDecodeError):
-            await extract_and_update_transition_states(
-                step=step,
-                session_id=session_id,
-                user_id=user_id,
-                planner_dynamic_or_workflow="workflow",
-                model_client=model_client,
-            )
-
-    cosmos_mock.update_step.assert_not_called()
-
-@pytest.mark.asyncio
-async def test_extract_and_update_transition_states_validation_error(reset_event_loop):
-    """Test handling of a response missing required fields."""
-    session_id = "test_session"
-    user_id = "test_user"
-    step = Step(
-        data_type="step",
-        plan_id="test_plan",
-        action="test_action",
-        agent="HumanAgent",
-        session_id=session_id,
-        user_id=user_id,
-        agent_reply="test_reply",
-    )
-    model_client = AsyncMock()
-    cosmos_mock = MagicMock()
-
-    invalid_response = {
-        "identifiedTargetState": "state1"
-    }  # Missing 'identifiedTargetTransition'
-    model_client.create.return_value = MagicMock(content=json.dumps(invalid_response))
-
-    with patch(
-        "src.backend.context.cosmos_memory.CosmosBufferedChatCompletionContext",
-        cosmos_mock,
-    ):
-        with pytest.raises(ValidationError):
-            await extract_and_update_transition_states(
-                step=step,
-                session_id=session_id,
-                user_id=user_id,
-                planner_dynamic_or_workflow="workflow",
-                model_client=model_client,
-            )
-
-    cosmos_mock.update_step.assert_not_called()
 
 def test_step_initialization():
     """Test Step initialization with valid data."""