From fcb342cb2cbbaea02404e0fdd39154c1e5991c04 Mon Sep 17 00:00:00 2001 From: Valentin Lorentz Date: Sun, 15 Dec 2019 20:19:17 +0100 Subject: [PATCH] test: Mock calls to time.sleep() The tests now take 7 seconds instead of 30. --- test/test_ircdb.py | 4 ++-- test/test_schedule.py | 30 +++++++++++++++--------------- test/test_utils.py | 6 +++--- 3 files changed, 20 insertions(+), 20 deletions(-) diff --git a/test/test_ircdb.py b/test/test_ircdb.py index 365402201..a195fa860 100644 --- a/test/test_ircdb.py +++ b/test/test_ircdb.py @@ -248,7 +248,7 @@ class IrcUserTestCase(IrcdbTestCase): u = ircdb.IrcUser() u.addAuth('foo!bar@baz') self.failUnless(u.checkHostmask('foo!bar@baz')) - time.sleep(2.1) + timeFastForward(2.1) self.failIf(u.checkHostmask('foo!bar@baz')) finally: conf.supybot.databases.users.timeoutIdentification.setValue(orig) @@ -265,7 +265,7 @@ class IrcUserTestCase(IrcdbTestCase): self.failUnless(len(u.auth) == 1) u.addAuth('boo!far@fizz') self.failUnless(u.checkHostmask('boo!far@fizz')) - time.sleep(2.1) + timeFastForward(2.1) self.failIf(u.checkHostmask('foo!bar@baz')) self.failIf(u.checkHostmask('boo!far@fizz')) finally: diff --git a/test/test_schedule.py b/test/test_schedule.py index 8749269b1..ae3b71cf2 100644 --- a/test/test_schedule.py +++ b/test/test_schedule.py @@ -53,10 +53,10 @@ class TestSchedule(SupyTestCase): sched.addEvent(add10, time.time() + 3) sched.addEvent(add1, time.time() + 1) - time.sleep(1.2) + timeFastForward(1.2) sched.run() self.assertEqual(i[0], 1) - time.sleep(1.9) + timeFastForward(1.9) sched.run() self.assertEqual(i[0], 11) @@ -65,7 +65,7 @@ class TestSchedule(SupyTestCase): self.assertEqual(i[0], 11) sched.removeEvent('test') self.assertEqual(i[0], 11) - time.sleep(3) + timeFastForward(3) self.assertEqual(i[0], 11) def testReschedule(self): @@ -75,10 +75,10 @@ class TestSchedule(SupyTestCase): i[0] += 1 n = sched.addEvent(inc, time.time() + 1) sched.rescheduleEvent(n, time.time() + 3) - time.sleep(1.2) + timeFastForward(1.2) sched.run() self.assertEqual(i[0], 0) - time.sleep(2) + timeFastForward(2) sched.run() self.assertEqual(i[0], 1) @@ -88,20 +88,20 @@ class TestSchedule(SupyTestCase): def inc(): i[0] += 1 n = sched.addPeriodicEvent(inc, 1, name='test_periodic') - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 0.6 self.assertEqual(i[0], 1) - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 1.2 self.assertEqual(i[0], 2) - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 1.8 self.assertEqual(i[0], 2) - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 2.4 self.assertEqual(i[0], 3) sched.removePeriodicEvent(n) - time.sleep(1) + timeFastForward(1) sched.run() # 3.4 self.assertEqual(i[0], 3) @@ -111,19 +111,19 @@ class TestSchedule(SupyTestCase): def inc(): i[0] += 1 n = sched.addPeriodicEvent(inc, 1, name='test_periodic', count=3) - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 0.6 self.assertEqual(i[0], 1) - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 1.2 self.assertEqual(i[0], 2) - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 1.8 self.assertEqual(i[0], 2) - time.sleep(0.6) + timeFastForward(0.6) sched.run() # 2.4 self.assertEqual(i[0], 3) - time.sleep(1) + timeFastForward(1) sched.run() # 3.4 self.assertEqual(i[0], 3) diff --git a/test/test_utils.py b/test/test_utils.py index 86908163e..931109152 100644 --- a/test/test_utils.py +++ b/test/test_utils.py @@ -1123,7 +1123,7 @@ class TestTimeoutQueue(SupyTestCase): q.enqueue(3) self.assertEqual(len(q), 3) self.assertEqual(sum(q), 6) - time.sleep(1.1) + timeFastForward(1.1) self.assertEqual(len(q), 0) self.assertEqual(sum(q), 0) @@ -1136,7 +1136,7 @@ class TestTimeoutQueue(SupyTestCase): q.enqueue(3) self.assertEqual(len(q), 3) self.assertEqual(sum(q), 6) - time.sleep(1.1) + timeFastForward(1.1) self.assertEqual(len(q), 0) self.assertEqual(sum(q), 0) @@ -1146,7 +1146,7 @@ class TestTimeoutQueue(SupyTestCase): self.failUnless(1 in q) self.failUnless(1 in q) # For some reason, the second one might fail. self.failIf(2 in q) - time.sleep(1.1) + timeFastForward(1.1) self.failIf(1 in q) def testReset(self):