def test_days_to_update_unstarted_two_days(conn):
journey_id = insert_journey_data(conn)
date = pendulum.Date(2013, 3, 31)
journey.queries.start_journey(conn, journey_id=journey_id, date=date)
next_date = date.add(days=2)
itr = journey.days_to_update(conn, journey_id, next_date)
days = list(itr)
assert days == [date, date.add(days=1)]
def test_days_to_update(conn: connection):
steps_data, body_reports = example_activity_data()
g_info = example_gargling_info()
journey_id = insert_journey_data(conn)
start_date = pendulum.Date(2013, 3, 31)
journey.queries.start_journey(conn, journey_id=journey_id, date=start_date)
with api_mocker():
data = journey.perform_daily_update(conn, journey_id, start_date,
steps_data, g_info)
assert data is not None
location, *_, finished = data
journey.store_update_data(conn, location, finished)
cur_date = start_date.add(days=2)
itr = journey.days_to_update(conn, journey_id, cur_date)
days = list(itr)
assert days == [cur_date.subtract(days=1)]
def test_generate_all_maps(mock_activity, conn):
steps_data, body_reports = example_activity_data()
mock_activity.return_value = (steps_data, body_reports)
journey_id = insert_journey_data(conn)
start_date = pendulum.Date(2013, 3, 31)
journey.queries.start_journey(conn, journey_id=journey_id, date=start_date)
g_info = example_gargling_info()
with api_mocker():
journey.perform_daily_update(conn, journey_id, start_date, steps_data,
g_info)
cur_date = start_date.add(days=4)
for date in journey.days_to_update(conn, journey_id, cur_date):
steps_data, body_reports = example_activity_data()
datum = journey.perform_daily_update(conn, journey_id, date,
steps_data, g_info)
if datum is None:
continue
location, *_, finished = datum
journey.store_update_data(conn, location, finished)
journey.store_steps(conn, steps_data, journey_id, date)
with patch("gargbot_3000.journey.mapping.render_map") as maps:
maps.return_value = Image.new("RGB", (1000, 600), (255, 255, 255))
mapping.generate_all_maps(conn, journey_id, write=False)
def test_journey_finished(mock_activity, conn: connection):
steps_data, body_reports = example_activity_data()
mock_activity.return_value = (steps_data, body_reports)
journey_id = insert_journey_data(conn)
start_date = pendulum.Date(2013, 3, 31)
journey.queries.start_journey(conn, journey_id=journey_id, date=start_date)
g_info = example_gargling_info()
with api_mocker():
steps_data, body_reports = example_activity_data()
journey.perform_daily_update(conn, journey_id, start_date, steps_data,
g_info)
cur_date = start_date.add(days=4)
data = []
for date in journey.days_to_update(conn, journey_id, cur_date):
steps_data, body_reports = example_activity_data()
datum = journey.perform_daily_update(conn, journey_id, date,
steps_data, g_info)
if datum is None:
continue
location, *_, finished = datum
journey.store_update_data(conn, location, finished)
data.append(finished)
last_fin = data[-1]
assert last_fin is True