feat: Vultr/DO/Cloudflare适配器分页处理(避免多VPS/域名同步遗漏)+ 分页测试
This commit is contained in:
@@ -220,3 +220,34 @@ def test_ai_capabilities_account_only():
|
||||
def test_registry_supports_ai_types():
|
||||
for t in ["deepseek-api", "moonshot-api", "openai-api", "minimax-api"]:
|
||||
assert registry.is_supported(t), f"{t} 应被支持"
|
||||
|
||||
|
||||
# --------------------------- 分页处理 --------------------------- #
|
||||
def test_vultr_pagination_cursor():
|
||||
adapter = VultrAdapter({"api_key": "x"})
|
||||
page1 = {"instances": [{"id": "a", "label": "a", "ram": 1024}], "meta": {"links": {"next": "cursor2"}}}
|
||||
page2 = {"instances": [{"id": "b", "label": "b", "ram": 2048}], "meta": {"links": {"next": ""}}}
|
||||
with patch.object(VultrAdapter, "_get", side_effect=[page1, page2]):
|
||||
result = adapter.list_vps()
|
||||
assert len(result) == 2
|
||||
assert {v.external_id for v in result} == {"a", "b"}
|
||||
|
||||
|
||||
def test_do_pagination_page():
|
||||
adapter = DigitalOceanAdapter({"api_key": "x"})
|
||||
page1 = {"droplets": [{"id": 1, "name": "d1", "memory": 1024, "networks": {"v4": []}}], "links": {"pages": {"next": "x"}}}
|
||||
page2 = {"droplets": [{"id": 2, "name": "d2", "memory": 1024, "networks": {"v4": []}}], "links": {}}
|
||||
with patch.object(DigitalOceanAdapter, "_get", side_effect=[page1, page2]):
|
||||
result = adapter.list_vps()
|
||||
assert len(result) == 2
|
||||
assert {v.external_id for v in result} == {"1", "2"}
|
||||
|
||||
|
||||
def test_cloudflare_pagination_result_info():
|
||||
adapter = CloudflareAdapter({"api_token": "x"})
|
||||
page1 = {"result": [{"id": "z1", "name": "a.com", "status": "active"}], "result_info": {"total_pages": 2, "page": 1}}
|
||||
page2 = {"result": [{"id": "z2", "name": "b.com", "status": "active"}], "result_info": {"total_pages": 2, "page": 2}}
|
||||
with patch.object(CloudflareAdapter, "_get", side_effect=[page1, page2]):
|
||||
result = adapter.list_domains()
|
||||
assert len(result) == 2
|
||||
assert {d.domain_name for d in result} == {"a.com", "b.com"}
|
||||
|
||||
Reference in New Issue
Block a user