Skip to content

Commit 6df652b

Browse files
OhYeeOhYee
authored
Merge pull request #91 from Serverless-Devs/fix-sdk-breaking-change
test(api): add unit tests for various control APIs
2 parents 03bf81d + 1dfc7ad commit 6df652b

13 files changed

Lines changed: 1030 additions & 10 deletions

File tree

‎agentrun/agent_runtime/api/control.py‎

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,6 @@
2121
CreateAgentRuntimeEndpointRequest,
2222
CreateAgentRuntimeInput,
2323
CreateAgentRuntimeRequest,
24-
DeleteAgentRuntimeEndpointRequest,
25-
DeleteAgentRuntimeRequest,
26-
GetAgentRuntimeEndpointRequest,
2724
GetAgentRuntimeRequest,
2825
ListAgentRuntimeEndpointsOutput,
2926
ListAgentRuntimeEndpointsRequest,
@@ -196,7 +193,6 @@ def delete_agent_runtime(
196193
client = self._get_client(config)
197194
response = client.delete_agent_runtime_with_options(
198195
agent_id,
199-
DeleteAgentRuntimeRequest(),
200196
headers=headers or {},
201197
runtime=RuntimeOptions(),
202198
)
@@ -252,7 +248,6 @@ async def delete_agent_runtime_async(
252248
client = self._get_client(config)
253249
response = await client.delete_agent_runtime_with_options_async(
254250
agent_id,
255-
DeleteAgentRuntimeRequest(),
256251
headers=headers or {},
257252
runtime=RuntimeOptions(),
258253
)
@@ -783,7 +778,6 @@ def delete_agent_runtime_endpoint(
783778
response = client.delete_agent_runtime_endpoint_with_options(
784779
agent_id,
785780
endpoint_id,
786-
DeleteAgentRuntimeEndpointRequest(),
787781
headers=headers or {},
788782
runtime=RuntimeOptions(),
789783
)
@@ -844,7 +838,6 @@ async def delete_agent_runtime_endpoint_async(
844838
await client.delete_agent_runtime_endpoint_with_options_async(
845839
agent_id,
846840
endpoint_id,
847-
DeleteAgentRuntimeEndpointRequest(),
848841
headers=headers or {},
849842
runtime=RuntimeOptions(),
850843
)
@@ -1035,7 +1028,6 @@ def get_agent_runtime_endpoint(
10351028
response = client.get_agent_runtime_endpoint_with_options(
10361029
agent_id,
10371030
endpoint_id,
1038-
GetAgentRuntimeEndpointRequest(),
10391031
headers=headers or {},
10401032
runtime=RuntimeOptions(),
10411033
)
@@ -1096,7 +1088,6 @@ async def get_agent_runtime_endpoint_async(
10961088
await client.get_agent_runtime_endpoint_with_options_async(
10971089
agent_id,
10981090
endpoint_id,
1099-
GetAgentRuntimeEndpointRequest(),
11001091
headers=headers or {},
11011092
runtime=RuntimeOptions(),
11021093
)

‎pyproject.toml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ dependencies = [
1313
"litellm>=1.79.3",
1414
"alibabacloud-devs20230714>=2.4.1",
1515
"pydash>=8.0.5",
16-
"alibabacloud-agentrun20250910>=5.6.1",
16+
"alibabacloud-agentrun20250910>=5.6.3",
1717
"alibabacloud_tea_openapi>=0.4.2",
1818
"alibabacloud_bailian20231229>=2.6.2",
1919
"agentrun-mem0ai>=0.0.10",

‎tests/unittests/agent_runtime/api/test_control.py‎

Lines changed: 201 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,201 @@
1+
"""
2+
测试 AgentRuntimeControlAPI 对底层 SDK client 的调用是否与 SDK 方法签名匹配。
3+
"""
4+
5+
from unittest.mock import create_autospec, MagicMock, patch
6+
7+
from alibabacloud_agentrun20250910.client import Client as AgentRunClient
8+
from alibabacloud_agentrun20250910.models import (
9+
CreateAgentRuntimeEndpointInput,
10+
CreateAgentRuntimeInput,
11+
GetAgentRuntimeRequest,
12+
ListAgentRuntimeEndpointsRequest,
13+
ListAgentRuntimesRequest,
14+
ListAgentRuntimeVersionsRequest,
15+
UpdateAgentRuntimeEndpointInput,
16+
UpdateAgentRuntimeInput,
17+
)
18+
import pytest
19+
20+
from agentrun.agent_runtime.api.control import AgentRuntimeControlAPI
21+
from agentrun.utils.config import Config
22+
23+
24+
@pytest.fixture
25+
def mock_config():
26+
return Config(
27+
access_key_id="test-ak",
28+
access_key_secret="test-sk",
29+
region_id="cn-hangzhou",
30+
control_endpoint="https://agentrun.cn-hangzhou.aliyuncs.com",
31+
)
32+
33+
34+
@pytest.fixture
35+
def mock_response():
36+
response = MagicMock()
37+
response.body.request_id = "test-request-id"
38+
response.body.data = MagicMock()
39+
return response
40+
41+
42+
@pytest.fixture
43+
def api_and_client(mock_config, mock_response):
44+
api = AgentRuntimeControlAPI(config=mock_config)
45+
mock_client = create_autospec(AgentRunClient, instance=True)
46+
47+
for attr in dir(AgentRunClient):
48+
if "with_options" in attr:
49+
getattr(mock_client, attr).return_value = mock_response
50+
51+
with patch.object(api, "_get_client", return_value=mock_client):
52+
yield api, mock_client
53+
54+
55+
class TestAgentRuntimeControlAPISignatures:
56+
57+
def test_create_agent_runtime(self, api_and_client):
58+
api, client = api_and_client
59+
input_data = MagicMock(spec=CreateAgentRuntimeInput)
60+
api.create_agent_runtime(input_data)
61+
client.create_agent_runtime_with_options.assert_called_once()
62+
63+
def test_delete_agent_runtime(self, api_and_client):
64+
api, client = api_and_client
65+
api.delete_agent_runtime("agent-123")
66+
client.delete_agent_runtime_with_options.assert_called_once()
67+
68+
def test_update_agent_runtime(self, api_and_client):
69+
api, client = api_and_client
70+
input_data = MagicMock(spec=UpdateAgentRuntimeInput)
71+
api.update_agent_runtime("agent-123", input_data)
72+
client.update_agent_runtime_with_options.assert_called_once()
73+
74+
def test_get_agent_runtime(self, api_and_client):
75+
api, client = api_and_client
76+
input_data = MagicMock(spec=GetAgentRuntimeRequest)
77+
api.get_agent_runtime("agent-123", input_data)
78+
client.get_agent_runtime_with_options.assert_called_once()
79+
80+
def test_list_agent_runtimes(self, api_and_client):
81+
api, client = api_and_client
82+
input_data = MagicMock(spec=ListAgentRuntimesRequest)
83+
api.list_agent_runtimes(input_data)
84+
client.list_agent_runtimes_with_options.assert_called_once()
85+
86+
def test_create_agent_runtime_endpoint(self, api_and_client):
87+
api, client = api_and_client
88+
input_data = MagicMock(spec=CreateAgentRuntimeEndpointInput)
89+
api.create_agent_runtime_endpoint("agent-123", input_data)
90+
client.create_agent_runtime_endpoint_with_options.assert_called_once()
91+
92+
def test_delete_agent_runtime_endpoint(self, api_and_client):
93+
api, client = api_and_client
94+
api.delete_agent_runtime_endpoint("agent-123", "endpoint-456")
95+
client.delete_agent_runtime_endpoint_with_options.assert_called_once()
96+
97+
def test_update_agent_runtime_endpoint(self, api_and_client):
98+
api, client = api_and_client
99+
input_data = MagicMock(spec=UpdateAgentRuntimeEndpointInput)
100+
api.update_agent_runtime_endpoint(
101+
"agent-123", "endpoint-456", input_data
102+
)
103+
client.update_agent_runtime_endpoint_with_options.assert_called_once()
104+
105+
def test_get_agent_runtime_endpoint(self, api_and_client):
106+
api, client = api_and_client
107+
api.get_agent_runtime_endpoint("agent-123", "endpoint-456")
108+
client.get_agent_runtime_endpoint_with_options.assert_called_once()
109+
110+
def test_list_agent_runtime_endpoints(self, api_and_client):
111+
api, client = api_and_client
112+
input_data = MagicMock(spec=ListAgentRuntimeEndpointsRequest)
113+
api.list_agent_runtime_endpoints("agent-123", input_data)
114+
client.list_agent_runtime_endpoints_with_options.assert_called_once()
115+
116+
def test_list_agent_runtime_versions(self, api_and_client):
117+
api, client = api_and_client
118+
input_data = MagicMock(spec=ListAgentRuntimeVersionsRequest)
119+
api.list_agent_runtime_versions("agent-123", input_data)
120+
client.list_agent_runtime_versions_with_options.assert_called_once()
121+
122+
123+
class TestAgentRuntimeControlAPIAsyncSignatures:
124+
125+
@pytest.mark.asyncio
126+
async def test_create_agent_runtime_async(self, api_and_client):
127+
api, client = api_and_client
128+
input_data = MagicMock(spec=CreateAgentRuntimeInput)
129+
await api.create_agent_runtime_async(input_data)
130+
client.create_agent_runtime_with_options_async.assert_called_once()
131+
132+
@pytest.mark.asyncio
133+
async def test_delete_agent_runtime_async(self, api_and_client):
134+
api, client = api_and_client
135+
await api.delete_agent_runtime_async("agent-123")
136+
client.delete_agent_runtime_with_options_async.assert_called_once()
137+
138+
@pytest.mark.asyncio
139+
async def test_update_agent_runtime_async(self, api_and_client):
140+
api, client = api_and_client
141+
input_data = MagicMock(spec=UpdateAgentRuntimeInput)
142+
await api.update_agent_runtime_async("agent-123", input_data)
143+
client.update_agent_runtime_with_options_async.assert_called_once()
144+
145+
@pytest.mark.asyncio
146+
async def test_get_agent_runtime_async(self, api_and_client):
147+
api, client = api_and_client
148+
input_data = MagicMock(spec=GetAgentRuntimeRequest)
149+
await api.get_agent_runtime_async("agent-123", input_data)
150+
client.get_agent_runtime_with_options_async.assert_called_once()
151+
152+
@pytest.mark.asyncio
153+
async def test_list_agent_runtimes_async(self, api_and_client):
154+
api, client = api_and_client
155+
input_data = MagicMock(spec=ListAgentRuntimesRequest)
156+
await api.list_agent_runtimes_async(input_data)
157+
client.list_agent_runtimes_with_options_async.assert_called_once()
158+
159+
@pytest.mark.asyncio
160+
async def test_create_agent_runtime_endpoint_async(self, api_and_client):
161+
api, client = api_and_client
162+
input_data = MagicMock(spec=CreateAgentRuntimeEndpointInput)
163+
await api.create_agent_runtime_endpoint_async("agent-123", input_data)
164+
client.create_agent_runtime_endpoint_with_options_async.assert_called_once()
165+
166+
@pytest.mark.asyncio
167+
async def test_delete_agent_runtime_endpoint_async(self, api_and_client):
168+
api, client = api_and_client
169+
await api.delete_agent_runtime_endpoint_async(
170+
"agent-123", "endpoint-456"
171+
)
172+
client.delete_agent_runtime_endpoint_with_options_async.assert_called_once()
173+
174+
@pytest.mark.asyncio
175+
async def test_update_agent_runtime_endpoint_async(self, api_and_client):
176+
api, client = api_and_client
177+
input_data = MagicMock(spec=UpdateAgentRuntimeEndpointInput)
178+
await api.update_agent_runtime_endpoint_async(
179+
"agent-123", "endpoint-456", input_data
180+
)
181+
client.update_agent_runtime_endpoint_with_options_async.assert_called_once()
182+
183+
@pytest.mark.asyncio
184+
async def test_get_agent_runtime_endpoint_async(self, api_and_client):
185+
api, client = api_and_client
186+
await api.get_agent_runtime_endpoint_async("agent-123", "endpoint-456")
187+
client.get_agent_runtime_endpoint_with_options_async.assert_called_once()
188+
189+
@pytest.mark.asyncio
190+
async def test_list_agent_runtime_endpoints_async(self, api_and_client):
191+
api, client = api_and_client
192+
input_data = MagicMock(spec=ListAgentRuntimeEndpointsRequest)
193+
await api.list_agent_runtime_endpoints_async("agent-123", input_data)
194+
client.list_agent_runtime_endpoints_with_options_async.assert_called_once()
195+
196+
@pytest.mark.asyncio
197+
async def test_list_agent_runtime_versions_async(self, api_and_client):
198+
api, client = api_and_client
199+
input_data = MagicMock(spec=ListAgentRuntimeVersionsRequest)
200+
await api.list_agent_runtime_versions_async("agent-123", input_data)
201+
client.list_agent_runtime_versions_with_options_async.assert_called_once()

‎tests/unittests/credential/api/__init__.py‎

Whitespace-only changes.

‎tests/unittests/credential/api/test_control.py‎

Lines changed: 114 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,114 @@
1+
"""
2+
测试 CredentialControlAPI 对底层 SDK client 的调用是否与 SDK 方法签名匹配。
3+
"""
4+
5+
from unittest.mock import create_autospec, MagicMock, patch
6+
7+
from alibabacloud_agentrun20250910.client import Client as AgentRunClient
8+
from alibabacloud_agentrun20250910.models import (
9+
CreateCredentialInput,
10+
ListCredentialsRequest,
11+
UpdateCredentialInput,
12+
)
13+
import pytest
14+
15+
from agentrun.credential.api.control import CredentialControlAPI
16+
from agentrun.utils.config import Config
17+
18+
19+
@pytest.fixture
20+
def mock_config():
21+
return Config(
22+
access_key_id="test-ak",
23+
access_key_secret="test-sk",
24+
region_id="cn-hangzhou",
25+
control_endpoint="https://agentrun.cn-hangzhou.aliyuncs.com",
26+
)
27+
28+
29+
@pytest.fixture
30+
def mock_response():
31+
response = MagicMock()
32+
response.body.request_id = "test-request-id"
33+
response.body.data = MagicMock()
34+
return response
35+
36+
37+
@pytest.fixture
38+
def api_and_client(mock_config, mock_response):
39+
api = CredentialControlAPI(config=mock_config)
40+
mock_client = create_autospec(AgentRunClient, instance=True)
41+
42+
for attr in dir(AgentRunClient):
43+
if "with_options" in attr:
44+
getattr(mock_client, attr).return_value = mock_response
45+
46+
with patch.object(api, "_get_client", return_value=mock_client):
47+
yield api, mock_client
48+
49+
50+
class TestCredentialControlAPISignatures:
51+
52+
def test_create_credential(self, api_and_client):
53+
api, client = api_and_client
54+
input_data = MagicMock(spec=CreateCredentialInput)
55+
api.create_credential(input_data)
56+
client.create_credential_with_options.assert_called_once()
57+
58+
def test_delete_credential(self, api_and_client):
59+
api, client = api_and_client
60+
api.delete_credential("test-cred")
61+
client.delete_credential_with_options.assert_called_once()
62+
63+
def test_update_credential(self, api_and_client):
64+
api, client = api_and_client
65+
input_data = MagicMock(spec=UpdateCredentialInput)
66+
api.update_credential("test-cred", input_data)
67+
client.update_credential_with_options.assert_called_once()
68+
69+
def test_get_credential(self, api_and_client):
70+
api, client = api_and_client
71+
api.get_credential("test-cred")
72+
client.get_credential_with_options.assert_called_once()
73+
74+
def test_list_credentials(self, api_and_client):
75+
api, client = api_and_client
76+
input_data = MagicMock(spec=ListCredentialsRequest)
77+
api.list_credentials(input_data)
78+
client.list_credentials_with_options.assert_called_once()
79+
80+
81+
class TestCredentialControlAPIAsyncSignatures:
82+
83+
@pytest.mark.asyncio
84+
async def test_create_credential_async(self, api_and_client):
85+
api, client = api_and_client
86+
input_data = MagicMock(spec=CreateCredentialInput)
87+
await api.create_credential_async(input_data)
88+
client.create_credential_with_options_async.assert_called_once()
89+
90+
@pytest.mark.asyncio
91+
async def test_delete_credential_async(self, api_and_client):
92+
api, client = api_and_client
93+
await api.delete_credential_async("test-cred")
94+
client.delete_credential_with_options_async.assert_called_once()
95+
96+
@pytest.mark.asyncio
97+
async def test_update_credential_async(self, api_and_client):
98+
api, client = api_and_client
99+
input_data = MagicMock(spec=UpdateCredentialInput)
100+
await api.update_credential_async("test-cred", input_data)
101+
client.update_credential_with_options_async.assert_called_once()
102+
103+
@pytest.mark.asyncio
104+
async def test_get_credential_async(self, api_and_client):
105+
api, client = api_and_client
106+
await api.get_credential_async("test-cred")
107+
client.get_credential_with_options_async.assert_called_once()
108+
109+
@pytest.mark.asyncio
110+
async def test_list_credentials_async(self, api_and_client):
111+
api, client = api_and_client
112+
input_data = MagicMock(spec=ListCredentialsRequest)
113+
await api.list_credentials_async(input_data)
114+
client.list_credentials_with_options_async.assert_called_once()

0 commit comments

Comments
 (0)