Merge pull request #164 from pusher/unsubsribe-all

Unsubsribe all

Author: hamchapman

Sources/PusherConnection.swift

@@ -176,6 +176,15 @@ public typealias PusherEventJSON = [String: AnyObject]
             self.channels.remove(name: channelName)
         }
     }
+    
+    /**
+        Unsubscribes from all PusherChannels
+    */
+    internal func unsubscribeAll() {
+        for (_, channel) in channels.channels {
+            unsubscribe(channelName: channel.name)
+        }
+    }
 
     /**
         Either writes a string directly to the websocket with the given event name

Sources/PusherSwift.swift

@@ -129,6 +129,13 @@ let CLIENT_NAME = "pusher-websocket-swift"
     open func unsubscribe(_ channelName: String) {
         self.connection.unsubscribe(channelName: channelName)
     }
+    
+    /**
+        Unsubscribes the client from all channels
+    */
+    open func unsubscribeAll() {
+        self.connection.unsubscribeAll()
+    }
 
     /**
         Binds the client's global channel to all events

Tests/Mocks.swift

@@ -60,6 +60,14 @@ open class MockWebSocket: WebSocket {
                     self.delegate?.websocketDidReceiveMessage(socket: self, text: "{\"event\":\"pusher_internal:subscription_succeeded\",\"channel\":\"test-channel\",\"data\":\"{}\"}")
                 }
             )
+        } else if string == "{\"data\":{\"channel\":\"test-channel2\"},\"event\":\"pusher:subscribe\"}" || string == "{\"event\":\"pusher:subscribe\",\"data\":{\"channel\":\"test-channel2\"}}" {
+            let _ = stubber.stub(
+                functionName: "writeString",
+                args: [string],
+                functionToCall: {
+                    self.delegate?.websocketDidReceiveMessage(socket: self, text: "{\"event\":\"pusher_internal:subscription_succeeded\",\"channel\":\"test-channel2\",\"data\":\"{}\"}")
+            }
+            )
         } else if stringContainsElements(string, elements: ["testkey123:6aae8814fabd5285245422096705abbed64ea59614648814ffb0bf2dc5d19168", "private-channel", "pusher:subscribe"]) {
             let _ = stubber.stub(
                 functionName: "writeString",
@@ -144,6 +152,12 @@ open class MockWebSocket: WebSocket {
                 args: [string],
                 functionToCall: nil
             )
+        } else if stringContainsElements(string, elements: ["test-channel2", "pusher:unsubscribe"]) {
+            let _ = stubber.stub(
+                functionName: "writeString",
+                args: [string],
+                functionToCall: nil
+            )
         } else if stringContainsElements(string, elements: ["testkey123:736f0b19c2e56f985f3e6faa38db5b69d39305bc8519952c8f9f5595d69fcb3d", "presence-test", "user_id", "123", "pusher:subscribe", "user_info", "twitter", "hamchapman"]) {
             let _ = stubber.stub(
                 functionName: "writeString",

Tests/PusherTopLevelAPITests.swift

@@ -232,6 +232,24 @@ class PusherTopLevelApiTests: XCTestCase {
         XCTAssertTrue(parsedEqualsExpected)
     }
 
+    func testUnsubscribingFromAllChannelsRemovesTheChannels() {
+        pusher.connect()
+        let _ = pusher.subscribe("test-channel")
+        let _ = pusher.subscribe("test-channel2")
+        XCTAssertEqual(pusher.connection.channels.channels.count, 2, "should have 2 channels")
+        XCTAssertEqual(socket.stubber.calls.last?.name, "writeString", "write function should have been called")
+        pusher.unsubscribeAll()
+
+        XCTAssertEqual(socket.stubber.calls.last?.name, "writeString", "write function should have been called")
+
+        let parsedSubscribeArgs = convertStringToDictionary(socket.stubber.calls.last?.args!.first as! String)
+        let expectedDict = ["data": ["channel": "test-channel2"], "event": "pusher:unsubscribe"] as [String: Any]
+        let parsedEqualsExpected = NSDictionary(dictionary: parsedSubscribeArgs!).isEqual(to: NSDictionary(dictionary: expectedDict) as [NSObject: AnyObject])
+
+        XCTAssertTrue(parsedEqualsExpected)
+        XCTAssertEqual(pusher.connection.channels.channels.count, 0, "should have no channels")
+    }
+
     /* global channel interactions */
 
     func testBindingToEventsGloballyAddsACallbackToTheGlobalChannel() {