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_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