Export MockType/AsyncMockType for type annotations (#415)

Fix #414 --------- Co-authored-by: Bruno Oliveira <bruno@soliv.dev>
pytest-dev · Mar 21, 2024 · 366966b · 366966b
1 parent 852116b
commit 366966b
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -1,6 +1,12 @@
 Releases
 ========
 
+UNRELEASED
+----------
+
+* `#415 <https://github.com/pytest-dev/pytest-mock/pull/415>`_: ``MockType`` and ``AsyncMockType`` can be imported from ``pytest_mock`` for type annotation purposes.
+
+
 3.13.0 (2024-03-21)
 -------------------
 

diff --git a/src/pytest_mock/__init__.py b/src/pytest_mock/__init__.py
@@ -1,4 +1,6 @@
+from pytest_mock.plugin import AsyncMockType
 from pytest_mock.plugin import MockerFixture
+from pytest_mock.plugin import MockType
 from pytest_mock.plugin import PytestMockWarning
 from pytest_mock.plugin import class_mocker
 from pytest_mock.plugin import mocker
@@ -11,8 +13,10 @@
 MockFixture = MockerFixture  # backward-compatibility only (#204)
 
 __all__ = [
+    "AsyncMockType",
     "MockerFixture",
     "MockFixture",
+    "MockType",
     "PytestMockWarning",
     "pytest_addoption",
     "pytest_configure",