move selection of entity types to test in a method to ease overriding 3.0
authorsylvain.thenault@logilab.fr
Thu, 15 Jan 2009 13:25:34 +0100
branch3.0
changeset 427 e894eec21a1b
parent 426 e96662444ec6
child 428 7d80331a91d7
move selection of entity types to test in a method to ease overriding
devtools/testlib.py
--- a/devtools/testlib.py	Thu Jan 15 11:25:29 2009 +0100
+++ b/devtools/testlib.py	Thu Jan 15 13:25:34 2009 +0100
@@ -259,10 +259,13 @@
                     msg+= '\nfor output:\n%s' % output
             raise AssertionError, msg, tcbk
 
-        
+
+    def to_test_etypes(self):
+        return unprotected_entities(self.schema, strict=True)
+    
     def iter_automatic_rsets(self):
         """generates basic resultsets for each entity type"""
-        etypes = unprotected_entities(self.schema, strict=True)
+        etypes = self.to_test_etypes()
         for etype in etypes:
             yield self.execute('Any X WHERE X is %s' % etype)