def test_invalidate_cache_midchain(flowmachine_connect):
"""
Test that invalidating a query in the middle of a chain drops the
top of the chain and this link, but not the bottom.
"""
dl1 = daily_location("2016-01-01")
dl1.store().result()
hl1 = ModalLocation(daily_location("2016-01-01"),
daily_location("2016-01-02"))
hl1.store().result()
hl2 = ModalLocation(daily_location("2016-01-03"),
daily_location("2016-01-04"))
flow = Flows(hl1, hl2)
flow.store().result()
assert dl1.is_stored
assert hl1.is_stored
assert flow.is_stored
hl1.invalidate_db_cache()
assert dl1.is_stored
assert not hl1.is_stored
assert not flow.is_stored
assert cache_table_exists(get_db(), dl1.query_id)
assert not cache_table_exists(get_db(), hl1.query_id)
assert not cache_table_exists(get_db(), flow.query_id)
has_deps = bool(get_db().fetch("SELECT * FROM cache.dependencies"))
assert has_deps # Daily location deps should remain
def test_invalidate_cache_midchain(flowmachine_connect):
"""
Test that invalidating a query in the middle of a chain drops the
top of the chain and this link, but not the bottom.
"""
dl1 = daily_location("2016-01-01")
dl1.store().result()
hl1 = ModalLocation(daily_location("2016-01-01"),
daily_location("2016-01-02"))
hl1.store().result()
hl2 = ModalLocation(daily_location("2016-01-03"),
daily_location("2016-01-04"))
flow = Flows(hl1, hl2)
flow.store().result()
hl1.invalidate_db_cache()
assert dl1.is_stored
assert not hl1.is_stored
assert not flow.is_stored
in_cache = bool(
flowmachine_connect.fetch(
f"SELECT * FROM cache.cached WHERE query_id='{dl1.md5}'"))
assert in_cache
in_cache = bool(
flowmachine_connect.fetch(
f"SELECT * FROM cache.cached WHERE query_id='{hl1.md5}'"))
assert not in_cache
in_cache = bool(
flowmachine_connect.fetch(
f"SELECT * FROM cache.cached WHERE query_id='{flow.md5}'"))
assert not in_cache
has_deps = bool(
flowmachine_connect.fetch("SELECT * FROM cache.dependencies"))
assert has_deps # Daily location deps should remain
