Skip to content

Commit 2c634de

Browse files
committed
fix mock datbase
add lifetime() method. caused tests to break. Use time.mie() - week + lifetime. set() method sets __timestamp if not set. add getall useful for debugging.
1 parent f05203c commit 2c634de

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

test/test_templating.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -56,19 +56,29 @@ def getclass(self, name):
5656
storage = {}
5757
def set(self, key, **props):
5858
MockDatabase.storage[key] = {}
59+
if '__timestamp' not in props:
60+
props['__timestamp'] = time.time() - 7*24*3600
5961
MockDatabase.storage[key].update(props)
6062

6163
def get(self, key, field, default=None):
6264
if key not in MockDatabase.storage:
6365
return default
6466
return MockDatabase.storage[key][field]
6567

68+
def getall(self, key):
69+
if key not in MockDatabase.storage:
70+
return default
71+
return MockDatabase.storage[key]
72+
6673
def exists(self,key):
6774
return key in MockDatabase.storage
6875

6976
def getOTKManager(self):
7077
return MockDatabase()
7178

79+
def lifetime(self, seconds):
80+
return time.time() - 7*24*3600 + seconds
81+
7282
class TemplatingTestCase(unittest.TestCase):
7383
def setUp(self):
7484
self.form = FieldStorage()

0 commit comments

Comments
 (0)