         socket = WebSocket(url: URL(string: "wss://echo.websocket.org")!)
         socket.delegate = self
     }
+
+    // MARK: - Test methods
 
     func testConnect() {
         connectExpectation = XCTestExpectation(description: "connectExpectation")


    func webSocketDidConnect(connection: WebSocketConnection) {
        connectExpectation?.fulfill()
        if shouldDisconnectImmediately {
            socket.disconnect()
        }

    }

    var shouldDisconnectImmediately: Bool!
    var receivedPongTimestamps = [Date]()
    static let expectationTimeout = 5.0
    static let repeatedPingInterval = 0.5

    override func setUp() {
        super.setUp()
