def test_store_with_table():
"""
Test that a subset of a table can be stored.
"""
t = Table("events.calls")
s = t.subset("id", ["5wNJA-PdRJ4-jxEdG-yOXpZ", "5wNJA-PdRJ4-jxEdG-yOXpZ"])
s.store().result()
assert s.is_stored
t.invalidate_db_cache()
assert not s.is_stored
assert t.is_stored
def subscriber_list_table(subscriber_list, flowmachine_connect):
engine = flowmachine_connect.engine
with engine.begin():
sql = """CREATE TABLE subscriber_list (subscriber TEXT)"""
engine.execute(sql)
formatted_subscribers = ",".join("('{}')".format(u)
for u in subscriber_list)
sql = """INSERT INTO subscriber_list (subscriber) VALUES {}""".format(
formatted_subscribers)
engine.execute(sql)
subs_table = Table("subscriber_list")
yield subs_table
subs_table.invalidate_db_cache(drop=True)