2
2
import sys
3
3
4
4
import pytest
5
-
6
5
from _pytest_mock_version import version
6
+
7
7
__version__ = version
8
8
9
9
@@ -34,7 +34,7 @@ class MockFixture(object):
34
34
def __init__ (self , config ):
35
35
self ._patches = [] # list of mock._patch objects
36
36
self ._mocks = [] # list of MagicMock objects
37
- self ._mock_module = mock_module = _get_mock_module (config )
37
+ self .mock_module = mock_module = _get_mock_module (config )
38
38
self .patch = self ._Patcher (self ._patches , self ._mocks , mock_module )
39
39
# aliases for convenience
40
40
self .Mock = mock_module .Mock
@@ -104,7 +104,7 @@ def stub(self, name=None):
104
104
:rtype: mock.MagicMock
105
105
:return: Stub object.
106
106
"""
107
- return self ._mock_module .MagicMock (spec = lambda * args , ** kwargs : None , name = name )
107
+ return self .mock_module .MagicMock (spec = lambda * args , ** kwargs : None , name = name )
108
108
109
109
class _Patcher (object ):
110
110
"""
@@ -115,7 +115,7 @@ class _Patcher(object):
115
115
def __init__ (self , patches , mocks , mock_module ):
116
116
self ._patches = patches
117
117
self ._mocks = mocks
118
- self ._mock_module = mock_module
118
+ self .mock_module = mock_module
119
119
120
120
def _start_patch (self , mock_func , * args , ** kwargs ):
121
121
"""Patches something by calling the given function from the mock
@@ -130,20 +130,20 @@ def _start_patch(self, mock_func, *args, **kwargs):
130
130
131
131
def object (self , * args , ** kwargs ):
132
132
"""API to mock.patch.object"""
133
- return self ._start_patch (self ._mock_module .patch .object , * args , ** kwargs )
133
+ return self ._start_patch (self .mock_module .patch .object , * args , ** kwargs )
134
134
135
135
def multiple (self , * args , ** kwargs ):
136
136
"""API to mock.patch.multiple"""
137
- return self ._start_patch (self ._mock_module .patch .multiple , * args ,
137
+ return self ._start_patch (self .mock_module .patch .multiple , * args ,
138
138
** kwargs )
139
139
140
140
def dict (self , * args , ** kwargs ):
141
141
"""API to mock.patch.dict"""
142
- return self ._start_patch (self ._mock_module .patch .dict , * args , ** kwargs )
142
+ return self ._start_patch (self .mock_module .patch .dict , * args , ** kwargs )
143
143
144
144
def __call__ (self , * args , ** kwargs ):
145
145
"""API to mock.patch"""
146
- return self ._start_patch (self ._mock_module .patch , * args , ** kwargs )
146
+ return self ._start_patch (self .mock_module .patch , * args , ** kwargs )
147
147
148
148
149
149
@pytest .yield_fixture
0 commit comments