From c6d000972e385b9533fdde424580b54647b4d6b5 Mon Sep 17 00:00:00 2001
From: Erik <nuclear.ace@gmail.com>
Date: Sun, 22 May 2016 14:10:43 -0400
Subject: [PATCH] add test

---
 SocketIO-MacTests/SocketSideEffectTest.swift | 6 ++++++
 Source/SocketIOClient.swift                  | 4 ++++
 Source/SocketIOClientSpec.swift              | 4 ----
 3 files changed, 10 insertions(+), 4 deletions(-)

diff --git a/SocketIO-MacTests/SocketSideEffectTest.swift b/SocketIO-MacTests/SocketSideEffectTest.swift
index a536693..5e5e8d8 100644
--- a/SocketIO-MacTests/SocketSideEffectTest.swift
+++ b/SocketIO-MacTests/SocketSideEffectTest.swift
@@ -137,6 +137,12 @@ class SocketSideEffectTest: XCTestCase {
         waitForExpectations(withTimeout: 3, handler: nil)
     }
     
+    func testSocketDataToAnyObject() {
+        let data = ["test", 1, 2.2, ["Hello": 2, "bob": 2.2], true, [1, 2], [1.1, 2]] as [SocketData]
+        
+        XCTAssertEqual(data.count, socket.socketDataToAnyObject(data: data).count)
+    }
+    
     func testHandleMultipleBinaryEvent() {
         let expect = expectation(withDescription: "handled multiple binary event")
         socket.on("test") {data, ack in
diff --git a/Source/SocketIOClient.swift b/Source/SocketIOClient.swift
index 5c8c380..72320a8 100644
--- a/Source/SocketIOClient.swift
+++ b/Source/SocketIOClient.swift
@@ -405,6 +405,10 @@ public final class SocketIOClient : NSObject, SocketEngineClient, SocketParsable
     public func removeAllHandlers() {
         handlers.removeAll(keepingCapacity: false)
     }
+    
+    func socketDataToAnyObject(data: [SocketData]) -> [AnyObject] {
+        return data.flatMap({$0 as? AnyObject})
+    }
 
     private func tryReconnect(reason: String) {
         if reconnecting {
diff --git a/Source/SocketIOClientSpec.swift b/Source/SocketIOClientSpec.swift
index 17be36c..1e2051c 100644
--- a/Source/SocketIOClientSpec.swift
+++ b/Source/SocketIOClientSpec.swift
@@ -40,8 +40,4 @@ extension SocketIOClientSpec {
         
         handleEvent("error", data: [reason as AnyObject], isInternalMessage: true, withAck: -1)
     }
-    
-    func socketDataToAnyObject(data: [SocketData]) -> [AnyObject] {
-        return data.flatMap({$0 as? AnyObject})
-    }
 }