X-Git-Url: http://code.vuplus.com/gitweb/?p=vuplus_dvbapp;a=blobdiff_plain;f=tests%2Ffake_time.py;h=3c77dedb0606a600af97eaa29ebcd6104241c055;hp=09f88b4977013a3503a163f58a6b597e02ee0814;hb=441ac3dbbc2c5dfd21b09b00f6d376a115ac32a5;hpb=283c17a74cf11f40b822807f2d9f52644136c749

diff --git a/tests/fake_time.py b/tests/fake_time.py
index 09f88b4..3c77ded 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