async def fixture_rest_async(
session_tmpdir: TempDirCreator, kafka_server: Optional[KafkaConfig], registry_async_client: Client
) -> AsyncIterator[KafkaRest]:
if not kafka_server:
assert REST_URI in os.environ
instance, _ = mock_factory("rest")()
yield instance
else:
config_path = os.path.join(session_tmpdir(), "karapace_config.json")
kafka_port = kafka_server.kafka_port
config = set_config_defaults({
"log_level": "WARNING",
"bootstrap_uri": f"127.0.0.1:{kafka_port}",
"admin_metadata_max_age": 0
})
write_config(config_path, config)
rest = KafkaRest(config_file_path=config_path, config=config)
assert rest.serializer.registry_client
assert rest.consumer_manager.deserializer.registry_client
rest.serializer.registry_client.client = registry_async_client
rest.consumer_manager.deserializer.registry_client.client = registry_async_client
try:
yield rest
finally:
rest.close()
await rest.close_producers()
async def fixture_registry_async(
tmp_path: Path, kafka_server: Optional[KafkaConfig]
) -> AsyncIterator[KarapaceSchemaRegistry]:
if not kafka_server:
assert REGISTRY_URI in os.environ or REST_URI in os.environ
instance, _ = mock_factory("registry")()
yield instance
else:
config_path = tmp_path / "karapace_config.json"
kafka_port = kafka_server.kafka_port
config = set_config_defaults({
"log_level":
"WARNING",
"bootstrap_uri":
f"127.0.0.1:{kafka_port}",
"topic_name":
new_random_name(),
"group_id":
new_random_name("schema_registry")
})
write_config(config_path, config)
registry = KarapaceSchemaRegistry(config_file_path=str(config_path),
config=set_config_defaults(config))
await registry.get_master()
try:
yield registry
finally:
registry.close()
async def fixture_registry_async(session_tmpdir, kafka_server):
if REGISTRY_URI in os.environ or REST_URI in os.environ:
instance, _ = mock_factory("registry")()
yield instance
else:
config_path = os.path.join(session_tmpdir(), "karapace_config.json")
kafka_port = kafka_server["kafka_port"]
write_config(
config_path, {
"log_level": "WARNING",
"bootstrap_uri": f"127.0.0.1:{kafka_port}",
"topic_name": new_random_name(),
"group_id": new_random_name("schema_registry")
})
registry = KarapaceSchemaRegistry(config_path)
await registry.get_master()
try:
yield registry
finally:
registry.close()
async def fixture_rest_async(session_tmpdir, kafka_server,
registry_async_client):
if REST_URI in os.environ:
instance, _ = mock_factory("rest")()
yield instance
else:
config_path = os.path.join(session_tmpdir(), "karapace_config.json")
kafka_port = kafka_server["kafka_port"]
write_config(
config_path, {
"log_level": "WARNING",
"bootstrap_uri": f"127.0.0.1:{kafka_port}",
"admin_metadata_max_age": 0
})
rest = KafkaRest(config_path)
rest.serializer.registry_client.client = registry_async_client
rest.consumer_manager.deserializer.registry_client.client = registry_async_client
try:
yield rest
finally:
rest.close()
await rest.close_producers()