267 lines
8.4 KiB
Swift
267 lines
8.4 KiB
Swift
//
|
|
// SocketSideEffectTest.swift
|
|
// Socket.IO-Client-Swift
|
|
//
|
|
// Created by Erik Little on 10/11/15.
|
|
//
|
|
//
|
|
|
|
import XCTest
|
|
@testable import SocketIO
|
|
|
|
class SocketSideEffectTest: XCTestCase {
|
|
func testInitialCurrentAck() {
|
|
XCTAssertEqual(socket.currentAck, -1)
|
|
}
|
|
|
|
func testFirstAck() {
|
|
socket.emitWithAck("test").timingOut(after: 0) {data in}
|
|
XCTAssertEqual(socket.currentAck, 0)
|
|
}
|
|
|
|
func testSecondAck() {
|
|
socket.emitWithAck("test").timingOut(after: 0) {data in}
|
|
socket.emitWithAck("test").timingOut(after: 0) {data in}
|
|
|
|
XCTAssertEqual(socket.currentAck, 1)
|
|
}
|
|
|
|
func testHandleAck() {
|
|
let expect = expectation(description: "handled ack")
|
|
socket.emitWithAck("test").timingOut(after: 0) {data in
|
|
XCTAssertEqual(data[0] as? String, "hello world")
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.parseSocketMessage("30[\"hello world\"]")
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testHandleAck2() {
|
|
let expect = expectation(description: "handled ack2")
|
|
socket.emitWithAck("test").timingOut(after: 0) {data in
|
|
XCTAssertTrue(data.count == 2, "Wrong number of ack items")
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.parseSocketMessage("61-0[{\"_placeholder\":true,\"num\":0},{\"test\":true}]")
|
|
socket.parseBinaryData(Data())
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testHandleEvent() {
|
|
let expect = expectation(description: "handled event")
|
|
socket.on("test") {data, ack in
|
|
XCTAssertEqual(data[0] as? String, "hello world")
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.parseSocketMessage("2[\"test\",\"hello world\"]")
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testHandleStringEventWithQuotes() {
|
|
let expect = expectation(description: "handled event")
|
|
socket.on("test") {data, ack in
|
|
XCTAssertEqual(data[0] as? String, "\"hello world\"")
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.parseSocketMessage("2[\"test\",\"\\\"hello world\\\"\"]")
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testHandleOnceEvent() {
|
|
let expect = expectation(description: "handled event")
|
|
socket.once("test") {data, ack in
|
|
XCTAssertEqual(data[0] as? String, "hello world")
|
|
XCTAssertEqual(self.socket.testHandlers.count, 0)
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.parseSocketMessage("2[\"test\",\"hello world\"]")
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testOffWithEvent() {
|
|
socket.on("test") {data, ack in }
|
|
XCTAssertEqual(socket.testHandlers.count, 1)
|
|
socket.on("test") {data, ack in }
|
|
XCTAssertEqual(socket.testHandlers.count, 2)
|
|
socket.off("test")
|
|
XCTAssertEqual(socket.testHandlers.count, 0)
|
|
}
|
|
|
|
func testOffWithId() {
|
|
let handler = socket.on("test") {data, ack in }
|
|
XCTAssertEqual(socket.testHandlers.count, 1)
|
|
socket.on("test") {data, ack in }
|
|
XCTAssertEqual(socket.testHandlers.count, 2)
|
|
socket.off(id: handler)
|
|
XCTAssertEqual(socket.testHandlers.count, 1)
|
|
}
|
|
|
|
func testHandlesErrorPacket() {
|
|
let expect = expectation(description: "Handled error")
|
|
socket.on("error") {data, ack in
|
|
if let error = data[0] as? String, error == "test error" {
|
|
expect.fulfill()
|
|
}
|
|
}
|
|
|
|
socket.parseSocketMessage("4\"test error\"")
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testHandleBinaryEvent() {
|
|
let expect = expectation(description: "handled binary event")
|
|
socket.on("test") {data, ack in
|
|
if let dict = data[0] as? NSDictionary, let data = dict["test"] as? NSData {
|
|
XCTAssertEqual(data as Data, self.data)
|
|
expect.fulfill()
|
|
}
|
|
}
|
|
|
|
socket.parseSocketMessage("51-[\"test\",{\"test\":{\"_placeholder\":true,\"num\":0}}]")
|
|
socket.parseBinaryData(data)
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testHandleMultipleBinaryEvent() {
|
|
let expect = expectation(description: "handled multiple binary event")
|
|
socket.on("test") {data, ack in
|
|
if let dict = data[0] as? NSDictionary, let data = dict["test"] as? NSData,
|
|
let data2 = dict["test2"] as? NSData {
|
|
XCTAssertEqual(data as Data, self.data)
|
|
XCTAssertEqual(data2 as Data, self.data2)
|
|
expect.fulfill()
|
|
}
|
|
}
|
|
|
|
socket.parseSocketMessage("52-[\"test\",{\"test\":{\"_placeholder\":true,\"num\":0},\"test2\":{\"_placeholder\":true,\"num\":1}}]")
|
|
socket.parseBinaryData(data)
|
|
socket.parseBinaryData(data2)
|
|
waitForExpectations(timeout: 3, handler: nil)
|
|
}
|
|
|
|
func testSocketManager() {
|
|
let manager = SocketClientManager.sharedManager
|
|
manager["test"] = socket
|
|
|
|
XCTAssert(manager["test"] === socket, "failed to get socket")
|
|
|
|
manager["test"] = nil
|
|
|
|
XCTAssert(manager["test"] == nil, "socket not removed")
|
|
|
|
}
|
|
|
|
func testChangingStatusCallsStatusChangeHandler() {
|
|
let expect = expectation(description: "The client should announce when the status changes")
|
|
let statusChange = SocketIOClientStatus.connecting
|
|
|
|
socket.on("statusChange") {data, ack in
|
|
guard let status = data[0] as? SocketIOClientStatus else {
|
|
XCTFail("Status should be one of the defined statuses")
|
|
|
|
return
|
|
}
|
|
|
|
XCTAssertEqual(status, statusChange, "The status changed should be the one set")
|
|
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.setTestStatus(statusChange)
|
|
|
|
waitForExpectations(timeout: 0.2)
|
|
}
|
|
|
|
func testOnClientEvent() {
|
|
let expect = expectation(description: "The client should call client event handlers")
|
|
let event = SocketClientEvent.disconnect
|
|
let closeReason = "testing"
|
|
|
|
socket.on(clientEvent: event) {data, ack in
|
|
guard let reason = data[0] as? String else {
|
|
XCTFail("Client should pass data for client events")
|
|
|
|
return
|
|
}
|
|
|
|
XCTAssertEqual(closeReason, reason, "The data should be what was sent to handleClientEvent")
|
|
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.handleClientEvent(event, data: [closeReason])
|
|
|
|
waitForExpectations(timeout: 0.2)
|
|
}
|
|
|
|
func testClientEventsAreBackwardsCompatible() {
|
|
let expect = expectation(description: "The client should call old style client event handlers")
|
|
let event = SocketClientEvent.disconnect
|
|
let closeReason = "testing"
|
|
|
|
socket.on("disconnect") {data, ack in
|
|
guard let reason = data[0] as? String else {
|
|
XCTFail("Client should pass data for client events")
|
|
|
|
return
|
|
}
|
|
|
|
XCTAssertEqual(closeReason, reason, "The data should be what was sent to handleClientEvent")
|
|
|
|
expect.fulfill()
|
|
}
|
|
|
|
socket.handleClientEvent(event, data: [closeReason])
|
|
|
|
waitForExpectations(timeout: 0.2)
|
|
}
|
|
|
|
func testConnectTimesOutIfNotConnected() {
|
|
let expect = expectation(description: "The client should call the timeout function")
|
|
|
|
socket.setTestStatus(.notConnected)
|
|
|
|
socket.connect(timeoutAfter: 1, withHandler: {
|
|
expect.fulfill()
|
|
})
|
|
|
|
waitForExpectations(timeout: 2)
|
|
}
|
|
|
|
func testConnectDoesNotTimeOutIfConnected() {
|
|
let expect = expectation(description: "The client should not call the timeout function")
|
|
|
|
socket.setTestStatus(.notConnected)
|
|
|
|
socket.connect(timeoutAfter: 1, withHandler: {
|
|
XCTFail("Should not call timeout handler if status is connected")
|
|
})
|
|
|
|
DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 0.2) {
|
|
// Fake connecting
|
|
self.socket.setTestStatus(.connected)
|
|
}
|
|
|
|
DispatchQueue.main.asyncAfter(deadline: DispatchTime.now() + 1.1) {
|
|
expect.fulfill()
|
|
}
|
|
|
|
waitForExpectations(timeout: 2)
|
|
}
|
|
|
|
let data = "test".data(using: String.Encoding.utf8)!
|
|
let data2 = "test2".data(using: String.Encoding.utf8)!
|
|
private var socket: SocketIOClient!
|
|
|
|
override func setUp() {
|
|
super.setUp()
|
|
socket = SocketIOClient(socketURL: URL(string: "http://localhost/")!)
|
|
socket.setTestable()
|
|
}
|
|
}
|