def test_rpc_consumer(get_rpc_exchange):
container = Mock(spec=ServiceContainer)
container.service_name = "exampleservice"
container.service_cls = Mock(rpcmethod=lambda: None)
exchange = Mock()
get_rpc_exchange.return_value = exchange
queue_consumer = Mock(spec=QueueConsumer)
queue_consumer.bind("queue_consumer", container)
consumer = RpcConsumer()
consumer.queue_consumer = queue_consumer
consumer.bind("rpc_consumer", container)
provider = RpcProvider()
provider.rpc_consumer = consumer
provider.bind("rpcmethod", container)
provider.prepare()
consumer.prepare()
queue_consumer.prepare()
queue = consumer.queue
assert queue.name == "rpc-exampleservice"
assert queue.routing_key == "exampleservice.*"
assert queue.exchange == exchange
assert queue.durable
queue_consumer.register_provider.assert_called_once_with(consumer)
consumer.register_provider(provider)
assert consumer._providers == set([provider])
routing_key = "exampleservice.rpcmethod"
assert consumer.get_provider_for_method(routing_key) == provider
routing_key = "exampleservice.invalidmethod"
with pytest.raises(MethodNotFound):
consumer.get_provider_for_method(routing_key)
consumer.unregister_provider(provider)
assert consumer._providers == set()
def test_rpc_consumer(get_rpc_exchange):
container = Mock(spec=ServiceContainer)
container.service_name = "exampleservice"
container.service_cls = Mock(rpcmethod=lambda: None)
exchange = Mock()
get_rpc_exchange.return_value = exchange
queue_consumer = Mock(spec=QueueConsumer)
queue_consumer.bind("queue_consumer", container)
consumer = RpcConsumer()
consumer.queue_consumer = queue_consumer
consumer.bind("rpc_consumer", container)
provider = RpcProvider()
provider.rpc_consumer = consumer
provider.bind("rpcmethod", container)
provider.prepare()
consumer.prepare()
queue_consumer.prepare()
queue = consumer.queue
assert queue.name == "rpc-exampleservice"
assert queue.routing_key == "exampleservice.*"
assert queue.exchange == exchange
assert queue.durable
queue_consumer.register_provider.assert_called_once_with(consumer)
consumer.register_provider(provider)
assert consumer._providers == set([provider])
routing_key = "exampleservice.rpcmethod"
assert consumer.get_provider_for_method(routing_key) == provider
routing_key = "exampleservice.invalidmethod"
with pytest.raises(MethodNotFound):
consumer.get_provider_for_method(routing_key)
consumer.unregister_provider(provider)
assert consumer._providers == set()
def test_rpc_consumer(get_rpc_exchange, queue_consumer, mock_container):
container = mock_container
container.shared_extensions = {}
container.config = {}
container.service_name = "exampleservice"
container.service_cls = Mock(rpcmethod=lambda: None)
exchange = Mock()
get_rpc_exchange.return_value = exchange
consumer = RpcConsumer().bind(container)
entrypoint = Rpc().bind(container, "rpcmethod")
entrypoint.rpc_consumer = consumer
entrypoint.setup()
consumer.setup()
queue_consumer.setup()
queue = consumer.queue
assert queue.name == "rpc-exampleservice"
assert queue.routing_key == "exampleservice.*"
assert queue.exchange == exchange
assert queue.durable
queue_consumer.register_provider.assert_called_once_with(consumer)
consumer.register_provider(entrypoint)
assert consumer._providers == set([entrypoint])
routing_key = "exampleservice.rpcmethod"
assert consumer.get_provider_for_method(routing_key) == entrypoint
routing_key = "exampleservice.invalidmethod"
with pytest.raises(MethodNotFound):
consumer.get_provider_for_method(routing_key)
consumer.unregister_provider(entrypoint)
assert consumer._providers == set()
def test_rpc_consumer(get_rpc_exchange, queue_consumer, mock_container):
container = mock_container
container.shared_extensions = {}
container.config = {}
container.service_name = "exampleservice"
container.service_cls = Mock(rpcmethod=lambda: None)
exchange = Mock()
get_rpc_exchange.return_value = exchange
consumer = RpcConsumer().bind(container)
entrypoint = Rpc().bind(container, "rpcmethod")
entrypoint.rpc_consumer = consumer
entrypoint.setup()
consumer.setup()
queue_consumer.setup()
queue = consumer.queue
assert queue.name == "rpc-exampleservice"
assert queue.routing_key == "exampleservice.*"
assert queue.exchange == exchange
assert queue.durable
queue_consumer.register_provider.assert_called_once_with(consumer)
consumer.register_provider(entrypoint)
assert consumer._providers == set([entrypoint])
routing_key = "exampleservice.rpcmethod"
assert consumer.get_provider_for_method(routing_key) == entrypoint
routing_key = "exampleservice.invalidmethod"
with pytest.raises(MethodNotFound):
consumer.get_provider_for_method(routing_key)
consumer.unregister_provider(entrypoint)
assert consumer._providers == set()