Skip to content

gh-126384: Add tests to verify the behavior of basic COM methods. #126610

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
encukou merged 12 commits into from
Nov 22, 2024
+188 −0
Merged

gh-126384: Add tests to verify the behavior of basic COM methods. #126610

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
ff06232
Add tests.
junkmd Nov 9, 2024
8049c1a
Add comments.
junkmd Nov 9, 2024
c85bd93
Add the platform bridge.
junkmd Nov 9, 2024
f77b18d
Update `IsEqualGUID`.
junkmd Nov 9, 2024
c43dfa4
Improve naming.
junkmd Nov 10, 2024
be31c06
Add comments.
junkmd Nov 10, 2024
c09a6a1
Add comments for typing.
junkmd Nov 10, 2024
f96a1ee
Improve platform bridges.
junkmd Nov 18, 2024
c90ae51
`ProtoComMethod` -> `create_proto_com_method`
junkmd Nov 18, 2024
cc07c44
`WindowsError` -> `OSError`
junkmd Nov 18, 2024
6c60512
Add `if __name__ == '__main__'`
junkmd Nov 21, 2024
f0d3ede
Merge branch 'main' into add_com_method_tests
junkmd Nov 22, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Add comments.
  • Loading branch information
@junkmd
junkmd committed Nov 10, 2024
commit be31c0638cfb4209e5bf1cf4102d4204f7072f65
10 changes: 10 additions & 0 deletions Lib/test/test_ctypes/test_win32_com_foreign_func.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@


class GUID(ctypes.Structure):
# https://learn.microsoft.com/en-us/windows/win32/api/guiddef/ns-guiddef-guid
_fields_ = [
("Data1", DWORD),
("Data2", WORD),
Expand Down Expand Up @@ -64,11 +65,13 @@ def __get__(self, instance, owner=None):

def create_guid(name):
guid = GUID()
# https://learn.microsoft.com/en-us/windows/win32/api/combaseapi/nf-combaseapi-clsidfromstring
ole32.CLSIDFromString(name, byref(guid))
return guid


def is_equal_guid(guid1, guid2):
# https://learn.microsoft.com/en-us/windows/win32/api/objbase/nf-objbase-isequalguid
return ole32.IsEqualGUID(byref(guid1), byref(guid2))


Expand All @@ -83,26 +86,33 @@ def is_equal_guid(guid1, guid2):
IID_IPersist = create_guid("{0000010C-0000-0000-C000-000000000046}")
CLSID_ShellLink = create_guid("{00021401-0000-0000-C000-000000000046}")

# https://learn.microsoft.com/en-us/windows/win32/api/unknwn/nf-unknwn-iunknown-queryinterface(refiid_void)
proto_query_interface = ProtoComMethod(
0, HRESULT, POINTER(GUID), POINTER(c_void_p)
)
# https://learn.microsoft.com/en-us/windows/win32/api/unknwn/nf-unknwn-iunknown-addref
proto_add_ref = ProtoComMethod(1, ctypes.c_long)
# https://learn.microsoft.com/en-us/windows/win32/api/unknwn/nf-unknwn-iunknown-release
proto_release = ProtoComMethod(2, ctypes.c_long)
# https://learn.microsoft.com/en-us/windows/win32/api/objidl/nf-objidl-ipersist-getclassid
proto_get_class_id = ProtoComMethod(3, HRESULT, POINTER(GUID))


@unittest.skipUnless(sys.platform == "win32", "Windows-specific test")
class ForeignFunctionsThatWillCallComMethodsTests(unittest.TestCase):
def setUp(self):
# https://learn.microsoft.com/en-us/windows/win32/api/combaseapi/nf-combaseapi-coinitializeex
ole32.CoInitializeEx(None, COINIT_APARTMENTTHREADED)

def tearDown(self):
# https://learn.microsoft.com/en-us/windows/win32/api/combaseapi/nf-combaseapi-couninitialize
ole32.CoUninitialize()
gc.collect()

@staticmethod
def create_shelllink_persist(typ):
ppst = typ()
# https://learn.microsoft.com/en-us/windows/win32/api/combaseapi/nf-combaseapi-cocreateinstance
ole32.CoCreateInstance(
byref(CLSID_ShellLink),
None,
Expand Down