diff options
| -rw-r--r-- | Lib/test/test_pyclbr.py | 158 |
1 files changed, 158 insertions, 0 deletions
diff --git a/Lib/test/test_pyclbr.py b/Lib/test/test_pyclbr.py new file mode 100644 index 0000000..476f1c7 --- /dev/null +++ b/Lib/test/test_pyclbr.py @@ -0,0 +1,158 @@ +''' + Test cases for pyclbr.py + Nick Mathewson +''' +from test_support import run_unittest +import unittest, sys +from types import ClassType, FunctionType, MethodType +import pyclbr + +# This next line triggers an error on old versions of pyclbr. + +from commands import getstatus + +# Here we test the python class browser code. +# +# The main function in this suite, 'testModule', compares the output +# of pyclbr with the introspected members of a module. Because pyclbr +# is imperfect (as designed), testModule is called with a set of +# members to ignore. + +class PyclbrTest(unittest.TestCase): + + def assertListEq(self, l1, l2, ignore): + ''' succeed iff {l1} - {ignore} == {l2} - {ignore} ''' + for p1, p2 in (l1, l2), (l2, l1): + for item in p1: + ok = (item in p2) or (item in ignore) + if not ok: + self.fail("%r missing" % item) + + + def assertHasattr(self, obj, attr, ignore): + ''' succeed iff hasattr(obj,attr) or attr in ignore. ''' + if attr in ignore: return + if not hasattr(obj, attr): print "???",attr + self.failUnless(hasattr(obj, attr)) + + + def assertHaskey(self, obj, key, ignore): + ''' succeed iff obj.has_key(key) or key in ignore. ''' + if key in ignore: return + if not obj.has_key(key): print "***",key + self.failUnless(obj.has_key(key)) + + def assertEquals(self, a, b, ignore=None): + ''' succeed iff a == b or a in ignore or b in ignore ''' + if (ignore == None) or (a in ignore) or (b in ignore): return + + unittest.TestCase.assertEquals(self, a, b) + + def checkModule(self, moduleName, module=None, ignore=()): + ''' succeed iff pyclbr.readmodule_ex(modulename) corresponds + to the actual module object, module. Any identifiers in + ignore are ignored. If no module is provided, the appropriate + module is loaded with __import__.''' + + if module == None: + module = __import__(moduleName, globals(), {}, []) + + dict = pyclbr.readmodule_ex(moduleName) + + # Make sure the toplevel functions and classes are the same. + for name, value in dict.items(): + if name in ignore: + continue + self.assertHasattr(module, name, ignore) + py_item = getattr(module, name) + if isinstance(value, pyclbr.Function): + self.assertEquals(type(py_item), FunctionType) + else: + self.assertEquals(type(py_item), ClassType) + real_bases = [base.__name__ for base in py_item.__bases__] + pyclbr_bases = [ getattr(base, 'name', base) + for base in value.super ] + + self.assertListEq(real_bases, pyclbr_bases, ignore) + + actualMethods = [] + for m in dir(py_item): + if type(getattr(py_item, m)) == MethodType: + actualMethods.append(m) + foundMethods = [] + for m in value.methods.keys(): + if m[:2] == '__' and m[-2:] != '__': + foundMethods.append('_'+name+m) + else: + foundMethods.append(m) + + self.assertListEq(foundMethods, actualMethods, ignore) + self.assertEquals(py_item.__module__, value.module) + + self.assertEquals(py_item.__name__, value.name, ignore) + # can't check file or lineno + + # Now check for missing stuff. + for name in dir(module): + item = getattr(module, name) + if type(item) in (ClassType, FunctionType): + self.assertHaskey(dict, name, ignore) + + def test_easy(self): + self.checkModule('pyclbr') + self.checkModule('doctest') + self.checkModule('rfc822') + self.checkModule('xmllib') + self.checkModule('difflib') + + def test_others(self): + cm = self.checkModule + + # these are about the 20 longest modules. + + cm('random', ignore=('_verify',)) # deleted + + cm('cgi', ignore=('f', 'g', # nested declarations + 'log')) # set with =, not def + + cm('mhlib', ignore=('do', # nested declaration + 'bisect')) # imported method, set with = + + cm('urllib', ignore=('getproxies_environment', # set with = + 'getproxies_registry')) # set with = + + #XXXX bad example + #cm('urllib2', ignore=('at_cnri', # defined inside __main__ + # '__super_init', # set with =. + # '_HTTPError__super_init', # set with =. + # 'http_error_301', # set with =. + # )) + + + + cm('pickle', ignore=('g',)) # deleted declaration + + cm('aifc', ignore=('openfp',)) # set with = + + cm('httplib', ignore=('error',)) # set with = + + + cm('Cookie', ignore=('__str__', 'Cookie')) # set with = + + cm('sre_parse', ignore=('literal', # nested def + 'makedict', 'dump' # from sre_constants + )) + + cm('test.test_pyclbr', + module=sys.modules[__name__]) + + # pydoc doesn't work because of string issues + # cm('pydoc', pydoc) + + # pdb plays too many dynamic games + # cm('pdb', pdb) + +run_unittest(PyclbrTest) + + + |