def test_delete_query_by_id(flowmachine_connect):
"""
Test that we can remove a query from cache by the query's id
"""
dl = daily_location("2016-01-01").store().result()
retrieved_query = invalidate_cache_by_id(flowmachine_connect, dl.query_id)
assert dl.query_id == retrieved_query.query_id
assert not dl.is_stored
def test_delete_query_by_id_does_not_cascade_by_default(flowmachine_connect):
"""
Test that removing a query by id doesn't cascade by default
"""
dl = daily_location("2016-01-01").store().result()
dl_agg = dl.aggregate().store().result()
retrieved_query = invalidate_cache_by_id(flowmachine_connect, dl.query_id)
assert dl.query_id == retrieved_query.query_id
assert not dl.is_stored
assert dl_agg.is_stored
def test_delete_query_by_id_can_cascade(flowmachine_connect):
"""
Test that removing a query by id can cascade
"""
dl = daily_location("2016-01-01").store().result()
dl_agg = dl.aggregate().store().result()
retrieved_query = invalidate_cache_by_id(get_db(), dl.query_id, cascade=True)
assert dl.query_id == retrieved_query.query_id
assert not dl.is_stored
assert not dl_agg.is_stored
