def test_bogus_payload(self, send_message):
p = PubSubProtocol()
p.send("topic", p)
p.send(p, {'data': 'bar'})
p.send(p, p)
self.assertFalse(send_message.called)
def test_bogus_payload(self, instance):
p = PubSubProtocol()
p.onMessage(b'', True)
p.onMessage(b'foo', True)
p.onMessage(b'{foo}', True)
p.onMessage(b'{"foo"}', True)
self.assertFalse(instance.called)
def test_jsonify(self, send_message):
now = datetime.datetime.now()
p = PubSubProtocol()
p.send("foo", now)
args, kwargs = send_message.call_args
actual = json.loads(args[0].decode('utf8'))
expected = {"topic": "foo", "data": now.isoformat()}
self.assertEqual(actual, expected)
def test_request_command(self, send, handle):
p = PubSubProtocol()
p.onMessage(b'{"request":"topic"}', True)
handle.assert_called_once_with({"request": "topic"})
send.assert_called_once_with("topic", "response")
def test_unsubscribe_command(self, unsubscribe):
p = PubSubProtocol()
p.onMessage(b'{"unsubscribe":"topic"}', True)
unsubscribe.assert_called_once_with(p, "topic")
def test_subscribe_publish_one(self, publish_one):
p = PubSubProtocol()
p.onMessage(b'{"subscribe":"topic"}', True)
publish_one.assert_called_once_with(p, "topic")
def test_unknown_command(self, instance):
p = PubSubProtocol()
p.onMessage(b'{"foo":"bar"}', True)
self.assertFalse(instance.called)
def test_unsubscribe_all(self, unsubscribe_all):
p = PubSubProtocol()
p.onClose(True, 0, None)
unsubscribe_all.assert_called_once_with(p)
def test_no_op(self):
p = PubSubProtocol()
p.onOpen()
def test_no_op(self):
p = PubSubProtocol()
p.onConnect(None)
def test_send_message(self, send_message):
p = PubSubProtocol()
p.send("foo", "bar")
self.assertEqual(send_message.call_count, 1)
def test_str(self):
p = PubSubProtocol()
p.peer = "foo"
self.assertEqual(str(p), "foo")
