Add new test cases

Author: leonarduschen

stdlib/@tests/test_cases/check_unittest.py

@@ -7,7 +7,7 @@
 from fractions import Fraction
 from typing import TypedDict
 from typing_extensions import assert_type
-from unittest.mock import MagicMock, Mock, patch
+from unittest.mock import MagicMock, Mock, patch, AsyncMock
 
 case = unittest.TestCase()
 
@@ -153,12 +153,16 @@ def f_default_new(i: int, mock: MagicMock) -> str:
 def f_explicit_new(i: int) -> str:
     return "asdf"
 
+@patch("sys.exit", new_callable=lambda: 42)
+def f_explicit_new_callable(i: int, new_callable_ret: int) -> str:
+    return "asdf"
 
 assert_type(f_default_new(1), str)
 f_default_new("a")  # Not an error due to ParamSpec limitations
 assert_type(f_explicit_new(1), str)
 f_explicit_new("a")  # type: ignore[arg-type]
-
+assert_type(f_explicit_new_callable(1), str)
+f_explicit_new_callable("a")  # Same as default new
 
 @patch("sys.exit", new=Mock())
 class TestXYZ(unittest.TestCase):
@@ -171,3 +175,49 @@ def method() -> int:
 
 assert_type(TestXYZ.attr, int)
 assert_type(TestXYZ.method(), int)
+
+
+with patch("sys.exit") as default_new_enter:
+    assert_type(default_new_enter, MagicMock | AsyncMock)
+
+with patch("sys.exit", new=42) as explicit_new_enter:
+    assert_type(explicit_new_enter, int)
+
+with patch("sys.exit", new_callable=lambda: 42) as explicit_new_callable_enter:
+    assert_type(explicit_new_callable_enter, int)
+
+
+###
+# Tests for mock.patch.object
+###
+
+@patch.object(Decimal, "exp")
+def obj_f_default_new(i: int, mock: MagicMock) -> str:
+    return "asdf"
+
+
+@patch.object(Decimal, "exp", new=42)
+def obj_f_explicit_new(i: int) -> str:
+    return "asdf"
+
+@patch.object(Decimal, "exp", new_callable=lambda: 42)
+def obj_f_explicit_new_callable(i: int, new_callable_ret: int) -> str:
+    return "asdf"
+
+assert_type(obj_f_default_new(1), str)
+obj_f_default_new("a")  # Not an error due to ParamSpec limitations
+assert_type(obj_f_explicit_new(1), str)
+obj_f_explicit_new("a")  # type: ignore[arg-type]
+assert_type(obj_f_explicit_new_callable(1), str)
+obj_f_explicit_new_callable("a")  # Same as default new
+
+
+
+with patch.object(Decimal, "exp") as obj_default_new_enter:
+    assert_type(obj_default_new_enter, MagicMock | AsyncMock)
+
+with patch.object(Decimal, "exp", new=42) as obj_explicit_new_enter:
+    assert_type(obj_explicit_new_enter, int)
+
+with patch.object(Decimal, "exp", new_callable=lambda: 42) as obj_explicit_new_callable_enter:
+    assert_type(obj_explicit_new_callable_enter, int)