def test_publish_user(self, producer_pool): payload = {'action': 'create', 'user': {'id': 'foobar'}} producer = producer_pool['foobar'].acquire().__enter__() exchange = realtime.get_exchange() request = DummyRequest() publisher = realtime.Publisher(request) publisher.publish_user(payload) producer.publish.assert_called_once_with(payload, exchange=exchange, declare=[exchange], routing_key='user')
def test_publish_user(self, matchers, producer_pool, pyramid_request): payload = {'action': 'create', 'user': {'id': 'foobar'}} producer = producer_pool['foobar'].acquire().__enter__() exchange = realtime.get_exchange() publisher = realtime.Publisher(pyramid_request) publisher.publish_user(payload) expected_headers = matchers.MappingContaining('timestamp') producer.publish.assert_called_once_with(payload, exchange=exchange, declare=[exchange], routing_key='user', headers=expected_headers)
def test_publish_user(self, producer_pool, pyramid_request, retry_policy): payload = {"action": "create", "user": {"id": "foobar"}} producer = producer_pool["foobar"].acquire().__enter__() exchange = realtime.get_exchange() publisher = realtime.Publisher(pyramid_request) publisher.publish_user(payload) producer.publish.assert_called_once_with( payload, exchange=exchange, declare=[exchange], routing_key="user", retry=True, retry_policy=retry_policy, )
def test_publish_annotation(self, producer_pool, pyramid_request, retry_policy): payload = {"action": "create", "annotation": {"id": "foobar"}} producer = producer_pool["foobar"].acquire().__enter__() exchange = realtime.get_exchange() publisher = realtime.Publisher(pyramid_request) publisher.publish_annotation(payload) producer.publish.assert_called_once_with( payload, exchange=exchange, declare=[exchange], routing_key="annotation", headers=Any.dict.containing(["timestamp"]), retry=True, retry_policy=retry_policy, )
def publisher(self, pyramid_request): return realtime.Publisher(pyramid_request)