Skip to content

Commit 30ebfe2

Browse files
committed
fix: update CredentialStrategy method calls to snake_case in integration tests
1 parent 00d0370 commit 30ebfe2

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

packages/toolbox-adk/tests/integration/test_integration.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ async def test_load_toolset_and_run(self):
4545
toolset = ToolboxToolset(
4646
server_url="http://localhost:5000",
4747
toolset_name="my-toolset",
48-
credentials=CredentialStrategy.TOOLBOX_IDENTITY(),
48+
credentials=CredentialStrategy.toolbox_identity(),
4949
)
5050

5151
try:
@@ -71,7 +71,7 @@ async def test_partial_loading_by_names(self):
7171
toolset = ToolboxToolset(
7272
server_url="http://localhost:5000",
7373
tool_names=["get-n-rows"],
74-
credentials=CredentialStrategy.TOOLBOX_IDENTITY(),
74+
credentials=CredentialStrategy.toolbox_identity(),
7575
)
7676
try:
7777
tools = await toolset.get_tools()
@@ -92,7 +92,7 @@ async def test_bound_params_e2e(self):
9292
server_url="http://localhost:5000",
9393
tool_names=["get-n-rows"],
9494
bound_params={"num_rows": "2"},
95-
credentials=CredentialStrategy.TOOLBOX_IDENTITY(),
95+
credentials=CredentialStrategy.toolbox_identity(),
9696
)
9797
try:
9898
tools = await toolset.get_tools()
@@ -108,7 +108,7 @@ async def test_3lo_flow_simulation(self):
108108
server_url="http://localhost:5000",
109109
# We use an existing toolset that contains auth tools or just any toolset if we can pick specific tools
110110
toolset_name="my-toolset",
111-
credentials=CredentialStrategy.USER_IDENTITY(
111+
credentials=CredentialStrategy.user_identity(
112112
client_id="test-client-id", client_secret="test-client-secret"
113113
),
114114
)
@@ -171,7 +171,7 @@ async def test_manual_token_integration(self):
171171
toolset = ToolboxToolset(
172172
server_url="http://localhost:5000",
173173
toolset_name="my-toolset",
174-
credentials=CredentialStrategy.MANUAL_TOKEN(token="fake-manual-token"),
174+
credentials=CredentialStrategy.manual_token(token="fake-manual-token"),
175175
)
176176
try:
177177
tools = await toolset.get_tools()
@@ -190,7 +190,7 @@ async def test_manual_creds_integration(self):
190190
toolset = ToolboxToolset(
191191
server_url="http://localhost:5000",
192192
toolset_name="my-toolset",
193-
credentials=CredentialStrategy.MANUAL_CREDS(credentials=mock_creds),
193+
credentials=CredentialStrategy.manual_creds(credentials=mock_creds),
194194
)
195195
try:
196196
tools = await toolset.get_tools()
@@ -211,7 +211,7 @@ async def test_header_collision(self):
211211
server_url="http://localhost:5000",
212212
toolset_name="my-toolset",
213213
additional_headers={"Authorization": manual_override},
214-
credentials=CredentialStrategy.MANUAL_TOKEN(token="strategy-token"),
214+
credentials=CredentialStrategy.manual_token(token="strategy-token"),
215215
)
216216

217217
# We need to inspect the client to see what it chose.

0 commit comments

Comments
 (0)