NeoAnd/adk-python
1
0
Fork 0
mirror of https://github.com/EvolutionAPI/adk-python.git synced 2025-07-14 01:41:25 -06:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix apihub toolset ut

Browse Source 
PiperOrigin-RevId: 758400203
This commit is contained in:
Xiang (Sean) Zhou 2025-05-13 15:31:47 -07:00 committed by Copybara-Service
parent 4def3f1bd3
commit af978fe194
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/unittests/tools/apihub_tool/test_apihub_toolset.py
View File

@ -24,7 +24,7 @@ import yaml
class MockAPIHubClient(BaseAPIHubClient): class MockAPIHubClient(BaseAPIHubClient):
def get_spec_content(self, apihub_resource_name: str) -> str: def get_spec_content(self, _apihub_resource_name: str) -> str:
return """ return """
openapi: 3.0.0 openapi: 3.0.0
info: info:
@ -81,8 +81,8 @@ def mock_auth_credential():
async def test_apihub_toolset_initialization(basic_apihub_toolset): async def test_apihub_toolset_initialization(basic_apihub_toolset):
assert basic_apihub_toolset.name == 'mock_api' assert basic_apihub_toolset.name == 'mock_api'
assert basic_apihub_toolset.description == 'Mock API Description' assert basic_apihub_toolset.description == 'Mock API Description'
assert basic_apihub_toolset.apihub_resource_name == 'test_resource' assert basic_apihub_toolset._apihub_resource_name == 'test_resource'
assert not basic_apihub_toolset.lazy_load_spec assert not basic_apihub_toolset._lazy_load_spec
generated_tools = await basic_apihub_toolset.get_tools() generated_tools = await basic_apihub_toolset.get_tools()
assert len(generated_tools) == 1 assert len(generated_tools) == 1
assert 'test_get' == generated_tools[0].name assert 'test_get' == generated_tools[0].name
@ -90,7 +90,7 @@ async def test_apihub_toolset_initialization(basic_apihub_toolset):
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_apihub_toolset_lazy_loading(lazy_apihub_toolset): async def test_apihub_toolset_lazy_loading(lazy_apihub_toolset):
assert lazy_apihub_toolset.lazy_load_spec assert lazy_apihub_toolset._lazy_load_spec
generated_tools = await lazy_apihub_toolset.get_tools() generated_tools = await lazy_apihub_toolset.get_tools()
assert generated_tools assert generated_tools
@ -116,7 +116,7 @@ paths:
class MockAPIHubClientEmptySpec(BaseAPIHubClient): class MockAPIHubClientEmptySpec(BaseAPIHubClient):
def get_spec_content(self, apihub_resource_name: str) -> str: def get_spec_content(self, _apihub_resource_name: str) -> str:
return spec return spec
apihub_client = MockAPIHubClientEmptySpec() apihub_client = MockAPIHubClientEmptySpec()
@ -146,7 +146,7 @@ paths:
class MockAPIHubClientEmptySpec(BaseAPIHubClient): class MockAPIHubClientEmptySpec(BaseAPIHubClient):
def get_spec_content(self, apihub_resource_name: str) -> str: def get_spec_content(self, _apihub_resource_name: str) -> str:
return spec return spec
apihub_client = MockAPIHubClientEmptySpec() apihub_client = MockAPIHubClientEmptySpec()
@ -177,7 +177,7 @@ async def test_apihub_toolset_get_tools_lazy_load_empty_spec():
class MockAPIHubClientEmptySpec(BaseAPIHubClient): class MockAPIHubClientEmptySpec(BaseAPIHubClient):
def get_spec_content(self, apihub_resource_name: str) -> str: def get_spec_content(self, _apihub_resource_name: str) -> str:
return '' return ''
apihub_client = MockAPIHubClientEmptySpec() apihub_client = MockAPIHubClientEmptySpec()
@ -195,7 +195,7 @@ async def test_apihub_toolset_get_tools_invalid_yaml():
class MockAPIHubClientInvalidYAML(BaseAPIHubClient): class MockAPIHubClientInvalidYAML(BaseAPIHubClient):
def get_spec_content(self, apihub_resource_name: str) -> str: def get_spec_content(self, _apihub_resource_name: str) -> str:
return '{invalid yaml' # Return invalid YAML return '{invalid yaml' # Return invalid YAML
with pytest.raises(yaml.YAMLError): with pytest.raises(yaml.YAMLError):