[test\test_storage.py] Use WinDLL instead of oledll/windll

Partially fix enthought#735
moi15moi · Jan 12, 2025 · 548661f · 548661f
1 parent a93a571
commit 548661f
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/comtypes/test/test_storage.py b/comtypes/test/test_storage.py
@@ -1,10 +1,10 @@
-from _ctypes import COMError
-from ctypes import byref, c_ubyte, POINTER
 import contextlib
-from pathlib import Path
 import unittest
+from _ctypes import COMError
+from ctypes import HRESULT, POINTER, WinDLL, byref, c_ubyte
+from ctypes.wintypes import DWORD, WCHAR
+from pathlib import Path
 
-import comtypes
 import comtypes.client
 
 
@@ -28,6 +28,11 @@
 
 STG_E_PATHNOTFOUND = -2147287038
 
+_ole32 = WinDLL("ole32")
+
+_StgCreateDocfile = _ole32.StgCreateDocfile
+_StgCreateDocfile.argtypes = [POINTER(WCHAR), DWORD, DWORD, POINTER(POINTER(IStorage))]
+_StgCreateDocfile.restype = HRESULT
 
 class Test_IStorage(unittest.TestCase):
     CREATE_DOC_FLAG = (
@@ -44,7 +49,7 @@ class Test_IStorage(unittest.TestCase):
 
     def _create_docfile(self) -> IStorage:
         stg = POINTER(IStorage)()
-        comtypes._ole32.StgCreateDocfile(None, self.CREATE_DOC_FLAG, 0, byref(stg))
+        _StgCreateDocfile(None, self.CREATE_DOC_FLAG, 0, byref(stg))
         return stg  # type: ignore
 
     def test_CreateStream(self):