chore: Fix integration tests.

Author: anubhav756

packages/toolbox-core/tests/test_e2e.py

@@ -133,11 +133,14 @@ async def test_run_tool_unauth_with_auth(
         self, toolbox: ToolboxClient, auth_token2: str
     ):
         """Tests running a tool that doesn't require auth, with auth provided."""
-        tool = await toolbox.load_tool(
-            "get-row-by-id", auth_token_getters={"my-test-auth": lambda: auth_token2}
-        )
-        response = await tool(id="2")
-        assert "row2" in response
+
+        with pytest.raises(
+            ValueError,
+            match=rf"Validation failed for tool 'get-row-by-id': unused auth tokens: my-test-auth",
+        ):
+            await toolbox.load_tool(
+                "get-row-by-id", auth_token_getters={"my-test-auth": lambda: auth_token2}
+            )
 
     async def test_run_tool_no_auth(self, toolbox: ToolboxClient):
         """Tests running a tool requiring auth without providing auth."""

packages/toolbox-core/tests/test_sync_e2e.py

@@ -115,11 +115,14 @@ def test_run_tool_unauth_with_auth(
         self, toolbox: ToolboxSyncClient, auth_token2: str
     ):
         """Tests running a tool that doesn't require auth, with auth provided."""
-        tool = toolbox.load_tool(
-            "get-row-by-id", auth_token_getters={"my-test-auth": lambda: auth_token2}
-        )
-        response = tool(id="2")
-        assert "row2" in response
+
+        with pytest.raises(
+            ValueError,
+            match=rf"Validation failed for tool 'get-row-by-id': unused auth tokens: my-test-auth",
+        ):
+            toolbox.load_tool(
+                "get-row-by-id", auth_token_getters={"my-test-auth": lambda: auth_token2}
+            )
 
     def test_run_tool_no_auth(self, toolbox: ToolboxSyncClient):
         """Tests running a tool requiring auth without providing auth."""