fix: Fix integration test

anubhav756 · anubhav756 · commit 3b068b905d2b · 2025-05-10T01:25:50.000+05:30
diff --git a/packages/toolbox-langchain/tests/test_e2e.py b/packages/toolbox-langchain/tests/test_e2e.py
@@ -286,7 +286,7 @@ def test_run_tool_wrong_auth(self, toolbox, auth_token2):
         )
         auth_tool = tool.add_auth_token_getter("my-test-auth", lambda: auth_token2)
         with pytest.raises(
-            ToolException,
+            Exception,
             match="tool invocation not authorized. Please make sure your specify correct auth headers",
         ):
             auth_tool.invoke({"id": "2"})

Original file line number	Diff line number	Diff line change
`@@ -286,7 +286,7 @@ def test_run_tool_wrong_auth(self, toolbox, auth_token2):`
`286`	`286`	`)`
`287`	`287`	`auth_tool = tool.add_auth_token_getter("my-test-auth", lambda: auth_token2)`
`288`	`288`	`with pytest.raises(`
`289`		`- ToolException,`
	`289`	`+ Exception,`
`290`	`290`	`match="tool invocation not authorized. Please make sure your specify correct auth headers",`
`291`	`291`	`):`
`292`	`292`	`auth_tool.invoke({"id": "2"})`