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_equal(dummy_pool): """Test the __eq__ method""" assert dummy_pool == TransportPool( transport_class=DebugEventTransport, transport_config=DebugEventTransport.Config(), config=Config.default(), )
def dummy_bus(loop, redis_server_url): # fmt: off dummy_bus = lightbus.creation.create( config=RootConfig( apis={ 'default': ApiConfig( rpc_transport=RpcTransportSelector( debug=DebugRpcTransport.Config()), result_transport=ResultTransportSelector( debug=DebugResultTransport.Config()), event_transport=EventTransportSelector( debug=DebugEventTransport.Config()), ) }, bus=BusConfig( schema=SchemaConfig(transport=SchemaTransportSelector( redis=RedisSchemaTransport.Config( url=redis_server_url)), ))), plugins=[], ) # fmt: on yield dummy_bus try: dummy_bus.client.close() except BusAlreadyClosed: pass
async def dummy_pool(): pool = TransportPool( transport_class=DebugEventTransport, transport_config=DebugEventTransport.Config(), config=Config.default(), ) yield pool await pool.close()
def test_hash_equal(dummy_pool): """Test the __hash__ method""" assert hash(dummy_pool) == hash( TransportPool( transport_class=DebugEventTransport, transport_config=DebugEventTransport.Config(), config=Config.default(), ))
def test_get_all_transports(redis_default_config): registry = TransportRegistry().load_config(redis_default_config) # Is equal to the default transport, so will not be included # in the get_all_transports() return value registry.set_event_transport("another", RedisEventTransport, RedisEventTransport.Config(), Config.default()) registry.set_event_transport("foo", DebugEventTransport, DebugEventTransport.Config(), Config.default()) # redis rpc + redis result + redis event + foo debug transport (above) = 4 assert len(registry.get_all_transports()) == 4