def test_consumer_passes_offsets_to_work_fn(db):
conn, _ = db
consumer = Consumer(conn, "foo", "offsets")
new_offset = consumer.new_offset()
consumer.commit(new_offset)
def work(old, new):
assert old == new_offset
consumer.do_work(work)
def test_consumer_throws_consumer_work_error_on_exception(db):
conn, _ = db
consumer = Consumer(conn, "foo", "offsets")
new_offset = consumer.new_offset()
consumer.commit(new_offset)
def work(old, new):
assert old != new_offset
with pytest.raises(ConsumerWorkError):
consumer.do_work(work)
def main():
env = Env()
conn = psycopg2.connect(
dbname=env("PG_DATABASE"),
user=env("PG_USER"),
host=env("PG_HOST"),
port=env("PG_PORT"),
password=env("PG_PASSWORD"),
)
conf_path = env("GOOGLE_ADS_YAML_PATH")
consumer = Consumer(conn, "signups_ga", "event_consumer_groups")
client = GoogleAdsClient.load_from_storage(conf_path)
customer_id = env("GOOGLE_CUSTOMER_ID")
work = lambda o, n: report_conversions(conn, client, customer_id, o, n)
consumer.do_work(work)
