169 lines
6.0 KiB
Python
169 lines
6.0 KiB
Python
'''
|
|
Test cases for pyclbr.py
|
|
Nick Mathewson
|
|
'''
|
|
from test.test_support import run_unittest
|
|
import unittest, sys
|
|
from types import ClassType, FunctionType, MethodType
|
|
import pyclbr
|
|
from unittest import TestCase
|
|
|
|
# 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(TestCase):
|
|
|
|
def assertListEq(self, l1, l2, ignore):
|
|
''' succeed iff {l1} - {ignore} == {l2} - {ignore} '''
|
|
l1.sort()
|
|
l2.sort()
|
|
try:
|
|
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)
|
|
except:
|
|
print >>sys.stderr, "l1=%r\nl2=%r\nignore=%r" % (l1, l2, ignore)
|
|
raise
|
|
|
|
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),
|
|
'expected hasattr(%r, %r)' % (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 >>sys.stderr, "***",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:
|
|
# Import it.
|
|
# ('<silly>' is to work around an API silliness in __import__)
|
|
module = __import__(moduleName, globals(), {}, ['<silly>'])
|
|
|
|
dict = pyclbr.readmodule_ex(moduleName)
|
|
|
|
def ismethod(obj, name):
|
|
if not isinstance(obj, MethodType):
|
|
return False
|
|
if obj.im_self is not None:
|
|
return False
|
|
objname = obj.__name__
|
|
if objname.startswith("__") and not objname.endswith("__"):
|
|
objname = "_%s%s" % (obj.im_class.__name__, objname)
|
|
return objname == name
|
|
|
|
# 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 ]
|
|
|
|
try:
|
|
self.assertListEq(real_bases, pyclbr_bases, ignore)
|
|
except:
|
|
print >>sys.stderr, "class=%s" % py_item
|
|
raise
|
|
|
|
actualMethods = []
|
|
for m in py_item.__dict__.keys():
|
|
if ismethod(getattr(py_item, m), m):
|
|
actualMethods.append(m)
|
|
foundMethods = []
|
|
for m in value.methods.keys():
|
|
if m[:2] == '__' and m[-2:] != '__':
|
|
foundMethods.append('_'+name+m)
|
|
else:
|
|
foundMethods.append(m)
|
|
|
|
try:
|
|
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
|
|
except:
|
|
print >>sys.stderr, "class=%s" % py_item
|
|
raise
|
|
|
|
# Now check for missing stuff.
|
|
def defined_in(item, module):
|
|
if isinstance(item, ClassType):
|
|
return item.__module__ == module.__name__
|
|
if isinstance(item, FunctionType):
|
|
return item.func_globals is module.__dict__
|
|
return False
|
|
for name in dir(module):
|
|
item = getattr(module, name)
|
|
if isinstance(item, (ClassType, FunctionType)):
|
|
if defined_in(item, module):
|
|
self.assertHaskey(dict, name, ignore)
|
|
|
|
def test_easy(self):
|
|
self.checkModule('pyclbr')
|
|
self.checkModule('doctest')
|
|
self.checkModule('rfc822')
|
|
self.checkModule('difflib')
|
|
|
|
def test_others(self):
|
|
cm = self.checkModule
|
|
|
|
# These were once about the 10 longest modules
|
|
cm('random', ignore=('Random',)) # from _random import Random as CoreGenerator
|
|
cm('cgi', ignore=('log',)) # set with = in module
|
|
cm('mhlib')
|
|
cm('urllib', ignore=('getproxies_registry',
|
|
'open_https')) # not on all platforms
|
|
cm('pickle', ignore=('g',)) # from types import *
|
|
cm('aifc', ignore=('openfp',)) # set with = in module
|
|
cm('Cookie')
|
|
cm('sre_parse', ignore=('dump',)) # from sre_constants import *
|
|
cm('pdb')
|
|
cm('pydoc')
|
|
|
|
# Tests for modules inside packages
|
|
cm('email.Parser')
|
|
cm('test.test_pyclbr')
|
|
|
|
|
|
def test_main():
|
|
run_unittest(PyclbrTest)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
test_main()
|