# Windows specific tests

from ctypes import *
import unittest

class WindowsTestCase(unittest.TestCase):
    def test_callconv_1(self):
        "Call functions with the wrong number of arguments, or the wrong calling convention"
        GetModuleHandle = windll.kernel32.GetModuleHandleA

        # ValueError: Procedure probably called with not enough arguments (4 bytes missing)
        self.assertRaises(ValueError, GetModuleHandle)

        # This one should succeeed...
        self.failUnless(GetModuleHandle(None))

        # ValueError: Procedure probably called with too many arguments (8 bytes in excess)
        self.assertRaises(ValueError, GetModuleHandle, 0, 0, 0)

    def test_callconv_2(self):
        "Call functions with the wrong number of arguments, or the wrong calling convention"
        GetModuleHandleA = cdll.kernel32.GetModuleHandleA

        # ValueError: Procedure called with not enough arguments (4 bytes missing)
        # or wrong calling convention
        self.assertRaises(ValueError, GetModuleHandleA, None)
        self.assertRaises(ValueError, GetModuleHandleA)

    def test_SEH(self):
        """Call functions with invalid arguments, and make sure that access violations
        are trapped and raise an exception"""
        self.assertRaises(WindowsError, windll.kernel32.GetModuleHandleA, 32)

    # TODO: Add tests for passing structures (and unions?) by value.
    def test_struct_by_value(self):
        from ctypes.wintypes import POINT, RECT

        pt = POINT(10, 10)
        rect = RECT(0, 0, 20, 20)
        self.failUnlessEqual(True, windll.user32.PtInRect(byref(rect), pt))

def get_suite():
    import os
    if os.name == "nt":
        return unittest.makeSuite(WindowsTestCase)
    else:
        return None

def test(verbose=0):
    runner = unittest.TextTestRunner(verbosity=verbose)
    runner.run(get_suite())

if __name__ == '__main__':
    unittest.main()