diff --git a/src/wormhole_transit_relay/test/test_stats.py b/src/wormhole_transit_relay/test/test_stats.py index ae37790..f9433ef 100644 --- a/src/wormhole_transit_relay/test/test_stats.py +++ b/src/wormhole_transit_relay/test/test_stats.py @@ -12,14 +12,15 @@ class DB(unittest.TestCase): return db def test_db(self): + T = 1519075308.0 d = self.mktemp() os.mkdir(d) usage_db = os.path.join(d, "usage.sqlite") - with mock.patch("time.time", return_value=456.0): + with mock.patch("time.time", return_value=T+0): t = Transit(blur_usage=None, log_file=None, usage_db=usage_db) db = self.open_db(usage_db) - with mock.patch("time.time", return_value=457.0): + with mock.patch("time.time", return_value=T+1): t.recordUsage(started=123, result="happy", total_bytes=100, total_time=10, waiting_time=2) self.assertEqual(db.execute("SELECT * FROM `usage`").fetchall(), @@ -27,11 +28,11 @@ class DB(unittest.TestCase): total_bytes=100, total_time=10, waiting_time=2), ]) self.assertEqual(db.execute("SELECT * FROM `current`").fetchone(), - dict(rebooted=456, updated=457, + dict(rebooted=T+0, updated=T+1, incomplete_bytes=0, waiting=0, connected=0)) - with mock.patch("time.time", return_value=458.0): + with mock.patch("time.time", return_value=T+2): t.recordUsage(started=150, result="errory", total_bytes=200, total_time=11, waiting_time=3) self.assertEqual(db.execute("SELECT * FROM `usage`").fetchall(), @@ -41,14 +42,14 @@ class DB(unittest.TestCase): total_bytes=200, total_time=11, waiting_time=3), ]) self.assertEqual(db.execute("SELECT * FROM `current`").fetchone(), - dict(rebooted=456, updated=458, + dict(rebooted=T+0, updated=T+2, incomplete_bytes=0, waiting=0, connected=0)) - with mock.patch("time.time", return_value=459.0): + with mock.patch("time.time", return_value=T+3): t.timerUpdateStats() self.assertEqual(db.execute("SELECT * FROM `current`").fetchone(), - dict(rebooted=456, updated=459, + dict(rebooted=T+0, updated=T+3, incomplete_bytes=0, waiting=0, connected=0))