70 lines
2.8 KiB
Python
70 lines
2.8 KiB
Python
|
# Test IShellItem and related interfaces
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from win32com.shell import knownfolders, shell, shellcon
|
||
|
|
|
||
|
|
|
||
|
|
class TestShellItem(unittest.TestCase):
|
||
|
|
def assertShellItemsEqual(self, i1, i2):
|
||
|
|
n1 = i1.GetDisplayName(shellcon.SHGDN_FORPARSING)
|
||
|
|
n2 = i2.GetDisplayName(shellcon.SHGDN_FORPARSING)
|
||
|
|
self.assertEqual(n1, n2)
|
||
|
|
|
||
|
|
def test_idlist_roundtrip(self):
|
||
|
|
pidl = shell.SHGetSpecialFolderLocation(0, shellcon.CSIDL_DESKTOP)
|
||
|
|
item = shell.SHCreateItemFromIDList(pidl, shell.IID_IShellItem)
|
||
|
|
pidl_back = shell.SHGetIDListFromObject(item)
|
||
|
|
self.assertEqual(pidl, pidl_back)
|
||
|
|
|
||
|
|
def test_parsing_name(self):
|
||
|
|
sf = shell.SHGetDesktopFolder()
|
||
|
|
flags = shellcon.SHCONTF_FOLDERS | shellcon.SHCONTF_NONFOLDERS
|
||
|
|
children = sf.EnumObjects(0, flags)
|
||
|
|
child_pidl = next(children)
|
||
|
|
name = sf.GetDisplayNameOf(child_pidl, shellcon.SHGDN_FORPARSING)
|
||
|
|
|
||
|
|
item = shell.SHCreateItemFromParsingName(name, None, shell.IID_IShellItem)
|
||
|
|
# test the name we get from the item is the same as from the folder.
|
||
|
|
self.assertEqual(name, item.GetDisplayName(shellcon.SHGDN_FORPARSING))
|
||
|
|
|
||
|
|
def test_parsing_relative(self):
|
||
|
|
desktop_pidl = shell.SHGetSpecialFolderLocation(0, shellcon.CSIDL_DESKTOP)
|
||
|
|
desktop_item = shell.SHCreateItemFromIDList(desktop_pidl, shell.IID_IShellItem)
|
||
|
|
|
||
|
|
sf = shell.SHGetDesktopFolder()
|
||
|
|
flags = shellcon.SHCONTF_FOLDERS | shellcon.SHCONTF_NONFOLDERS
|
||
|
|
children = sf.EnumObjects(0, flags)
|
||
|
|
child_pidl = next(children)
|
||
|
|
name_flags = shellcon.SHGDN_FORPARSING | shellcon.SHGDN_INFOLDER
|
||
|
|
name = sf.GetDisplayNameOf(child_pidl, name_flags)
|
||
|
|
|
||
|
|
item = shell.SHCreateItemFromRelativeName(
|
||
|
|
desktop_item, name, None, shell.IID_IShellItem
|
||
|
|
)
|
||
|
|
# test the name we get from the item is the same as from the folder.
|
||
|
|
self.assertEqual(name, item.GetDisplayName(name_flags))
|
||
|
|
|
||
|
|
def test_create_in_known_folder(self):
|
||
|
|
item = shell.SHCreateItemInKnownFolder(
|
||
|
|
knownfolders.FOLDERID_Desktop, 0, None, shell.IID_IShellItem
|
||
|
|
)
|
||
|
|
# this will do for now :)
|
||
|
|
|
||
|
|
def test_create_item_with_parent(self):
|
||
|
|
desktop_pidl = shell.SHGetSpecialFolderLocation(0, shellcon.CSIDL_DESKTOP)
|
||
|
|
desktop_item = shell.SHCreateItemFromIDList(desktop_pidl, shell.IID_IShellItem)
|
||
|
|
|
||
|
|
sf = shell.SHGetDesktopFolder()
|
||
|
|
flags = shellcon.SHCONTF_FOLDERS | shellcon.SHCONTF_NONFOLDERS
|
||
|
|
children = sf.EnumObjects(0, flags)
|
||
|
|
child_pidl = next(children)
|
||
|
|
item1 = shell.SHCreateItemWithParent(
|
||
|
|
desktop_pidl, None, child_pidl, shell.IID_IShellItem
|
||
|
|
)
|
||
|
|
item2 = shell.SHCreateItemWithParent(None, sf, child_pidl, shell.IID_IShellItem)
|
||
|
|
self.assertShellItemsEqual(item1, item2)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|