224 lines
8.2 KiB
Python
224 lines
8.2 KiB
Python
|
"""Test pywin32's error semantics"""
|
||
|
|
import sys
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
import pythoncom
|
||
|
|
import pywintypes
|
||
|
|
import win32api
|
||
|
|
import win32file
|
||
|
|
import winerror
|
||
|
|
|
||
|
|
|
||
|
|
class TestBase(unittest.TestCase):
|
||
|
|
def _testExceptionIndex(self, exc, index, expected):
|
||
|
|
# check the exception itself can be indexed if not py3k
|
||
|
|
if sys.version_info < (3,):
|
||
|
|
self.assertEqual(exc[index], expected)
|
||
|
|
# and that exception.args can is the same.
|
||
|
|
self.assertEqual(exc.args[index], expected)
|
||
|
|
|
||
|
|
|
||
|
|
class TestAPISimple(TestBase):
|
||
|
|
def _getInvalidHandleException(self):
|
||
|
|
try:
|
||
|
|
win32api.CloseHandle(1)
|
||
|
|
except win32api.error as exc:
|
||
|
|
return exc
|
||
|
|
self.fail("Didn't get invalid-handle exception.")
|
||
|
|
|
||
|
|
def testSimple(self):
|
||
|
|
self.assertRaises(pywintypes.error, win32api.CloseHandle, 1)
|
||
|
|
|
||
|
|
def testErrnoIndex(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
self._testExceptionIndex(exc, 0, winerror.ERROR_INVALID_HANDLE)
|
||
|
|
|
||
|
|
def testFuncIndex(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
self._testExceptionIndex(exc, 1, "CloseHandle")
|
||
|
|
|
||
|
|
def testMessageIndex(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
expected = win32api.FormatMessage(winerror.ERROR_INVALID_HANDLE).rstrip()
|
||
|
|
self._testExceptionIndex(exc, 2, expected)
|
||
|
|
|
||
|
|
def testUnpack(self):
|
||
|
|
try:
|
||
|
|
win32api.CloseHandle(1)
|
||
|
|
self.fail("expected exception!")
|
||
|
|
except win32api.error as exc:
|
||
|
|
self.assertEqual(exc.winerror, winerror.ERROR_INVALID_HANDLE)
|
||
|
|
self.assertEqual(exc.funcname, "CloseHandle")
|
||
|
|
expected_msg = win32api.FormatMessage(
|
||
|
|
winerror.ERROR_INVALID_HANDLE
|
||
|
|
).rstrip()
|
||
|
|
self.assertEqual(exc.strerror, expected_msg)
|
||
|
|
|
||
|
|
def testAsStr(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
err_msg = win32api.FormatMessage(winerror.ERROR_INVALID_HANDLE).rstrip()
|
||
|
|
# early on the result actually *was* a tuple - it must always look like one
|
||
|
|
err_tuple = (winerror.ERROR_INVALID_HANDLE, "CloseHandle", err_msg)
|
||
|
|
self.assertEqual(str(exc), str(err_tuple))
|
||
|
|
|
||
|
|
def testAsTuple(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
err_msg = win32api.FormatMessage(winerror.ERROR_INVALID_HANDLE).rstrip()
|
||
|
|
# early on the result actually *was* a tuple - it must be able to be one
|
||
|
|
err_tuple = (winerror.ERROR_INVALID_HANDLE, "CloseHandle", err_msg)
|
||
|
|
if sys.version_info < (3,):
|
||
|
|
self.assertEqual(tuple(exc), err_tuple)
|
||
|
|
else:
|
||
|
|
self.assertEqual(exc.args, err_tuple)
|
||
|
|
|
||
|
|
def testClassName(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
# The error class has always been named 'error'. That's not ideal :(
|
||
|
|
self.assertEqual(exc.__class__.__name__, "error")
|
||
|
|
|
||
|
|
def testIdentity(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
self.assertTrue(exc.__class__ is pywintypes.error)
|
||
|
|
|
||
|
|
def testBaseClass(self):
|
||
|
|
self.assertEqual(pywintypes.error.__bases__, (Exception,))
|
||
|
|
|
||
|
|
def testAttributes(self):
|
||
|
|
exc = self._getInvalidHandleException()
|
||
|
|
err_msg = win32api.FormatMessage(winerror.ERROR_INVALID_HANDLE).rstrip()
|
||
|
|
self.assertEqual(exc.winerror, winerror.ERROR_INVALID_HANDLE)
|
||
|
|
self.assertEqual(exc.strerror, err_msg)
|
||
|
|
self.assertEqual(exc.funcname, "CloseHandle")
|
||
|
|
|
||
|
|
# some tests for 'insane' args.
|
||
|
|
def testStrangeArgsNone(self):
|
||
|
|
try:
|
||
|
|
raise pywintypes.error()
|
||
|
|
self.fail("Expected exception")
|
||
|
|
except pywintypes.error as exc:
|
||
|
|
self.assertEqual(exc.args, ())
|
||
|
|
self.assertEqual(exc.winerror, None)
|
||
|
|
self.assertEqual(exc.funcname, None)
|
||
|
|
self.assertEqual(exc.strerror, None)
|
||
|
|
|
||
|
|
def testStrangeArgsNotEnough(self):
|
||
|
|
try:
|
||
|
|
raise pywintypes.error("foo")
|
||
|
|
self.fail("Expected exception")
|
||
|
|
except pywintypes.error as exc:
|
||
|
|
assert exc.args[0] == "foo"
|
||
|
|
# 'winerror' always args[0]
|
||
|
|
self.assertEqual(exc.winerror, "foo")
|
||
|
|
self.assertEqual(exc.funcname, None)
|
||
|
|
self.assertEqual(exc.strerror, None)
|
||
|
|
|
||
|
|
def testStrangeArgsTooMany(self):
|
||
|
|
try:
|
||
|
|
raise pywintypes.error("foo", "bar", "you", "never", "kn", 0)
|
||
|
|
self.fail("Expected exception")
|
||
|
|
except pywintypes.error as exc:
|
||
|
|
self.assertEqual(exc.args[0], "foo")
|
||
|
|
self.assertEqual(exc.args[-1], 0)
|
||
|
|
self.assertEqual(exc.winerror, "foo")
|
||
|
|
self.assertEqual(exc.funcname, "bar")
|
||
|
|
self.assertEqual(exc.strerror, "you")
|
||
|
|
|
||
|
|
|
||
|
|
class TestCOMSimple(TestBase):
|
||
|
|
def _getException(self):
|
||
|
|
try:
|
||
|
|
pythoncom.StgOpenStorage("foo", None, 0)
|
||
|
|
except pythoncom.com_error as exc:
|
||
|
|
return exc
|
||
|
|
self.fail("Didn't get storage exception.")
|
||
|
|
|
||
|
|
def testIs(self):
|
||
|
|
self.assertTrue(pythoncom.com_error is pywintypes.com_error)
|
||
|
|
|
||
|
|
def testSimple(self):
|
||
|
|
self.assertRaises(pythoncom.com_error, pythoncom.StgOpenStorage, "foo", None, 0)
|
||
|
|
|
||
|
|
def testErrnoIndex(self):
|
||
|
|
exc = self._getException()
|
||
|
|
self._testExceptionIndex(exc, 0, winerror.STG_E_INVALIDFLAG)
|
||
|
|
|
||
|
|
def testMessageIndex(self):
|
||
|
|
exc = self._getException()
|
||
|
|
expected = win32api.FormatMessage(winerror.STG_E_INVALIDFLAG).rstrip()
|
||
|
|
self._testExceptionIndex(exc, 1, expected)
|
||
|
|
|
||
|
|
def testAsStr(self):
|
||
|
|
exc = self._getException()
|
||
|
|
err_msg = win32api.FormatMessage(winerror.STG_E_INVALIDFLAG).rstrip()
|
||
|
|
# early on the result actually *was* a tuple - it must always look like one
|
||
|
|
err_tuple = (winerror.STG_E_INVALIDFLAG, err_msg, None, None)
|
||
|
|
self.assertEqual(str(exc), str(err_tuple))
|
||
|
|
|
||
|
|
def testAsTuple(self):
|
||
|
|
exc = self._getException()
|
||
|
|
err_msg = win32api.FormatMessage(winerror.STG_E_INVALIDFLAG).rstrip()
|
||
|
|
# early on the result actually *was* a tuple - it must be able to be one
|
||
|
|
err_tuple = (winerror.STG_E_INVALIDFLAG, err_msg, None, None)
|
||
|
|
if sys.version_info < (3,):
|
||
|
|
self.assertEqual(tuple(exc), err_tuple)
|
||
|
|
else:
|
||
|
|
self.assertEqual(exc.args, err_tuple)
|
||
|
|
|
||
|
|
def testClassName(self):
|
||
|
|
exc = self._getException()
|
||
|
|
self.assertEqual(exc.__class__.__name__, "com_error")
|
||
|
|
|
||
|
|
def testIdentity(self):
|
||
|
|
exc = self._getException()
|
||
|
|
self.assertTrue(exc.__class__ is pywintypes.com_error)
|
||
|
|
|
||
|
|
def testBaseClass(self):
|
||
|
|
exc = self._getException()
|
||
|
|
self.assertEqual(pywintypes.com_error.__bases__, (Exception,))
|
||
|
|
|
||
|
|
def testAttributes(self):
|
||
|
|
exc = self._getException()
|
||
|
|
err_msg = win32api.FormatMessage(winerror.STG_E_INVALIDFLAG).rstrip()
|
||
|
|
self.assertEqual(exc.hresult, winerror.STG_E_INVALIDFLAG)
|
||
|
|
self.assertEqual(exc.strerror, err_msg)
|
||
|
|
self.assertEqual(exc.argerror, None)
|
||
|
|
self.assertEqual(exc.excepinfo, None)
|
||
|
|
|
||
|
|
def testStrangeArgsNone(self):
|
||
|
|
try:
|
||
|
|
raise pywintypes.com_error()
|
||
|
|
self.fail("Expected exception")
|
||
|
|
except pywintypes.com_error as exc:
|
||
|
|
self.assertEqual(exc.args, ())
|
||
|
|
self.assertEqual(exc.hresult, None)
|
||
|
|
self.assertEqual(exc.strerror, None)
|
||
|
|
self.assertEqual(exc.argerror, None)
|
||
|
|
self.assertEqual(exc.excepinfo, None)
|
||
|
|
|
||
|
|
def testStrangeArgsNotEnough(self):
|
||
|
|
try:
|
||
|
|
raise pywintypes.com_error("foo")
|
||
|
|
self.fail("Expected exception")
|
||
|
|
except pywintypes.com_error as exc:
|
||
|
|
self.assertEqual(exc.args[0], "foo")
|
||
|
|
self.assertEqual(exc.hresult, "foo")
|
||
|
|
self.assertEqual(exc.strerror, None)
|
||
|
|
self.assertEqual(exc.excepinfo, None)
|
||
|
|
self.assertEqual(exc.argerror, None)
|
||
|
|
|
||
|
|
def testStrangeArgsTooMany(self):
|
||
|
|
try:
|
||
|
|
raise pywintypes.com_error("foo", "bar", "you", "never", "kn", 0)
|
||
|
|
self.fail("Expected exception")
|
||
|
|
except pywintypes.com_error as exc:
|
||
|
|
self.assertEqual(exc.args[0], "foo")
|
||
|
|
self.assertEqual(exc.args[-1], 0)
|
||
|
|
self.assertEqual(exc.hresult, "foo")
|
||
|
|
self.assertEqual(exc.strerror, "bar")
|
||
|
|
self.assertEqual(exc.excepinfo, "you")
|
||
|
|
self.assertEqual(exc.argerror, "never")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|