35 lines
1.1 KiB
Swift
35 lines
1.1 KiB
Swift
|
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)
|
||
|
|
}
|
||
|
|
}
|