def retrieve_all_records(client: KvDbClient) -> List[AirbyteRecordMessage]:
"""retrieves and formats all records in kvdb as Airbyte messages"""
all_records = client.list_keys(list_values=True)
out = []
for record in all_records:
key = record[0]
stream = key.split("__ab__")[0]
value = record[1]
out.append(_record(stream, value["str_col"], value["int_col"]))
return out
def teardown(config: Mapping):
yield
client = KvDbClient(**config)
client.delete(list(client.list_keys()))