from win32com import storagecon
import pythoncom, os, win32api
import win32com.test.util

import unittest

class TestEnum(win32com.test.util.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()