@@ -113,9 +113,7 @@ async def test_run_tool_unauth_with_auth(self, toolbox, auth_token2):
113
113
114
114
async def test_run_tool_no_auth (self , toolbox ):
115
115
"""Tests running a tool requiring auth without providing auth."""
116
- tool = await toolbox .load_tool (
117
- "get-row-by-id-auth" ,
118
- )
116
+ tool = await toolbox .load_tool ("get-row-by-id-auth" )
119
117
with pytest .raises (
120
118
Exception ,
121
119
match = "tool invocation not authorized. Please make sure your specify correct auth headers" ,
@@ -125,9 +123,7 @@ async def test_run_tool_no_auth(self, toolbox):
125
123
async def test_run_tool_wrong_auth (self , toolbox , auth_token2 ):
126
124
"""Tests running a tool with incorrect auth. The tool
127
125
requires a different authentication than the one provided."""
128
- tool = await toolbox .load_tool (
129
- "get-row-by-id-auth" ,
130
- )
126
+ tool = await toolbox .load_tool ("get-row-by-id-auth" )
131
127
auth_tool = tool .add_auth_token_getters ({"my-test-auth" : lambda : auth_token2 })
132
128
with pytest .raises (
133
129
Exception ,
@@ -137,9 +133,7 @@ async def test_run_tool_wrong_auth(self, toolbox, auth_token2):
137
133
138
134
async def test_run_tool_auth (self , toolbox , auth_token1 ):
139
135
"""Tests running a tool with correct auth."""
140
- tool = await toolbox .load_tool (
141
- "get-row-by-id-auth" ,
142
- )
136
+ tool = await toolbox .load_tool ("get-row-by-id-auth" )
143
137
auth_tool = tool .add_auth_token_getters ({"my-test-auth" : lambda : auth_token1 })
144
138
response = await auth_tool (id = "2" )
145
139
assert "row2" in response
0 commit comments