Mercurial > p > roundup > code
comparison test/test_templating.py @ 6337:316c2c32dace
Make testing mock smarter. Make getclass fail for invalid classes.
| author | John Rouillard <rouilj@ieee.org> |
|---|---|
| date | Fri, 12 Mar 2021 01:49:38 -0500 |
| parents | 6f89cdc7c938 |
| children | a70a0c138dd9 |
comparison
equal
deleted
inserted
replaced
| 6336:6f89cdc7c938 | 6337:316c2c32dace |
|---|---|
| 42 | 42 |
| 43 from roundup.anypy.strings import u2s, s2u | 43 from roundup.anypy.strings import u2s, s2u |
| 44 | 44 |
| 45 class MockDatabase(MockNull): | 45 class MockDatabase(MockNull): |
| 46 def getclass(self, name): | 46 def getclass(self, name): |
| 47 # limit class names | |
| 48 if name not in [ 'issue', 'user' ]: | |
| 49 raise KeyError('There is no class called "%s"' % name) | |
| 47 # Class returned must have hasnode(id) method that returns true | 50 # Class returned must have hasnode(id) method that returns true |
| 48 # otherwise designators like 'issue1' can't be hyperlinked. | 51 # otherwise designators like 'issue1' can't be hyperlinked. |
| 49 self.classes[name].hasnode = lambda id: True | 52 self.classes[name].hasnode = lambda id: True |
| 50 return self.classes[name] | 53 return self.classes[name] |
| 51 | 54 |