make timer test working

1 files changed, 23 insertions, 2 deletions

diff --git a/tests/fake_time.py b/tests/fake_time.py
index 09f88b49..3c77dedb 100644
--- a/tests/fake_time.py
+++ b/tests/fake_time.py
@@ -1,10 +1,31 @@
 import time
 
-real_time = time.time
+real_time = None
+time_offset = 0
 
-time_offset = real_time()
+def setRealtime():
+	global real_time
+	real_time = time.time
+
+def setIdealtime():
+	global real_time
+	real_time = lambda: 0
+
+def setTime(now):
+	global time_offset
+	time_offset = real_time() - now
+
+setIdealtime()
+setTime(0)
 
 def my_time():
 	return real_time() - time_offset
 
 time.time = my_time
+
+def my_sleep(sleep):
+	global time_offset
+	time_offset -= sleep
+	print "(faking %f seconds)" % sleep
+
+time.sleep = my_sleep