pywin32-checkins

[<mha...@us...>]

Update of /cvsroot/pywin32/pywin32/com/win32com/test
In directory sc8-pr-cvs1:/tmp/cvs-serv5029

Added Files:
	testROT.py testShell.py testStorage.py 
Log Message:
New tests.


--- NEW FILE: testROT.py ---
import pythoncom
import unittest

class TestROT(unittest.TestCase):
    def testit(self):
        ctx = pythoncom.CreateBindCtx()
        rot = pythoncom.GetRunningObjectTable()
        num = 0
        for mk in rot:
            name = mk.GetDisplayName(ctx, None)
            num += 1
            # Monikers themselves can iterate their contents
            for sub in mk:
                num += 1

        if num < 2:
            print "Only", num, "objects in the ROT - this is unusual"

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

--- NEW FILE: testShell.py ---
import sys, os
import unittest

import pythoncom
from win32com.shell import shell
from win32com.shell.shellcon import *
from win32com.storagecon import *

class ShellTester(unittest.TestCase):
    def testShellLink(self):
        desktop = str(shell.SHGetSpecialFolderPath(0, CSIDL_DESKTOP))
        num = 0
        shellLink = pythoncom.CoCreateInstance(shell.CLSID_ShellLink, None, pythoncom.CLSCTX_INPROC_SERVER, shell.IID_IShellLink)
        persistFile = shellLink.QueryInterface(pythoncom.IID_IPersistFile)
        for base_name in os.listdir(desktop):
            name = os.path.join(desktop, base_name)
            try:
                persistFile.Load(name,STGM_READ)
            except pythoncom.com_error:
                continue
            # Resolve is slow - avoid it for our tests.
            #shellLink.Resolve(0, shell.SLR_ANY_MATCH | shell.SLR_NO_UI)
            fname, findData = shellLink.GetPath(0)
            unc = shellLink.GetPath(shell.SLGP_UNCPRIORITY)[0]
            num += 1
        self.failIf(num<3, "Only found %d items on the desktop??" % num)

    def testShellFolder(self):
        sf = shell.SHGetDesktopFolder()
        names_1 = []
        for i in sf: # Magically calls EnumObjects
            name = sf.GetDisplayNameOf(i, SHGDN_NORMAL)
            names_1.append(name)
        
        # And get the enumerator manually    
        enum = sf.EnumObjects(0, SHCONTF_FOLDERS | SHCONTF_NONFOLDERS | SHCONTF_INCLUDEHIDDEN)
        names_2 = []
        for i in enum:
            name = sf.GetDisplayNameOf(i, SHGDN_NORMAL)
            names_2.append(name)
        names_1.sort()
        names_2.sort()
        self.assertEqual(names_1, names_2)

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

--- NEW FILE: testStorage.py ---
from win32com import storagecon
import pythoncom, os, win32api

import unittest

class TestEnum(unittest.TestCase):
    def testit(self):
        fname, tmp = win32api.GetTempFileName(win32api.GetTempPath(),'stg')
        m=storagecon.STGM_READWRITE | storagecon.STGM_SHARE_EXCLUSIVE
        ##  file, mode, format, attrs (always 0), IID (IStorage or IPropertySetStorage, storage options(only used with STGFMT_DOCFILE)
        pss=pythoncom.StgOpenStorageEx(fname, m, storagecon.STGFMT_FILE, 0 , pythoncom.IID_IPropertySetStorage)
        ###                               {"Version":2,"reserved":0,"SectorSize":512,"TemplateFile":u'somefilename'})
        
        ## FMTID_SummaryInformation FMTID_DocSummaryInformation FMTID_UserDefinedProperties 
        psuser=pss.Create(pythoncom.FMTID_UserDefinedProperties,
                      pythoncom.IID_IPropertySetStorage,
                      storagecon.PROPSETFLAG_DEFAULT,
                      storagecon.STGM_READWRITE|storagecon.STGM_CREATE|storagecon.STGM_SHARE_EXCLUSIVE) ## its very picky about flag combinations!
        psuser.WriteMultiple((3,4),('hey','bubba'))
        psuser.WritePropertyNames((3,4),('property3','property4'))
        expected_summaries = []
        expected_summaries.append( ('property3', 3, pythoncom.VT_BSTR))
        expected_summaries.append( ('property4', 4, pythoncom.VT_BSTR))
        psuser=None
        
        pssum=pss.Create(pythoncom.FMTID_SummaryInformation,
                      pythoncom.IID_IPropertySetStorage,
                      storagecon.PROPSETFLAG_DEFAULT,
                      storagecon.STGM_READWRITE|storagecon.STGM_CREATE|storagecon.STGM_SHARE_EXCLUSIVE)
        pssum.WriteMultiple((storagecon.PIDSI_AUTHOR,storagecon.PIDSI_COMMENTS),('me', 'comment'))
        
        pssum=None
        pss=None    ## doesn't seem to be a close or release method, and you can't even reopen it from the same process until previous object is gone 
        
        pssread=pythoncom.StgOpenStorageEx(fname, storagecon.STGM_READ|storagecon.STGM_SHARE_EXCLUSIVE, storagecon.STGFMT_FILE, 0 , pythoncom.IID_IPropertySetStorage)
        found_summaries = []
        for psstat in pssread:
            ps=pssread.Open(psstat[0],storagecon.STGM_READ|storagecon.STGM_SHARE_EXCLUSIVE)
            for p in ps:
                p_val = ps.ReadMultiple((p[1],))[0]
                if (p[1]==storagecon.PIDSI_AUTHOR and p_val=='me') or \
                   (p[1]==storagecon.PIDSI_COMMENTS and p_val=='comment'):
                    pass
                else:
                    self.fail("Uxexpected property %s/%s" % (p, p_val))
            ps=None
            ## FMTID_UserDefinedProperties can't exist without FMTID_DocSummaryInformation, and isn't returned independently from Enum
            ## also can't be open at same time
            if psstat[0]==pythoncom.FMTID_DocSummaryInformation:
                ps=pssread.Open(pythoncom.FMTID_UserDefinedProperties,storagecon.STGM_READ|storagecon.STGM_SHARE_EXCLUSIVE)
                for p in ps:
                    found_summaries.append(p)
                ps=None
        psread=None
        expected_summaries.sort()
        found_summaries.sort()
        self.assertEqual(expected_summaries, found_summaries)

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