147 lines
4.4 KiB
Python
147 lines
4.4 KiB
Python
# 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 as xxx_todo_changeme:
|
|
(hr, desc, exc, argErr) = xxx_todo_changeme.args
|
|
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 as xxx_todo_changeme1:
|
|
(hr, desc, exc, argErr) = xxx_todo_changeme1.args
|
|
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 as xxx_todo_changeme2:
|
|
(hr, desc, exc, argErr) = xxx_todo_changeme2.args
|
|
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 as xxx_todo_changeme3:
|
|
(hr, fn, desc, arg) = xxx_todo_changeme3.args
|
|
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()
|