wehub-resource-sync
230 lines
8.2 KiB
Swift
230 lines
8.2 KiB
Swift
import XCTest
|
||
@testable import IPOverviewPlugin
|
||
|
||
final class IPOverviewParserTests: XCTestCase {
|
||
func testParsesIPAddressFromJSONPayload() {
|
||
let data = #"{"ip":"203.0.113.8"}"#.data(using: .utf8)!
|
||
|
||
XCTAssertEqual(IPOverviewParser.ipFromJSON(data), "203.0.113.8")
|
||
}
|
||
|
||
func testParsesIPAddressFromCloudflareTracePayload() {
|
||
let data = """
|
||
fl=123
|
||
h=4.ipcheck.ing
|
||
ip=2001:db8::1
|
||
ts=1710000000
|
||
""".data(using: .utf8)!
|
||
|
||
XCTAssertEqual(IPOverviewParser.ipFromCloudflareTrace(data), "2001:db8::1")
|
||
}
|
||
|
||
func testParsesIPAddressFromBilibiliPayload() {
|
||
let data = """
|
||
{"code":0,"data":{"addr":"203.0.113.8","country":"中国"}}
|
||
""".data(using: .utf8)!
|
||
|
||
XCTAssertEqual(IPOverviewParser.ipFromBilibiliIPService(data), "203.0.113.8")
|
||
}
|
||
|
||
func testParsesIPAddressFromCIPText() {
|
||
let data = """
|
||
IP\t: 203.0.113.8
|
||
地址\t: 中国 北京 北京
|
||
""".data(using: .utf8)!
|
||
|
||
XCTAssertEqual(IPOverviewParser.ipFromCIPText(data), "203.0.113.8")
|
||
}
|
||
|
||
func testParsesIPAddressFromIPIPText() {
|
||
let data = "当前 IP：203.0.113.8 来自于：中国 北京 北京 电信".data(using: .utf8)!
|
||
|
||
XCTAssertEqual(IPOverviewParser.ipFromIPIPText(data), "203.0.113.8")
|
||
}
|
||
|
||
func testParsesIPWhoisGeoPayload() {
|
||
let data = """
|
||
{
|
||
"success": true,
|
||
"country": "United States",
|
||
"region": "California",
|
||
"city": "Los Angeles",
|
||
"connection": {
|
||
"asn": 15169,
|
||
"org": "Google LLC",
|
||
"isp": "Google"
|
||
},
|
||
"timezone": {
|
||
"id": "America/Los_Angeles"
|
||
}
|
||
}
|
||
""".data(using: .utf8)!
|
||
|
||
let info = IPOverviewParser.geoInfoFromIPWhois(data, ip: "8.8.8.8")
|
||
|
||
XCTAssertEqual(info?.locationText, "United States / California / Los Angeles")
|
||
XCTAssertEqual(info?.asn, "AS15169")
|
||
XCTAssertEqual(info?.organization, "Google LLC")
|
||
XCTAssertEqual(info?.timezone, "America/Los_Angeles")
|
||
}
|
||
|
||
func testMapsPreferredLocalizationToGeoLanguageCode() {
|
||
XCTAssertEqual(IPOverviewLocale.geoLanguageCode(preferredLocalization: "zh-Hans"), "zh-CN")
|
||
XCTAssertEqual(IPOverviewLocale.geoLanguageCode(preferredLocalization: "en"), "en")
|
||
XCTAssertNil(IPOverviewLocale.geoLanguageCode(preferredLocalization: "ja"))
|
||
}
|
||
|
||
func testIPWhoisGeoURLUsesLanguageCode() {
|
||
let url = IPOverviewGeoSource.ipwhois.url(ip: "8.8.8.8", languageCode: "zh-CN")
|
||
|
||
XCTAssertEqual(url.absoluteString, "https://ipwho.is/8.8.8.8?lang=zh-CN")
|
||
}
|
||
|
||
func testParsesIPAPIGeoPayload() {
|
||
let data = """
|
||
{
|
||
"country_name": "United States",
|
||
"region": "California",
|
||
"city": "Mountain View",
|
||
"org": "GOOGLE",
|
||
"asn": "AS15169",
|
||
"timezone": "America/Los_Angeles"
|
||
}
|
||
""".data(using: .utf8)!
|
||
|
||
let info = IPOverviewParser.geoInfoFromIPAPI(data, ip: "8.8.8.8")
|
||
|
||
XCTAssertEqual(info?.locationText, "United States / California / Mountain View")
|
||
XCTAssertEqual(info?.networkText, "AS15169 GOOGLE")
|
||
XCTAssertEqual(info?.source, "ipapi.co")
|
||
}
|
||
|
||
func testParsesNetworkQualityPayload() {
|
||
let data = """
|
||
{
|
||
"base_rtt": 42.5,
|
||
"dl_throughput": 125000000,
|
||
"ul_throughput": 32000000,
|
||
"dl_responsiveness": 80,
|
||
"ul_responsiveness": 91.5,
|
||
"dl_phase_duration": 4.2,
|
||
"ul_phase_duration": 3.8,
|
||
"interface_name": "en0",
|
||
"test_endpoint": "example.apple.com",
|
||
"start_date": "2026-06-24 18:02:05.338",
|
||
"end_date": "2026-06-24 18:02:11.095"
|
||
}
|
||
""".data(using: .utf8)!
|
||
|
||
let measurement = IPOverviewNetworkQualityParser.measurement(from: data)
|
||
|
||
XCTAssertEqual(measurement?.baseRTTMilliseconds, 42.5)
|
||
XCTAssertEqual(measurement?.downloadMbps, 125)
|
||
XCTAssertEqual(measurement?.uploadMbps, 32)
|
||
XCTAssertEqual(measurement?.uploadResponsivenessRPM, 91.5)
|
||
XCTAssertEqual(measurement?.totalPhaseDuration, 8)
|
||
XCTAssertEqual(measurement?.interfaceName, "en0")
|
||
XCTAssertEqual(measurement?.testEndpoint, "example.apple.com")
|
||
}
|
||
|
||
func testParsesNetworkQualityVerbosePayload() {
|
||
let data = """
|
||
==== Verbose Results ====
|
||
Uplink capacity: 29.192 Mbps
|
||
Downlink capacity: 21.801 Mbps
|
||
Idle Latency: 200.707 milliseconds | 298 RPM
|
||
Uplink Responsiveness: Low (304.435 milliseconds | 197 RPM)
|
||
Downlink Responsiveness: Low (356.107 milliseconds | 168 RPM)
|
||
Test Endpoint: example.apple.com
|
||
Interface: en0
|
||
Start: 2026-06-24 19:10:11.260
|
||
End: 2026-06-24 19:10:19.946
|
||
Downlink Phase Length: 3.60s
|
||
Uplink Phase Length: 3.54s
|
||
""".data(using: .utf8)!
|
||
|
||
let measurement = IPOverviewNetworkQualityParser.measurement(from: data)
|
||
|
||
XCTAssertEqual(measurement?.downloadMbps, 21.801)
|
||
XCTAssertEqual(measurement?.uploadMbps, 29.192)
|
||
XCTAssertEqual(measurement?.baseRTTMilliseconds, 200.707)
|
||
XCTAssertEqual(measurement?.uploadResponsivenessRPM, 197)
|
||
XCTAssertEqual(measurement?.downloadResponsivenessRPM, 168)
|
||
XCTAssertEqual(measurement?.downloadPhaseDuration, 3.60)
|
||
XCTAssertEqual(measurement?.uploadPhaseDuration, 3.54)
|
||
XCTAssertEqual(measurement?.interfaceName, "en0")
|
||
XCTAssertEqual(measurement?.testEndpoint, "example.apple.com")
|
||
}
|
||
|
||
func testParsesNetworkQualityProgressEvents() {
|
||
let events = IPOverviewNetworkQualityParser.progressEvents(from: """
|
||
Downlink capacity: 21.801 Mbps
|
||
Uplink capacity: 29.192 Mbps
|
||
Idle Latency: 200.707 milliseconds | 298 RPM
|
||
""")
|
||
|
||
XCTAssertTrue(events.contains(.phase(.measuringDownload)))
|
||
XCTAssertTrue(events.contains(.phase(.measuringUpload)))
|
||
XCTAssertTrue(events.contains(.phase(.measuringLatency)))
|
||
XCTAssertTrue(events.contains(.download(21.801)))
|
||
XCTAssertTrue(events.contains(.upload(29.192)))
|
||
}
|
||
|
||
func testParsesNetworkQualityTTYProgressEvents() {
|
||
let events = IPOverviewNetworkQualityParser.progressEvents(from: """
|
||
\u{1B}[2K
|
||
Downlink: capacity 26.079 Mbps, responsiveness 147 RPM (17.608 MB, 6 flows) - Uplink: capacity 0.000 Mbps, responsiveness 0 RPM (0 B, 0 flows)
|
||
""")
|
||
|
||
XCTAssertTrue(events.contains(.phase(.measuringDownload)))
|
||
XCTAssertTrue(events.contains(.download(26.079)))
|
||
XCTAssertFalse(events.contains(.phase(.measuringUpload)))
|
||
XCTAssertFalse(events.contains(.upload(0)))
|
||
}
|
||
|
||
func testParsesNetworkQualityTTYUploadProgressEvents() {
|
||
let events = IPOverviewNetworkQualityParser.progressEvents(from: """
|
||
\u{1B}[2K
|
||
Downlink: capacity 26.079 Mbps, responsiveness 147 RPM (17.608 MB, 6 flows) - Uplink: capacity 24.039 Mbps, responsiveness 173 RPM (9.562 MB, 5 flows)
|
||
""")
|
||
|
||
XCTAssertTrue(events.contains(.phase(.measuringUpload)))
|
||
XCTAssertTrue(events.contains(.upload(24.039)))
|
||
XCTAssertFalse(events.contains(.download(26.079)))
|
||
}
|
||
|
||
func testGradesNetworkQualityMeasurement() {
|
||
XCTAssertEqual(
|
||
IPOverviewNetworkQualityGrade.evaluate(
|
||
baseRTTMilliseconds: 45,
|
||
downloadMbps: 150,
|
||
uploadMbps: 50
|
||
),
|
||
.excellent
|
||
)
|
||
XCTAssertEqual(
|
||
IPOverviewNetworkQualityGrade.evaluate(
|
||
baseRTTMilliseconds: 120,
|
||
downloadMbps: 8,
|
||
uploadMbps: 4
|
||
),
|
||
.fair
|
||
)
|
||
XCTAssertEqual(
|
||
IPOverviewNetworkQualityGrade.evaluate(
|
||
baseRTTMilliseconds: 240,
|
||
downloadMbps: 2,
|
||
uploadMbps: 0.5
|
||
),
|
||
.poor
|
||
)
|
||
}
|
||
|
||
func testValidatesIPAddressFamily() {
|
||
XCTAssertTrue(IPOverviewService.isValidIPAddress("203.0.113.8", family: .ipv4))
|
||
XCTAssertFalse(IPOverviewService.isValidIPAddress("203.0.113.8", family: .ipv6))
|
||
XCTAssertTrue(IPOverviewService.isValidIPAddress("2001:db8::1", family: .ipv6))
|
||
XCTAssertFalse(IPOverviewService.isValidIPAddress("not-an-ip"))
|
||
}
|
||
}
|