Add misc types for mock and unittest.mock (#3923)

python · Apr 24, 2020 · 31705fc · 31705fc
1 parent 92f52a2
commit 31705fc
Show file tree

Hide file tree

Showing 3 changed files with 38 additions and 2 deletions.
diff --git a/stdlib/3/unittest/mock.pyi b/stdlib/3/unittest/mock.pyi
@@ -5,6 +5,25 @@ from typing import Any, List, Optional, Text, Tuple, Type, TypeVar
 
 _T = TypeVar("_T")
 
+__all__ = [
+    'Mock',
+    'MagicMock',
+    'patch',
+    'sentinel',
+    'DEFAULT',
+    'ANY',
+    'call',
+    'create_autospec',
+    'AsyncMock',
+    'FILTER_DIR',
+    'NonCallableMock',
+    'NonCallableMagicMock',
+    'mock_open',
+    'PropertyMock',
+    'seal',
+]
+__version__: str
+
 FILTER_DIR: Any
 
 class _slotted: ...

diff --git a/tests/stubtest_whitelists/py38.txt b/tests/stubtest_whitelists/py38.txt
@@ -149,8 +149,6 @@ types.WrapperDescriptorType.__get__
 unittest.TestCase.addCleanup
 unittest.case.TestCase.addCleanup
 unittest.doModuleCleanups
-unittest.mock.MagicProxy.__call__
-unittest.mock._MockIter.__iter__
 weakref.WeakValueDictionary.__init__
 weakref.WeakValueDictionary.update
 weakref.getweakrefcount

diff --git a/third_party/2and3/mock.pyi b/third_party/2and3/mock.pyi
@@ -5,6 +5,25 @@ from typing import Any, List, Optional, Text, Tuple, Type, TypeVar
 
 _T = TypeVar("_T")
 
+__all__ = [
+    'Mock',
+    'MagicMock',
+    'patch',
+    'sentinel',
+    'DEFAULT',
+    'ANY',
+    'call',
+    'create_autospec',
+    'AsyncMock',
+    'FILTER_DIR',
+    'NonCallableMock',
+    'NonCallableMagicMock',
+    'mock_open',
+    'PropertyMock',
+    'seal',
+]
+__version__: str
+
 FILTER_DIR: Any
 
 class _slotted: ...