"""Test pywin32's error semantics"""
import sys
import unittest
import win32api, win32file, pywintypes
import pythoncom
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.failUnlessEqual(exc[index], expected)
# and that exception.args can is the same.
self.failUnlessEqual(exc.args[index], expected)
class TestAPISimple(TestBase):
def _getInvalidHandleException(self):
try:
win32api.CloseHandle(1)
except win32api.error, 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, exc:
self.failUnlessEqual(exc.winerror, winerror.ERROR_INVALID_HANDLE)
self.failUnlessEqual(exc.funcname, "CloseHandle")
expected_msg = win32api.FormatMessage(winerror.ERROR_INVALID_HANDLE).rstrip()
self.failUnlessEqual(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.failUnlessEqual(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.failUnlessEqual(tuple(exc), err_tuple)
else:
self.failUnlessEqual(exc.args, err_tuple)
def testClassName(self):
exc = self._getInvalidHandleException()
# The error class has always been named 'error'. That's not ideal :(
self.failUnlessEqual(exc.__class__.__name__, "error")
def testIdentity(self):
exc = self._getInvalidHandleException()
self.failUnless(exc.__class__ is pywintypes.error)
def testBaseClass(self):
self.failUnlessEqual(pywintypes.error.__bases__, (Exception,))
def testAttributes(self):
exc = self._getInvalidHandleException()
err_msg = win32api.FormatMessage(winerror.ERROR_INVALID_HANDLE).rstrip()
self.failUnlessEqual(exc.winerror, winerror.ERROR_INVALID_HANDLE)
self.failUnlessEqual(exc.strerror, err_msg)
self.failUnlessEqual(exc.funcname, 'CloseHandle')
# some tests for 'insane' args.
def testStrangeArgsNone(self):
try:
raise pywintypes.error()
self.fail("Expected exception")
except pywintypes.error, exc:
self.failUnlessEqual(exc.args, ())
self.failUnlessEqual(exc.winerror, None)
self.failUnlessEqual(exc.funcname, None)
self.failUnlessEqual(exc.strerror, None)
def testStrangeArgsNotEnough(self):
try:
raise pywintypes.error("foo")
self.fail("Expected exception")
except pywintypes.error, exc:
assert exc.args[0] == "foo"
# 'winerror' always args[0]
self.failUnlessEqual(exc.winerror, "foo")
self.failUnlessEqual(exc.funcname, None)
self.failUnlessEqual(exc.strerror, None)
def testStrangeArgsTooMany(self):
try:
raise pywintypes.error("foo", "bar", "you", "never", "kn", 0)
self.fail("Expected exception")
except pywintypes.error, exc:
self.failUnlessEqual(exc.args[0], "foo")
self.failUnlessEqual(exc.args[-1], 0)
self.failUnlessEqual(exc.winerror, "foo")
self.failUnlessEqual(exc.funcname, "bar")
self.failUnlessEqual(exc.strerror, "you")
class TestCOMSimple(TestBase):
def _getException(self):
try:
pythoncom.StgOpenStorage("foo", None, 0)
except pythoncom.com_error, exc:
return exc
self.fail("Didn't get storage exception.")
def testIs(self):
self.failUnless(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.failUnlessEqual(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.failUnlessEqual(tuple(exc), err_tuple)
else:
self.failUnlessEqual(exc.args, err_tuple)
def testClassName(self):
exc = self._getException()
self.failUnlessEqual(exc.__class__.__name__, "com_error")
def testIdentity(self):
exc = self._getException()
self.failUnless(exc.__class__ is pywintypes.com_error)
def testBaseClass(self):
exc = self._getException()
self.failUnlessEqual(pywintypes.com_error.__bases__, (Exception,))
def testAttributes(self):
exc = self._getException()
err_msg = win32api.FormatMessage(winerror.STG_E_INVALIDFLAG).rstrip()
self.failUnlessEqual(exc.hresult, winerror.STG_E_INVALIDFLAG)
self.failUnlessEqual(exc.strerror, err_msg)
self.failUnlessEqual(exc.argerror, None)
self.failUnlessEqual(exc.excepinfo, None)
def testStrangeArgsNone(self):
try:
raise pywintypes.com_error()
self.fail("Expected exception")
except pywintypes.com_error, exc:
self.failUnlessEqual(exc.args, ())
self.failUnlessEqual(exc.hresult, None)
self.failUnlessEqual(exc.strerror, None)
self.failUnlessEqual(exc.argerror, None)
self.failUnlessEqual(exc.excepinfo, None)
def testStrangeArgsNotEnough(self):
try:
raise pywintypes.com_error("foo")
self.fail("Expected exception")
except pywintypes.com_error, exc:
self.failUnlessEqual(exc.args[0], "foo")
self.failUnlessEqual(exc.hresult, "foo")
self.failUnlessEqual(exc.strerror, None)
self.failUnlessEqual(exc.excepinfo, None)
self.failUnlessEqual(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, exc:
self.failUnlessEqual(exc.args[0], "foo")
self.failUnlessEqual(exc.args[-1], 0)
self.failUnlessEqual(exc.hresult, "foo")
self.failUnlessEqual(exc.strerror, "bar")
self.failUnlessEqual(exc.excepinfo, "you")
self.failUnlessEqual(exc.argerror, "never")
if __name__ == '__main__':
unittest.main()
|