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