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)