def test_size_of_cache(flowmachine_connect):
"""
Test that cache size is reported correctly.
"""
dl = daily_location("2016-01-01").store().result()
dl_aggregate = dl.aggregate().store().result()
total_cache_size = get_size_of_cache(flowmachine_connect)
removed_query, table_size_a = shrink_one(flowmachine_connect)
removed_query, table_size_b = shrink_one(flowmachine_connect)
assert total_cache_size == table_size_a + table_size_b
assert 0 == get_size_of_cache(flowmachine_connect)
def test_shrink_one(flowmachine_connect):
"""
Test that shrink_one removes a cache record.
"""
dl = daily_location("2016-01-01").store().result()
dl_aggregate = dl.aggregate().store().result()
flowmachine_connect.engine.execute(
f"UPDATE cache.cached SET cache_score_multiplier = 100 WHERE query_id='{dl_aggregate.md5}'"
)
flowmachine_connect.engine.execute(
f"UPDATE cache.cached SET cache_score_multiplier = 0.5 WHERE query_id='{dl.md5}'"
)
removed_query, table_size = shrink_one(flowmachine_connect)
assert dl.md5 == removed_query.md5
assert not dl.is_stored
assert dl_aggregate.is_stored
def test_shrink_one(flowmachine_connect):
"""
Test that shrink_one removes a cache record.
"""
dl = daily_location("2016-01-01").store().result()
dl_aggregate = dl.aggregate().store().result()
get_db().engine.execute(
f"UPDATE cache.cached SET cache_score_multiplier = 1 WHERE query_id='{dl_aggregate.query_id}'"
)
get_db().engine.execute(
f"UPDATE cache.cached SET cache_score_multiplier = 0.5 WHERE query_id='{dl.query_id}'"
)
removed_query, table_size = shrink_one(get_db(), protected_period=-1)
assert dl.query_id == removed_query.query_id
assert not dl.is_stored
assert dl_aggregate.is_stored
