add PropertyMock to mocker

pytest-dev · Feb 20, 2016 · a1d1665 · a1d1665
1 parent e70168f
commit a1d1665
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/pytest_mock.py b/pytest_mock.py
@@ -19,6 +19,7 @@ class MockFixture(object):
 
     Mock = mock_module.Mock
     MagicMock = mock_module.MagicMock
+    PropertyMock = mock_module.PropertyMock
     ANY = mock_module.ANY
 
     def __init__(self):

diff --git a/test_pytest_mock.py b/test_pytest_mock.py
@@ -139,6 +139,13 @@ def test_mocker_has_magic_mock_class_as_attribute_for_instantiation():
     assert isinstance(mocker.MagicMock(), mock_module.MagicMock)
 
 
+def test_mocker_has_property_mock_class_as_attribute_for_instantiation():
+    from pytest_mock import mock_module, MockFixture
+
+    mocker = MockFixture()
+    assert isinstance(mocker.PropertyMock(), mock_module.PropertyMock)
+
+
 def test_mocker_has_mock_class_as_attribute_for_instantiation():
     from pytest_mock import mock_module, MockFixture