def test_transport_registry_get_event_transports(redis_default_config):
registry = TransportRegistry().load_config(redis_default_config)
# See comment in test_transport_registry_get_rpc_transports
registry.set_event_transport("redis1", RedisEventTransport,
RedisEventTransport.Config(batch_size=99),
Config.default())
registry.set_event_transport("redis2", RedisEventTransport,
RedisEventTransport.Config(batch_size=99),
Config.default())
registry.set_event_transport("debug1", DebugEventTransport,
DebugEventTransport.Config(),
Config.default())
registry.set_event_transport("debug2", DebugEventTransport,
DebugEventTransport.Config(),
Config.default())
transports = registry.get_event_transports(
["default", "foo", "bar", "redis1", "redis2", "debug1", "debug2"])
default_transport = registry.get_event_transport("default")
redis_transport = registry.get_event_transport("redis1")
debug_transport = registry.get_event_transport("debug1")
assert set(transports[default_transport]) == {"default", "foo", "bar"}
assert set(transports[debug_transport]) == {"debug1", "debug2"}
assert set(transports[redis_transport]) == {"redis1", "redis2"}
def test_transport_registry_get_does_not_exist_default():
registry = TransportRegistry()
with pytest.raises(TransportNotFound):
assert not registry.get_rpc_transport("default")
with pytest.raises(TransportNotFound):
assert not registry.get_result_transport("default")
with pytest.raises(TransportNotFound):
assert not registry.get_event_transport("default")
with pytest.raises(TransportNotFound):
assert not registry.get_schema_transport()
def test_transport_registry_load_config(redis_default_config):
registry = TransportRegistry().load_config(redis_default_config)
assert registry.get_rpc_transport(
"default").transport_class == RedisRpcTransport
assert registry.get_result_transport(
"default").transport_class == RedisResultTransport
assert registry.get_event_transport(
"default").transport_class == RedisEventTransport
assert registry.get_schema_transport(
"default").transport_class == RedisSchemaTransport
def test_transport_registry_get_specific_api(redis_other_config):
registry = TransportRegistry().load_config(redis_other_config)
assert registry.get_rpc_transport(
"other").transport_class == RedisRpcTransport
assert registry.get_result_transport(
"other").transport_class == RedisResultTransport
assert registry.get_event_transport(
"other").transport_class == RedisEventTransport
assert registry.get_schema_transport(
"other").transport_class == RedisSchemaTransport
def log_welcome_message(
logger: Logger,
transport_registry: TransportRegistry,
schema: "Schema",
plugin_registry: PluginRegistry,
config: Config,
):
"""Show the worker-startup welcome message
"""
logger.info(
LBullets(
"Lightbus is setting up",
items={
"service_name (set with -s or LIGHTBUS_SERVICE_NAME)":
Bold(config.service_name),
"process_name (with with -p or LIGHTBUS_PROCESS_NAME)":
Bold(config.process_name),
},
))
# Log the transport information
rpc_transport = transport_registry.get_rpc_transport("default",
default=None)
result_transport = transport_registry.get_result_transport("default",
default=None)
event_transport = transport_registry.get_event_transport("default",
default=None)
log_transport_information(rpc_transport, result_transport, event_transport,
schema.schema_transport, logger)
if plugin_registry._plugins:
logger.info(
LBullets(
"Loaded the following plugins ({})".format(
len(plugin_registry._plugins)),
items=plugin_registry._plugins,
))
else:
logger.info("No plugins loaded")
def test_transport_registry_get_does_not_exist_default_default_value():
registry = TransportRegistry()
assert registry.get_rpc_transport("default", default=None) is None
assert registry.get_result_transport("default", default=None) is None
assert registry.get_event_transport("default", default=None) is None
assert registry.get_schema_transport(default=None) is None
