Test with extrem values

Author: aisk

Lib/test/test_msvcrt.py

@@ -2,6 +2,7 @@
 import subprocess
 import sys
 import unittest
+from contextlib import contextmanager
 from textwrap import dedent
 
 from test.support import os_helper, requires_resource, Py_DEBUG
@@ -109,6 +110,18 @@ def test_putwch(self):
 
 
 class TestOther(unittest.TestCase):
+    # We can't suppress the msgbox with invalid input without debug build
+    @unittest.skipUnless(Py_DEBUG, "only available under debug build")
+    @contextmanager
+    def set_crt_dbg_report_to_stderr(self):
+        mode = msvcrt.CrtSetReportMode(msvcrt.CRT_ASSERT, msvcrt.CRTDBG_MODE_FILE)
+        file = msvcrt.CrtSetReportFile(msvcrt.CRT_ASSERT, msvcrt.CRTDBG_FILE_STDERR)
+        try:
+            yield
+        finally:
+            msvcrt.CrtSetReportMode(msvcrt.CRT_ASSERT, mode)
+            msvcrt.CrtSetReportFile(msvcrt.CRT_ASSERT, file)
+
     def test_heap_min(self):
         try:
             msvcrt.heapmin()
@@ -128,6 +141,10 @@ def test_SetErrorMode(self):
         self.assertIs(type(returned), int)
         self.assertEqual(old, returned)
 
+        msvcrt.SetErrorMode(-1)
+        msvcrt.SetErrorMode(2**32)  # max unsigned int
+        msvcrt.SetErrorMode(-2**31)
+
     @unittest.skipUnless(Py_DEBUG, "only available under debug build")
     def test_set_error_mode(self):
         old = msvcrt.set_error_mode(msvcrt.REPORT_ERRMODE)
@@ -141,6 +158,11 @@ def test_set_error_mode(self):
         returned = msvcrt.set_error_mode(msvcrt.REPORT_ERRMODE)
         self.assertEqual(returned, msvcrt.OUT_TO_STDERR)
 
+        with self.set_crt_dbg_report_to_stderr():
+            self.assertEqual(msvcrt.set_error_mode(-1), -1)
+            self.assertEqual(msvcrt.set_error_mode(2**31-1), -1)
+            self.assertEqual(msvcrt.set_error_mode(-2**31), -1)
+
     @unittest.skipUnless(Py_DEBUG, "only available under debug build")
     def test_CrtSetReportMode(self):
         old = msvcrt.CrtSetReportMode(msvcrt.CRT_WARN,
@@ -157,6 +179,10 @@ def test_CrtSetReportMode(self):
                                            msvcrt.CRTDBG_REPORT_MODE)
         self.assertEqual(returned, msvcrt.CRTDBG_MODE_DEBUG)
 
+        self.assertRaises(OSError, msvcrt.CrtSetReportMode, -1, -1)
+        self.assertRaises(OSError, msvcrt.CrtSetReportMode, 2**31-1, 2**31-1)
+        self.assertRaises(OSError, msvcrt.CrtSetReportMode, -2**31, -2**31)
+
     @unittest.skipUnless(Py_DEBUG, "only available under debug build")
     def test_CrtSetReportFile(self):
         # Set the report mode to CRTDBG_REPORT_FILE at first.
@@ -176,6 +202,10 @@ def test_CrtSetReportFile(self):
                                            msvcrt.CRTDBG_REPORT_FILE)
         self.assertEqual(returned, msvcrt.get_osfhandle(sys.stdout.fileno()))
 
+        msvcrt.CrtSetReportFile(-1, -1)
+        msvcrt.CrtSetReportFile(2**31-1, 2**31-1)
+        msvcrt.CrtSetReportFile(-2**31, -2**31)
+
 
 if __name__ == "__main__":
     unittest.main()