Mercurial > p > roundup > code
comparison test/test_templating.py @ 6817:38b73f7495f8
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.
| author | John Rouillard <rouilj@ieee.org> |
|---|---|
| date | Thu, 04 Aug 2022 16:11:59 -0400 |
| parents | 994893cf3e1a |
| children | 28b906a237d8 |
comparison
equal
deleted
inserted
replaced
| 6816:a4652d809ec5 | 6817:38b73f7495f8 |
|---|---|
| 54 | 54 |
| 55 # setup for csrf testing of otks database api | 55 # setup for csrf testing of otks database api |
| 56 storage = {} | 56 storage = {} |
| 57 def set(self, key, **props): | 57 def set(self, key, **props): |
| 58 MockDatabase.storage[key] = {} | 58 MockDatabase.storage[key] = {} |
| 59 if '__timestamp' not in props: | |
| 60 props['__timestamp'] = time.time() - 7*24*3600 | |
| 59 MockDatabase.storage[key].update(props) | 61 MockDatabase.storage[key].update(props) |
| 60 | 62 |
| 61 def get(self, key, field, default=None): | 63 def get(self, key, field, default=None): |
| 62 if key not in MockDatabase.storage: | 64 if key not in MockDatabase.storage: |
| 63 return default | 65 return default |
| 64 return MockDatabase.storage[key][field] | 66 return MockDatabase.storage[key][field] |
| 65 | 67 |
| 68 def getall(self, key): | |
| 69 if key not in MockDatabase.storage: | |
| 70 return default | |
| 71 return MockDatabase.storage[key] | |
| 72 | |
| 66 def exists(self,key): | 73 def exists(self,key): |
| 67 return key in MockDatabase.storage | 74 return key in MockDatabase.storage |
| 68 | 75 |
| 69 def getOTKManager(self): | 76 def getOTKManager(self): |
| 70 return MockDatabase() | 77 return MockDatabase() |
| 78 | |
| 79 def lifetime(self, seconds): | |
| 80 return time.time() - 7*24*3600 + seconds | |
| 71 | 81 |
| 72 class TemplatingTestCase(unittest.TestCase): | 82 class TemplatingTestCase(unittest.TestCase): |
| 73 def setUp(self): | 83 def setUp(self): |
| 74 self.form = FieldStorage() | 84 self.form = FieldStorage() |
| 75 self.client = MockNull() | 85 self.client = MockNull() |