TestSocketPort.swift

// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
#if os(Windows)
import WinSDK
#elseif canImport(Android)
@preconcurrencyimport Android
#endif

classTestPortDelegateWithBlock:NSObject,PortDelegate{
letblock:(PortMessage)->Void

init(block:@escaping(PortMessage)->Void){
self.block = block
}

func handle(_ message:PortMessage){
block(message)
}
}

classTestSocketPort:XCTestCase{

func tcpOrUdpPort(of socketPort:SocketPort)->Int?{
letdata= socketPort.address

#if canImport(Darwin) || os(FreeBSD) || os(OpenBSD)
letfamilyOffset=1
#else
letfamilyOffset=0
#endif

ifdata[data.startIndex + familyOffset]== AF_INET {
return data.withUnsafeBytes{(buffer)in
varsin=sockaddr_in()
withUnsafeMutableBytes(of:&sin){
 $0.copyMemory(from: buffer)
}

returnInt(sin.sin_port.bigEndian)
}
}elseifdata[data.startIndex + familyOffset]== AF_INET6 {
return data.withUnsafeBytes{(buffer)in
varsin=sockaddr_in6()
withUnsafeMutableBytes(of:&sin){
 $0.copyMemory(from: buffer)
}

returnInt(sin.sin6_port.bigEndian)
}
}else{
returnnil
}
}

func testRemoteSocketPortsAreUniqued(){
leta=SocketPort(remoteWithTCPPort:10000, host:"localhost")
letb=SocketPort(remoteWithTCPPort:10000, host:"localhost")
XCTAssertEqual(a, b)
}

func testInitPicksATCPPort()throws{
letlocal=tryXCTUnwrap(SocketPort(tcpPort:0))
defer{ local.invalidate()}

letport=tryXCTUnwrap(tcpOrUdpPort(of: local))
XCTAssertNotEqual(port,0)
XCTAssert(port >=1024)
}

func testSendingOneMessageRemoteToLocal()throws{
letlocal=tryXCTUnwrap(SocketPort(tcpPort:0))
defer{ local.invalidate()}

lettcpPort=tryUInt16(XCTUnwrap(tcpOrUdpPort(of: local)))

letremote=tryXCTUnwrap(SocketPort(remoteWithTCPPort: tcpPort, host:"localhost"))
defer{ remote.invalidate()}

letdata=Data("I cannot weave".utf8)

letreceived=expectation(description:"Message received")
letdelegate=TestPortDelegateWithBlock{ message in
XCTAssertEqual(message.components as?[AnyHashable],[data asNSData])
 received.fulfill()
}

withExtendedLifetime(delegate){
 local.setDelegate(delegate)
 local.schedule(in:.main, forMode:.default)
 remote.schedule(in:.main, forMode:.default)

defer{
 local.setDelegate(nil)
 local.remove(from:.main, forMode:.default)
 remote.remove(from:.main, forMode:.default)
}

letsent= remote.send(before:Date(timeIntervalSinceNow:5), components:NSMutableArray(array:[data]), from:nil, reserved:0)
XCTAssertTrue(sent)

waitForExpectations(timeout:5.5)
}
}

func testSendingMultipleMessagesRemoteToLocal()throws{
varlocalPorts=[SocketPort]()
varremotePorts=[SocketPort]()
vardelegates=[TestPortDelegateWithBlock]()

letdata=Data("I cannot weave".utf8)

for_in0..<128{
letlocal=tryXCTUnwrap(SocketPort(tcpPort:0))
lettcpPort=tryUInt16(XCTUnwrap(tcpOrUdpPort(of: local)))
letremote=tryXCTUnwrap(SocketPort(remoteWithTCPPort: tcpPort, host:"localhost"))

letreceived=expectation(description:"Message received")

letlocalDelegate=TestPortDelegateWithBlock{ message in
XCTAssertEqual(message.components as?[AnyHashable],[data asNSData])
 received.fulfill()
}

 localPorts.append(local)
 remotePorts.append(remote)
 delegates.append(localDelegate)

 local.setDelegate(localDelegate)
 local.schedule(in:.main, forMode:.default)
 remote.schedule(in:.main, forMode:.default)
}

withExtendedLifetime(delegates){
forremotein remotePorts {
letsent= remote.send(before:Date(timeIntervalSinceNow:5), components:NSMutableArray(array:[data]), from:nil, reserved:0)
XCTAssertTrue(sent)
}
waitForExpectations(timeout:5.0)
}

forportin localPorts + remotePorts {
 port.setDelegate(nil)
 port.remove(from:.main, forMode:.default)
 port.invalidate()
}
}
}