From 5f0c5f284ec1f7080f6017d444bc56df18a6116f Mon Sep 17 00:00:00 2001 From: Jan Fajerski Date: Thu, 28 May 2020 13:47:50 +0200 Subject: [PATCH] pybing/mgr/snap_schedule: add more unit tests Signed-off-by: Jan Fajerski --- .../snap_schedule/tests/fs/test_schedule.py | 67 +++++++++++++++++++ 1 file changed, 67 insertions(+) diff --git a/src/pybind/mgr/snap_schedule/tests/fs/test_schedule.py b/src/pybind/mgr/snap_schedule/tests/fs/test_schedule.py index a6239798f3cc6..4eaa908dd6135 100644 --- a/src/pybind/mgr/snap_schedule/tests/fs/test_schedule.py +++ b/src/pybind/mgr/snap_schedule/tests/fs/test_schedule.py @@ -50,3 +50,70 @@ class TestSchedule(object): rows = db.execute(SELECT_ALL).fetchall() assert len(rows) == len(simple_schedules) + + def test_update_last(self, db, simple_schedule): + simple_schedule.store_schedule(db) + + row = () + with db: + row = db.execute(SELECT_ALL).fetchone() + assert row['last'] is None + + first_time = datetime.datetime.now(datetime.timezone.utc) + simple_schedule.update_last(first_time, db) + + with db: + row = db.execute(SELECT_ALL).fetchone() + assert datetime.datetime.fromisoformat(row['last']) == first_time + + second_time = datetime.datetime.now(datetime.timezone.utc) + simple_schedule.update_last(second_time, db) + + with db: + row = db.execute(SELECT_ALL).fetchone() + assert datetime.datetime.fromisoformat(row['last']) == second_time + + def test_update_created_count(self, db, simple_schedule): + simple_schedule.store_schedule(db) + + row = () + with db: + row = db.execute(SELECT_ALL).fetchone() + assert row['created_count'] == 0 + + first_time = datetime.datetime.now(datetime.timezone.utc) + simple_schedule.update_last(first_time, db) + + with db: + row = db.execute(SELECT_ALL).fetchone() + assert row['created_count'] == 1 + + second_time = datetime.datetime.now(datetime.timezone.utc) + simple_schedule.update_last(second_time, db) + + with db: + row = db.execute(SELECT_ALL).fetchone() + assert row['created_count'] == 2 + + def test_update_first(self, db, simple_schedule): + simple_schedule.store_schedule(db) + + row = () + with db: + row = db.execute(SELECT_ALL).fetchone() + assert row['first'] is None + + first_time = datetime.datetime.now(datetime.timezone.utc) + simple_schedule.update_last(first_time, db) + + with db: + row = db.execute(SELECT_ALL).fetchone() + assert datetime.datetime.fromisoformat(row['first']) == first_time + + second_time = datetime.datetime.now(datetime.timezone.utc) + simple_schedule.update_last(second_time, db) + + with db: + row = db.execute(SELECT_ALL).fetchone() + assert datetime.datetime.fromisoformat(row['first']) == first_time + -- 2.39.5