# testCollections.py
#
# This code tests both the client and server side of collections
# and enumerators.
#
# Also has the side effect of testing some of the PythonCOM error semantics.
import sys
import win32com.server.util
import win32com.test.util
import win32com.client
import traceback
import pythoncom
import pywintypes
import winerror
L=pywintypes.Unicode
import unittest
error = "collection test error"
def MakeEmptyEnum():
# create the Python enumerator object as a real COM object
o = win32com.server.util.wrap( win32com.server.util.Collection() )
return win32com.client.Dispatch(o)
def MakeTestEnum():
# create a sub-collection, just to make sure it works :-)
sub = win32com.server.util.wrap( win32com.server.util.Collection( ['Sub1', 2, 'Sub3']) )
# create the Python enumerator object as a real COM object
o = win32com.server.util.wrap( win32com.server.util.Collection( [1,'Two',3, sub]))
return win32com.client.Dispatch(o)
def TestEnumAgainst(o,check):
for i in range(len(check)):
if o(i) != check[i]:
raise error("Using default method gave the incorrect value - %s/%s" % (repr(o(i)), repr(check[i])))
for i in range(len(check)):
if o.Item(i) != check[i]:
raise error("Using Item method gave the incorrect value - %s/%s" % (repr(o(i)), repr(check[i])))
# First try looping.
cmp = []
for s in o:
cmp.append(s)
if cmp[:len(check)] != check:
raise error("Result after looping isnt correct - %s/%s" % (repr(cmp[:len(check)]), repr(check)))
for i in range(len(check)):
if o[i] != check[i]:
raise error("Using indexing gave the incorrect value")
def TestEnum(quiet=None):
if quiet is None:
quiet = not "-v" in sys.argv
if not quiet: print "Simple enum test"
o = MakeTestEnum()
check = [1,'Two',3]
TestEnumAgainst(o, check)
if not quiet: print "sub-collection test"
sub = o[3]
TestEnumAgainst(sub ,['Sub1', 2, 'Sub3'])
# Remove the sublist for this test!
o.Remove(o.Count()-1)
if not quiet: print "Remove item test"
del check[1]
o.Remove(1)
TestEnumAgainst(o, check)
if not quiet: print "Add item test"
o.Add('New Item')
check.append('New Item')
TestEnumAgainst(o, check)
if not quiet: print "Insert item test"
o.Insert(2, -1)
check.insert(2, -1)
TestEnumAgainst(o, check)
### This does not work!
# if not quiet: print "Indexed replace item test"
# o[2] = 'Replaced Item'
# check[2] = 'Replaced Item'
# TestEnumAgainst(o, check)
try:
o()
raise error("default method with no args worked when it shouldnt have!")
except pythoncom.com_error, (hr, desc, exc, argErr):
if hr != winerror.DISP_E_BADPARAMCOUNT:
raise error("Expected DISP_E_BADPARAMCOUNT - got %d (%s)" % (hr, desc))
try:
o.Insert("foo", 2)
raise error("Insert worked when it shouldnt have!")
except pythoncom.com_error, (hr, desc, exc, argErr):
if hr != winerror.DISP_E_TYPEMISMATCH:
raise error("Expected DISP_E_TYPEMISMATCH - got %d (%s)" % (hr, desc))
# Remove the sublist for this test!
try:
o.Remove(o.Count())
raise error("Remove worked when it shouldnt have!")
except pythoncom.com_error, (hr, desc, exc, argErr):
if hr != winerror.DISP_E_BADINDEX:
raise error("Expected DISP_E_BADINDEX - got %d (%s)" % (hr, desc))
# Test an empty collection
if not quiet: print "Empty collection test"
o = MakeEmptyEnum()
for item in o:
raise error("Empty list performed an iteration")
try:
ob = o[1]
raise error("Empty list could be indexed")
except IndexError:
pass
try:
ob = o[0]
raise error("Empty list could be indexed")
except IndexError:
pass
try:
ob = o(0)
raise error("Empty list could be indexed")
except pythoncom.com_error, (hr, fn, desc, arg):
if hr != winerror.DISP_E_BADINDEX:
raise error("Expected DISP_E_BADINDEX - got %d (%s)" % (hr, desc))
class TestCase(win32com.test.util.TestCase):
def testEnum(self):
TestEnum()
if __name__=='__main__':
unittest.main()
|