Skip to content

Commit 389ec73

Browse files
committed
Make testing mock smarter. Make getclass fail for invalid classes.
1 parent d6c12b8 commit 389ec73

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

test/test_templating.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,9 @@
4444

4545
class MockDatabase(MockNull):
4646
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)
4750
# Class returned must have hasnode(id) method that returns true
4851
# otherwise designators like 'issue1' can't be hyperlinked.
4952
self.classes[name].hasnode = lambda id: True

0 commit comments

Comments
 (0)