chore: Fix integration tests

Author: anubhav756

packages/toolbox-core/tests/test_e2e.py

@@ -258,7 +258,7 @@ async def test_run_tool_with_optional_params_omitted(self, toolbox: ToolboxClien
 
         response = await tool(email="[email protected]")
         assert isinstance(response, str)
-        assert 'email="[email protected]"' in response
+        assert '"email":"[email protected]"' in response
         assert "row1" not in response
         assert "row2" in response
         assert "row3" not in response
@@ -272,7 +272,7 @@ async def test_run_tool_with_optional_data_provided(self, toolbox: ToolboxClient
 
         response = await tool(email="[email protected]", data="row3")
         assert isinstance(response, str)
-        assert 'email="[email protected]"' in response
+        assert '"email":"[email protected]"' in response
         assert "row1" not in response
         assert "row2" not in response
         assert "row3" in response
@@ -286,7 +286,7 @@ async def test_run_tool_with_optional_data_null(self, toolbox: ToolboxClient):
 
         response = await tool(email="[email protected]", data=None)
         assert isinstance(response, str)
-        assert 'email="[email protected]"' in response
+        assert '"email":"[email protected]"' in response
         assert "row1" not in response
         assert "row2" in response
         assert "row3" not in response
@@ -300,21 +300,15 @@ async def test_run_tool_with_optional_id_provided(self, toolbox: ToolboxClient):
 
         response = await tool(email="[email protected]", id=1)
         assert isinstance(response, str)
-        assert 'email="[email protected]"' in response
-        assert "row1" in response
-        assert "row2" not in response
-        assert "row3" not in response
-        assert "row4" not in response
-        assert "row5" not in response
-        assert "row6" not in response
+        assert response == "null"
 
     async def test_run_tool_with_optional_id_null(self, toolbox: ToolboxClient):
         """Invoke a tool providing both required and optional parameters."""
         tool = await toolbox.load_tool("search-rows")
 
         response = await tool(email="[email protected]", id=None)
         assert isinstance(response, str)
-        assert 'email="[email protected]"' in response
+        assert '"email":"[email protected]"' in response
         assert "row1" not in response
         assert "row2" in response
         assert "row3" not in response
@@ -331,7 +325,7 @@ async def test_run_tool_with_missing_required_param(self, toolbox: ToolboxClient
     async def test_run_tool_with_required_param_null(self, toolbox: ToolboxClient):
         """Invoke a tool without its required parameter."""
         tool = await toolbox.load_tool("search-rows")
-        with pytest.raises(TypeError, match="missing a required argument: 'email'"):
+        with pytest.raises(ValidationError, match="missing a required argument: 'email'"):
             await tool(email=None, id=5, data="row5")
 
     async def test_run_tool_with_all_default_params(self, toolbox: ToolboxClient):
@@ -340,7 +334,7 @@ async def test_run_tool_with_all_default_params(self, toolbox: ToolboxClient):
 
         response = await tool(email="[email protected]", id=0, data="row2")
         assert isinstance(response, str)
-        assert 'email="[email protected]"' in response
+        assert '"email":"[email protected]"' in response
         assert "row1" not in response
         assert "row2" in response
         assert "row3" not in response
@@ -354,7 +348,7 @@ async def test_run_tool_with_all_valid_params(self, toolbox: ToolboxClient):
 
         response = await tool(email="[email protected]", id=3, data="row3")
         assert isinstance(response, str)
-        assert 'email="[email protected]"' in response
+        assert '"email":"[email protected]"' in response
         assert "row1" not in response
         assert "row2" not in response
         assert "row3" in response