aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorJon Moss <me@jonathanmoss.me>2016-05-27 14:37:26 -0400
committerJon Moss <me@jonathanmoss.me>2016-05-31 13:06:11 -0400
commitfb0f0c7f6eddcc6a566093bed5a6e94404f883ca (patch)
tree0570d2f180a87eebf713b5a330f19e750952b70e
parent66901c1849efae74c8a58fe0cb36afd487c067cc (diff)
downloadrails-fb0f0c7f6eddcc6a566093bed5a6e94404f883ca.tar.gz
rails-fb0f0c7f6eddcc6a566093bed5a6e94404f883ca.tar.bz2
rails-fb0f0c7f6eddcc6a566093bed5a6e94404f883ca.zip
Reorganize `MockWebSocket`
-rw-r--r--actioncable/test/javascript/src/test_helpers/mock_websocket.coffee3
-rw-r--r--actioncable/test/javascript/src/unit/consumer_test.coffee51
2 files changed, 32 insertions, 22 deletions
diff --git a/actioncable/test/javascript/src/test_helpers/mock_websocket.coffee b/actioncable/test/javascript/src/test_helpers/mock_websocket.coffee
index 6508984c37..147a336efb 100644
--- a/actioncable/test/javascript/src/test_helpers/mock_websocket.coffee
+++ b/actioncable/test/javascript/src/test_helpers/mock_websocket.coffee
@@ -1,10 +1,11 @@
#= require mock-socket
+ActionCable.TestHelpers.MockWebSocket = MockWebSocket
+
server = null
consumer = null
ActionCable.TestHelpers.createConsumer = (url, callback) ->
- ActionCable.WebSocket = MockWebSocket
server = new MockServer url
consumer = ActionCable.createConsumer(url)
callback(consumer, server)
diff --git a/actioncable/test/javascript/src/unit/consumer_test.coffee b/actioncable/test/javascript/src/unit/consumer_test.coffee
index d8b1450ad8..fd76b086ed 100644
--- a/actioncable/test/javascript/src/unit/consumer_test.coffee
+++ b/actioncable/test/javascript/src/unit/consumer_test.coffee
@@ -1,31 +1,40 @@
{module, test} = QUnit
-{testURL, createConsumer} = ActionCable.TestHelpers
+{testURL, MockWebSocket, createConsumer} = ActionCable.TestHelpers
module "ActionCable.Consumer", ->
- test "#connect", (assert) ->
- done = assert.async()
+ module 'with MockWebSocket', (hooks) ->
+ originalWebSocket = ActionCable.WebSocket
- createConsumer testURL, (consumer, server) ->
- server.on "connection", ->
- clients = server.clients()
- assert.equal clients.length, 1
- assert.equal clients[0].readyState, WebSocket.OPEN
- done()
+ hooks.beforeEach ->
+ ActionCable.WebSocket = MockWebSocket
- consumer.connect()
+ hooks.afterEach ->
+ ActionCable.WebSocket = originalWebSocket
- test "#disconnect", (assert) ->
- done = assert.async()
+ test "#connect", (assert) ->
+ done = assert.async()
- createConsumer testURL, (consumer, server) ->
- server.on "connection", ->
- clients = server.clients()
- assert.equal clients.length, 1
-
- clients[0].addEventListener "close", (event) ->
- assert.equal event.type, "close"
+ createConsumer testURL, (consumer, server) ->
+ server.on "connection", ->
+ clients = server.clients()
+ assert.equal clients.length, 1
+ assert.equal clients[0].readyState, WebSocket.OPEN
done()
- consumer.disconnect()
+ consumer.connect()
+
+ test "#disconnect", (assert) ->
+ done = assert.async()
+
+ createConsumer testURL, (consumer, server) ->
+ server.on "connection", ->
+ clients = server.clients()
+ assert.equal clients.length, 1
+
+ clients[0].addEventListener "close", (event) ->
+ assert.equal event.type, "close"
+ done()
+
+ consumer.disconnect()
- consumer.connect()
+ consumer.connect()