diff 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
line wrap: on
line diff
--- a/test/test_templating.py	Fri Mar 12 01:30:22 2021 -0500
+++ b/test/test_templating.py	Fri Mar 12 01:49:38 2021 -0500
@@ -44,6 +44,9 @@
 
 class MockDatabase(MockNull):
     def getclass(self, name):
+        # limit class names
+        if name not in [ 'issue', 'user' ]:
+            raise KeyError('There is no class called "%s"' % name)
         # Class returned must have hasnode(id) method that returns true
         # otherwise designators like 'issue1' can't be hyperlinked.
         self.classes[name].hasnode = lambda id: True

Roundup Issue Tracker: http://roundup-tracker.org/