unit test fixes

agrimk · agrimk · commit f45350a63f76 · 2025-11-19T13:27:32.000+05:30
diff --git a/tests/unitary/with_extras/aqua/test_deployment_handler.py b/tests/unitary/with_extras/aqua/test_deployment_handler.py
@@ -274,8 +274,7 @@ def test_post(self, mock_get_model_deployment_response):
 
         mock_get_model_deployment_response.assert_called_with(
             "mock-deployment-id",
-            {"prompt": "Hello", "model": "some-model"},
-            "test-route",
+            {"prompt": "Hello", "model": "some-model"}
         )
         self.handler.write.assert_any_call("chunk1")
         self.handler.write.assert_any_call("chunk2")

Original file line number	Diff line number	Diff line change
`@@ -274,8 +274,7 @@ def test_post(self, mock_get_model_deployment_response):`
`274`	`274`
`275`	`275`	`mock_get_model_deployment_response.assert_called_with(`
`276`	`276`	`"mock-deployment-id",`
`277`		`- {"prompt": "Hello", "model": "some-model"},`
`278`		`- "test-route",`
	`277`	`+ {"prompt": "Hello", "model": "some-model"}`
`279`	`278`	`)`
`280`	`279`	`self.handler.write.assert_any_call("chunk1")`
`281`	`280`	`self.handler.write.assert_any_call("chunk2")`