[ctypes-commit] ctypes/comtypes/unittests test_basic.py,NONE,1.1 runtests.py,NONE,1.1 .cvsignore,NONE,1.1

[Thomas H. <th...@us...>]

Update of /cvsroot/ctypes/ctypes/comtypes/unittests
In directory sc8-pr-cvs1.sourceforge.net:/tmp/cvs-serv28228

Added Files:
	test_basic.py runtests.py .cvsignore 
Log Message:
comtypes is advanced enough to be checked in.
Not that it really works, though ;-)


--- NEW FILE: .cvsignore ---
test_basic.pyc

--- NEW FILE: test_basic.py ---
import unittest
from ctypes import windll, POINTER, byref
from comtypes import IUnknown

class BasicTest(unittest.TestCase):
    def test_IUnknown(self):
        from comtypes import IUnknown
        self.failUnlessEqual(IUnknown._nummethods_, 3)

    def test_release(self):
        POINTER(IUnknown)()

    def test_refcounts(self):
        p = POINTER(IUnknown)()
        windll.oleaut32.CreateTypeLib(1, u"blabla", byref(p))
        # initial refcount is 2
        for i in range(2, 10):
            self.failUnlessEqual(p.AddRef(), i)
        for i in range(8, 0, -1):
            self.failUnlessEqual(p.Release(), i)

    def test_qi(self):
        p = POINTER(IUnknown)()
        windll.oleaut32.CreateTypeLib(1, u"blabla", byref(p))
        self.failUnlessEqual(p.AddRef(), 2)
        self.failUnlessEqual(p.Release(), 1)

        other = p.QueryInterface(IUnknown)
        self.failUnlessEqual(other.AddRef(), 3)
        self.failUnlessEqual(p.AddRef(), 4)
        self.failUnlessEqual(p.Release(), 3)
        self.failUnlessEqual(other.Release(), 2)

        del p # calls p.Release()

        self.failUnlessEqual(other.AddRef(), 2)
        self.failUnlessEqual(other.Release(), 1)
        
    def test_derived(self):

        class IMyInterface(IUnknown):
            pass

        self.failUnlessEqual(IMyInterface._nummethods_, 3)

        IMyInterface._methods_ = []
        self.failUnlessEqual(IMyInterface._nummethods_, 3)

    def test_mro(self):
        mro = POINTER(IUnknown).__mro__

        self.failUnlessEqual(mro[0], POINTER(IUnknown))
        self.failUnlessEqual(mro[1], IUnknown)

        # the IUnknown class has the actual methods:
        self.failUnless(IUnknown.__dict__.get("QueryInterface"))

if __name__ == "__main__":
    unittest.main()

--- NEW FILE: runtests.py ---
"""Usage: runtests.py [-q] [-v] [mask]

Run all tests found in this directory, and print a summary of the results.
Command line flags:
  -v     verbose mode: print the test currently executed
  -q     quiet mode: don't prnt anything while the tests are running
  mask   mask to select filenames containing testcases, wildcards allowed
"""
import glob, os, sys, unittest, getopt

def get_suite(mask):
    if not mask.endswith(".py"):
        mask += ".py"
    test_suites = []
    for fname in glob.glob(mask):
        try:
            mod = __import__(os.path.splitext(fname)[0])
        except Exception, detail:
            print "Warning: could not import %s: %s" % (fname, detail)
            continue
        for name in dir(mod):
            if name.startswith("_"):
                continue
            o = getattr(mod, name)
            if type(o) is type(unittest.TestCase) and issubclass(o, unittest.TestCase):
                test_suites.append(unittest.makeSuite(o))
    return unittest.TestSuite(test_suites)

def usage():
    print __doc__
    return 1

def main():
    verbosity = 1
    mask = "test_*.py"
    try:
        opts, args = getopt.getopt(sys.argv[1:], "qv")
    except getopt.error:
        return usage()
    for flag, value in opts:
        if flag == "-q":
            verbosity -= 1
        elif flag == "-v":
            verbosity += 1
    if args:
        mask = args[0]
    suite = get_suite(mask)
    runner = unittest.TextTestRunner(verbosity=verbosity)
    return bool(runner.run(suite).errors)


if __name__ == "__main__":
    sys.exit(main())