def test_basic_broadcast(self, ws1, ws2):
WebSocketDispatcher.broadcast('bar', trigger='manual')
ws1.trigger.assert_called_with(client='client-1',
callback='callback-1',
trigger='manual')
assert not ws2.trigger.called
assert not ws1.unsubscribe_all.called and not ws2.unsubscribe_all.called
def test_broadcast_error(ws4, monkeypatch):
monkeypatch.setattr(log, 'warn', Mock())
WebSocketDispatcher.broadcast('foo')
assert not ws4.trigger.called and not log.warn.called
WebSocketDispatcher.broadcast('baf')
assert ws4.trigger.called and log.warn.called
def test_multi_broadcast(ws1, ws2, ws3, ws4):
WebSocketDispatcher.broadcast(['foo', 'bar'])
assert ws1.trigger.called and ws2.trigger.called and not ws3.trigger.called and not ws4.trigger.called
def test_broadcast_with_originating_client(ws1, ws2):
WebSocketDispatcher.broadcast('foo', originating_client='client-1')
assert ws2.trigger.called and not ws1.trigger.called
def test_broadcast_error(ws4, monkeypatch):
monkeypatch.setattr(log, 'warn', Mock())
WebSocketDispatcher.broadcast('foo')
assert not ws4.trigger.called and not log.warn.called
WebSocketDispatcher.broadcast('baf')
assert ws4.trigger.called and log.warn.called
def test_multi_broadcast(ws1, ws2, ws3, ws4):
WebSocketDispatcher.broadcast(['foo', 'bar'])
assert ws1.trigger.called and ws2.trigger.called and not ws3.trigger.called and not ws4.trigger.called
def test_broadcast_with_originating_client(ws1, ws2):
WebSocketDispatcher.broadcast('foo', originating_client='client-1')
assert ws2.trigger.called and not ws1.trigger.called
def test_basic_broadcast(ws1, ws2):
WebSocketDispatcher.broadcast('bar', trigger='manual')
ws1.trigger.assert_called_with(client='client-1', callback='callback-1', trigger='manual')
assert not ws2.trigger.called
def test_broadcast_closed(self, ws1, ws2):
ws1.is_closed = True
WebSocketDispatcher.broadcast('foo')
assert ws2.trigger.called and not ws1.trigger.called
assert ws1.unsubscribe_all.called and not ws2.unsubscribe_all.called
