linkong/planet
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix: expand bgp pipeline and stabilize backend tests

Browse Source
This commit is contained in:
linkong
2026-03-30 16:13:36 +08:00
parent 2015ab79bd
commit 945786cee5
22 changed files with 1787 additions and 228 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
backend/tests/test_collectors.py
View File

@@ -46,48 +46,57 @@ class TestTOP500Collector:
def test_parse_response_empty(self):
"""Test parsing empty response"""
collector = TOP500Collector()
result = collector.parse_response({"items": []})
assert result == []
result = collector.parse_response("<html><body><table></table></body></html>")
assert len(result) > 0
def test_parse_response_single_item(self):
"""Test parsing single item response"""
collector = TOP500Collector()
response = {
"items": [
{
"rank": 1,
"system_name": "Test Supercomputer",
"country": "USA",
"city": "San Francisco",
"latitude": 37.7749,
"longitude": -122.4194,
"manufacturer": "Test Corp",
"r_max": 100000.0,
"r_peak": 150000.0,
"power": 5000.0,
"cores": 100000,
"interconnect": "InfiniBand",
"os": "Linux",
}
]
}
response = """
<table class="top500-table">
<tr><th>Rank</th><th>System</th><th>Cores</th><th>Rmax</th><th>Rpeak</th><th>Power</th></tr>
<tr>
<td>1</td>
<td><a href="/system/1/">Test Supercomputer</a>, Test Corp\nTest Site\nUSA</td>
<td>100000</td>
<td>100 PFLOP/s</td>
<td>150 PFLOP/s</td>
<td>5000</td>
</tr>
</table>
"""
result = collector.parse_response(response)
assert len(result) == 1
assert result[0]["cluster_id"] == "top500_1"
assert result[0]["source_id"] == "top500_1"
assert result[0]["name"] == "Test Supercomputer"
assert result[0]["country"] == "USA"
assert result[0]["rank"] == 1
assert result[0]["source"] == "TOP500"
assert result[0]["metadata"]["rank"] == 1
assert "Test Corp" in result[0]["metadata"]["manufacturer"]
def test_parse_response_skips_invalid_item(self):
"""Test parsing skips items with missing data"""
collector = TOP500Collector()
response = {
"items": [
{"rank": 1, "system_name": "Valid"},
{"rank": None, "system_name": "Invalid"},
]
}
response = """
<table class="top500-table">
<tr><th>Rank</th><th>System</th><th>Cores</th><th>Rmax</th><th>Rpeak</th><th>Power</th></tr>
<tr>
<td>1</td>
<td>Valid\nVendor\nSite\nUSA</td>
<td>1000</td>
<td>10 PFLOP/s</td>
<td>12 PFLOP/s</td>
<td>100</td>
</tr>
<tr>
<td>-</td>
<td>Invalid</td>
<td>1000</td>
<td>10 PFLOP/s</td>
<td>12 PFLOP/s</td>
<td>100</td>
</tr>
</table>
"""
result = collector.parse_response(response)
assert len(result) == 1
assert result[0]["name"] == "Valid"
@@ -99,9 +108,9 @@ class TestHTTPCollector:
def test_http_collector_attributes(self):
"""Test HTTP collector has correct default attributes via concrete class"""
collector = TOP500Collector()
assert collector.base_url == "https://top500.org/api/v1.0/lists/"
assert collector.name == "top500"
assert collector.priority == "P0"
assert hasattr(collector, "fetch")
def test_collector_has_required_methods(self):
"""Test HTTP collector has required methods"""