Skip to content

Commit 906bdae

Browse files
committed
fix(tests): resolve unit test failures in tool and client wrappers
1 parent a615c9e commit 906bdae

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

packages/toolbox-adk/tests/unit/test_client.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@ def test_init_no_auth(self, mock_core_client):
2525
client = ToolboxClient("http://server", credentials=creds)
2626

2727
mock_core_client.assert_called_with(
28-
server_url="http://server",
28+
url="http://server",
2929
client_headers={}
3030
)
3131

@@ -35,7 +35,7 @@ def test_init_manual_token(self, mock_core_client):
3535
client = ToolboxClient("http://server", credentials=creds)
3636

3737
mock_core_client.assert_called_with(
38-
server_url="http://server",
38+
url="http://server",
3939
client_headers={"Authorization": "Bearer abc"}
4040
)
4141

@@ -50,7 +50,7 @@ def test_init_additional_headers(self, mock_core_client):
5050
)
5151

5252
mock_core_client.assert_called_with(
53-
server_url="http://server",
53+
url="http://server",
5454
client_headers={"X-Custom": "Val"}
5555
)
5656

packages/toolbox-adk/tests/unit/test_tool.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@
1515
import pytest
1616
from unittest.mock import MagicMock, AsyncMock
1717
from toolbox_adk.tool import ToolboxTool, ToolboxContext
18+
from toolbox_adk.credentials import CredentialConfig, CredentialType
1819

1920
class TestToolboxTool:
2021

0 commit comments

Comments
 (0)