mirror of
https://github.com/BigBodyCobain/Shadowbroker.git
synced 2026-06-09 15:53:56 +02:00
adust09
910d1fd633
- Add 68 military bases (PLA, Russia, DPRK, ROC, Philippines, Australia) with data-driven color coding (red/blue/green) on the map - Add 6 news RSS feeds (Yonhap, Nikkei Asia, Taipei Times, Asia Times, Defense News, Japan Times) and 15 geocoding keywords for islands, straits, and disputed areas - Extend ICAO country ranges for Russia, Australia, Philippines, Singapore, DPRK and add Russian aircraft classification (fighters, bombers, cargo, recon) - Create PLAN/CCG vessel enrichment module (90+ ships) following yacht_alert pattern for automatic MMSI-based identification - Update frontend types and popup styling for adversary/allied/ROC color distinction Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
92 lines
3.0 KiB
Python
92 lines
3.0 KiB
Python
"""Tests for ICAO country enrichment and military type classification."""
|
|
import pytest
|
|
from services.fetchers.military import _enrich_country, _classify_military_type
|
|
|
|
|
|
class TestEnrichCountry:
|
|
def test_china_range(self):
|
|
assert _enrich_country("780000", "Unknown") == ("China", "PLA")
|
|
|
|
def test_japan_range(self):
|
|
assert _enrich_country("840000", "Unknown") == ("Japan", "JSDF")
|
|
|
|
def test_taiwan_range(self):
|
|
assert _enrich_country("E80000", "Unknown") == ("Taiwan", "ROC")
|
|
|
|
def test_south_korea_range(self):
|
|
assert _enrich_country("700000", "Unknown") == ("South Korea", "ROK")
|
|
|
|
def test_out_of_range_unknown_flag(self):
|
|
assert _enrich_country("A00000", "Unknown") == ("Unknown", "")
|
|
|
|
def test_valid_flag_preserved(self):
|
|
country, force = _enrich_country("780000", "United States")
|
|
assert country == "United States"
|
|
assert force == ""
|
|
|
|
def test_empty_flag_uses_icao(self):
|
|
assert _enrich_country("840000", "") == ("Japan", "JSDF")
|
|
|
|
def test_military_asset_flag_uses_icao(self):
|
|
assert _enrich_country("E80000", "Military Asset") == ("Taiwan", "ROC")
|
|
|
|
def test_invalid_hex_with_unknown(self):
|
|
assert _enrich_country("ZZZZ", "Unknown") == ("Unknown", "")
|
|
|
|
def test_invalid_hex_with_empty(self):
|
|
assert _enrich_country("ZZZZ", "") == ("Military Asset", "")
|
|
|
|
def test_russia_range(self):
|
|
assert _enrich_country("150000", "Unknown") == ("Russia", "VKS")
|
|
|
|
def test_russia_range_end(self):
|
|
assert _enrich_country("157FFF", "Unknown") == ("Russia", "VKS")
|
|
|
|
def test_australia_range(self):
|
|
assert _enrich_country("7C0000", "Unknown") == ("Australia", "RAAF")
|
|
|
|
def test_philippines_range(self):
|
|
assert _enrich_country("758000", "Unknown") == ("Philippines", "PAF")
|
|
|
|
def test_singapore_range(self):
|
|
assert _enrich_country("768000", "Unknown") == ("Singapore", "RSAF")
|
|
|
|
def test_north_korea_range(self):
|
|
assert _enrich_country("720000", "Unknown") == ("North Korea", "KPAF")
|
|
|
|
|
|
class TestClassifyMilitaryType:
|
|
@pytest.mark.parametrize("model,expected", [
|
|
("J-20", "fighter"),
|
|
("Y-20", "cargo"),
|
|
("KJ-500", "recon"),
|
|
("YY-20", "tanker"),
|
|
("F-15J", "fighter"),
|
|
("FA-50", "fighter"),
|
|
("E-2K", "recon"),
|
|
("F16", "fighter"),
|
|
("C17", "cargo"),
|
|
("P8", "recon"),
|
|
("H60", "heli"),
|
|
("K35", "tanker"),
|
|
("Boeing 737", "default"),
|
|
# Russian aircraft
|
|
("SU-27", "fighter"),
|
|
("SU-30", "fighter"),
|
|
("SU-35", "fighter"),
|
|
("SU-57", "fighter"),
|
|
("MiG-29", "fighter"),
|
|
("MiG-31", "fighter"),
|
|
("Tu-95", "bomber"),
|
|
("Tu-160", "bomber"),
|
|
("Tu-22", "bomber"),
|
|
("IL-76", "cargo"),
|
|
("AN-124", "cargo"),
|
|
("AN-12", "cargo"),
|
|
("A-50", "recon"),
|
|
("Tu-214R", "recon"),
|
|
("IL-20", "recon"),
|
|
])
|
|
def test_classification(self, model: str, expected: str):
|
|
assert _classify_military_type(model) == expected
|