Skip to content

Commit 310cfe1

Browse files
committed
Implement UpdateTimestamp for memorydb and test.
Also test proper DbType when running memory test.
1 parent a8d7b2e commit 310cfe1

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

roundup/test/memorydb.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,10 @@ def commit(self):
194194
def close(self):
195195
pass
196196
def updateTimestamp(self, sessid):
197-
pass
197+
sess = self.get(sessid, '__timestamp', None)
198+
now = time.time()
199+
if sess is None or now > sess + 60:
200+
self.set(sessid, __timestamp=now)
198201
def clean(self):
199202
pass
200203

test/test_memorydb.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,11 +62,16 @@ def setUp(self):
6262
self.db = self.module.Database(config, 'admin')
6363
setupSchema(self.db, 1, self.module)
6464
self.sessions = self.db.sessions
65+
self.db.Session = self.sessions
6566
self.otks = self.db.otks
67+
self.db.Otk = self.otks
6668

67-
# doesn't work for memory as it uses a mock for session db.
68-
def testUpdateTimestamp(self):
69-
self.skipTest("This test is not implemented for memorydb.")
69+
def get_ts(self):
70+
return (self.sessions.get('random_session', '__timestamp'),)
71+
72+
def testDbType(self):
73+
self.assertIn("memorydb", repr(self.db))
74+
self.assertIn("{}", repr(self.db.Session))
7075

7176
# vim: set filetype=python ts=4 sw=4 et si
7277

0 commit comments

Comments
 (0)