Initial commit: NetworkCity — network traffic as a neon city
A native macOS network monitor that renders live traffic as a top-down neon city: your Mac is downtown, remote orgs are glowing districts, and traffic is cars of light driving the roads. Architecture: - NetworkCityCore: swappable data layer behind a ConnectionSource protocol (nettop-backed today), plus pure/tested logic — traffic diffing, org classification (reverse-DNS + CIDR), traffic classes, hub-and-spoke expansion policy, label anti-overlap, and latency→distance layout. - NetworkCityApp: SwiftUI + SpriteKit city — auto-expanding districts, protocol-coloured cars, click-to-inspect panel, and latency-as-distance (districts glide to their measured RTT). - nettop-probe: CLI proof of the data layer. 44 tests passing. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,49 @@
|
||||
import XCTest
|
||||
@testable import NetworkCityCore
|
||||
|
||||
final class ExpansionPolicyTests: XCTestCase {
|
||||
typealias D = ExpansionPolicy.District
|
||||
|
||||
func testBusyMultiEndpointOrgExpands() {
|
||||
let policy = ExpansionPolicy()
|
||||
let expanded = policy.update([D(key: "onedrive", bytes: 500_000, endpointCount: 15)])
|
||||
XCTAssertTrue(expanded.contains("onedrive"))
|
||||
}
|
||||
|
||||
func testSingleEndpointNeverExpandsEvenIfHuge() {
|
||||
// The exact case we saw live: Google at 126 MB but only one active IP.
|
||||
let policy = ExpansionPolicy()
|
||||
let expanded = policy.update([D(key: "google", bytes: 126_000_000, endpointCount: 1)])
|
||||
XCTAssertTrue(expanded.isEmpty)
|
||||
}
|
||||
|
||||
func testQuietOrgDoesNotExpand() {
|
||||
let policy = ExpansionPolicy(floorBytes: 8 * 1024)
|
||||
let expanded = policy.update([D(key: "apple", bytes: 100, endpointCount: 5)])
|
||||
XCTAssertTrue(expanded.isEmpty)
|
||||
}
|
||||
|
||||
func testOnlyTopKExpand() {
|
||||
let policy = ExpansionPolicy(topK: 2)
|
||||
let districts = (0..<5).map { D(key: "org\($0)", bytes: UInt64(1000 - $0) * 1024, endpointCount: 3) }
|
||||
let expanded = policy.update(districts)
|
||||
XCTAssertEqual(expanded, ["org0", "org1"]) // two busiest only
|
||||
}
|
||||
|
||||
func testHysteresisKeepsExpandedThenCollapses() {
|
||||
let policy = ExpansionPolicy(ttlTicks: 3)
|
||||
// Tick 1: busy -> expands.
|
||||
XCTAssertTrue(policy.update([D(key: "x", bytes: 1_000_000, endpointCount: 4)]).contains("x"))
|
||||
// Now idle, but still has endpoints: stays up for the TTL tail.
|
||||
XCTAssertTrue(policy.update([D(key: "x", bytes: 0, endpointCount: 4)]).contains("x")) // ttl 2
|
||||
XCTAssertTrue(policy.update([D(key: "x", bytes: 0, endpointCount: 4)]).contains("x")) // ttl 1
|
||||
XCTAssertFalse(policy.update([D(key: "x", bytes: 0, endpointCount: 4)]).contains("x")) // ttl 0 -> collapsed
|
||||
}
|
||||
|
||||
func testExpiredTimerWithLostEndpointsStaysCollapsed() {
|
||||
let policy = ExpansionPolicy(ttlTicks: 2)
|
||||
_ = policy.update([D(key: "x", bytes: 1_000_000, endpointCount: 4)])
|
||||
// Endpoints drop to 1 while timer still alive -> nothing to fan out.
|
||||
XCTAssertFalse(policy.update([D(key: "x", bytes: 0, endpointCount: 1)]).contains("x"))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user