88 lines
2.1 KiB
Swift
88 lines
2.1 KiB
Swift
|
//
|
||
|
|
// MoonrakerSocketManagerTests.swift
|
||
|
|
// SoyuzTests
|
||
|
|
//
|
||
|
|
// Created by maddiefuzz on 2/21/23.
|
||
|
|
//
|
||
|
|
|
||
|
|
import XCTest
|
||
|
|
import Starscream
|
||
|
|
import Combine
|
||
|
|
import Network
|
||
|
|
@testable import Soyuz
|
||
|
|
|
||
|
|
class DummyEngine: Engine {
|
||
|
|
var delegate: Starscream.EngineDelegate?
|
||
|
|
|
||
|
|
@Published var startCalled = false
|
||
|
|
|
||
|
|
func register(delegate: Starscream.EngineDelegate) {
|
||
|
|
self.delegate = delegate
|
||
|
|
}
|
||
|
|
|
||
|
|
func start(request: URLRequest) {
|
||
|
|
startCalled = true
|
||
|
|
}
|
||
|
|
|
||
|
|
func stop(closeCode: UInt16) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func forceStop() {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func write(data: Data, opcode: Starscream.FrameOpCode, completion: (() -> ())?) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func write(string: String, completion: (() -> ())?) {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
class MoonrakerSocketManagerTests: XCTestCase {
|
||
|
|
var socketManager: MoonrakerSocketManager?
|
||
|
|
var bonjourListener: NWListener?
|
||
|
|
var engine: DummyEngine!
|
||
|
|
var cancellable: AnyCancellable?
|
||
|
|
|
||
|
|
override func setUp() {
|
||
|
|
engine = DummyEngine()
|
||
|
|
socketManager = MoonrakerSocketManager(starscreamEngine: engine!)
|
||
|
|
|
||
|
|
do {
|
||
|
|
bonjourListener = try NWListener(using: .tcp, on: .http)
|
||
|
|
|
||
|
|
bonjourListener!.newConnectionHandler = { newConnection in
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
bonjourListener!.start(queue: DispatchQueue.main)
|
||
|
|
} catch {
|
||
|
|
print("Error: \(error)")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
|
||
|
|
func testConnectToBonjourEndpoint() {
|
||
|
|
let endpoint = NWEndpoint.hostPort(host: "localhost", port: .http)
|
||
|
|
print("Trying to connect to bonjour endpoint \(endpoint)")
|
||
|
|
|
||
|
|
let expectation = XCTestExpectation(description: "MoonrakerSocketManager.connectToBonjourEndpoint opens Starscream socket")
|
||
|
|
|
||
|
|
cancellable = engine.$startCalled
|
||
|
|
.sink(receiveValue: { newValue in
|
||
|
|
if newValue == true {
|
||
|
|
expectation.fulfill()
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
socketManager?.connectToBonjourEndpoint(endpoint)
|
||
|
|
wait(for: [expectation], timeout: 2)
|
||
|
|
XCTAssertTrue(engine.startCalled)
|
||
|
|
}
|
||
|
|
}
|