import XCTest
@testable import NetworkCityCore

final class TrafficClassTests: XCTestCase {

    func testQuicByProtocol() {
        XCTAssertEqual(TrafficClass.classify(proto: .quic4, port: 443), .quic)
        XCTAssertEqual(TrafficClass.classify(proto: .quic6, port: 12345), .quic)
    }

    func testHTTPSOverTCP() {
        XCTAssertEqual(TrafficClass.classify(proto: .tcp4, port: 443), .https)
    }

    func testUDP443IsQuic() {
        XCTAssertEqual(TrafficClass.classify(proto: .udp4, port: 443), .quic)
    }

    func testDNS() {
        XCTAssertEqual(TrafficClass.classify(proto: .udp4, port: 53), .dns)
        XCTAssertEqual(TrafficClass.classify(proto: .udp6, port: 5353), .dns)
        XCTAssertEqual(TrafficClass.classify(proto: .tcp4, port: 53), .dns)
    }

    func testHTTP() {
        XCTAssertEqual(TrafficClass.classify(proto: .tcp4, port: 80), .http)
        XCTAssertEqual(TrafficClass.classify(proto: .tcp4, port: 8080), .http)
    }

    func testOther() {
        XCTAssertEqual(TrafficClass.classify(proto: .tcp4, port: 22), .other)
        XCTAssertEqual(TrafficClass.classify(proto: .tcp4, port: nil), .other)
    }
}