def test_json_decoded(temp_file: str, mocked_redis: MockedRedis,
query: RedisQuery):
mocked_redis.add({
'product:a': '{ "name": "kaas" }',
'product:b': '{ "name": "kees" }',
'product:c': '{ "name": "koos" }',
'something-else': '{ "name": "kies" }',
})
with JsonProcessor(temp_file, True) as p:
query.query_with_processor("product:", p)
with open(temp_file, "r") as f:
obj = json.load(f)
assert obj == [{
"key": "product:a",
"value": {
"name": "kaas"
}
}, {
"key": "product:b",
"value": {
"name": "kees"
}
}, {
"key": "product:c",
"value": {
"name": "koos"
}
}]
def test_csv_omit_header(temp_file: str, mocked_redis: MockedRedis,
query: RedisQuery):
mocked_redis.add({
'product:a': '11',
'product:b': '22',
'product:c': '33',
'something-else': '44'
})
with CsvProcessor(temp_file, omit_csv_header=True) as p:
query.query_with_processor("product:", p)
with open(temp_file, "r") as f:
lines = f.read().splitlines()
assert lines == ['product:a,11', 'product:b,22', 'product:c,33']
def test_txt(temp_file: str, mocked_redis: MockedRedis, query: RedisQuery):
mocked_redis.add({
'product:a': '11',
'product:b': '22',
'product:c': '33',
'something-else': '44'
})
with TextProcessor(temp_file) as p:
query.query_with_processor("product:", p)
with open(temp_file, "r") as f:
lines = f.read().splitlines()
assert lines == ['product:a', '11', 'product:b', '22', 'product:c', '33']