1
- from collections import namedtuple
2
1
import os
3
2
4
3
import pytest
@@ -149,50 +148,45 @@ def foo(on_something):
149
148
stub .assert_called_once_with ('foo' , 'bar' )
150
149
151
150
152
- def test_mocker_spy (mocker_spy_cases ):
153
- foo , expect_obj = mocker_spy_cases
154
-
155
- assert foo .bar (arg = 10 ) == 20
156
- if expect_obj :
157
- foo .bar .assert_called_once_with (foo , arg = 10 )
158
- else :
159
- foo .bar .assert_called_once_with (arg = 10 )
160
-
161
-
162
- def create_instance_method_spy (mocker ):
151
+ def test_instance_method_spy (mocker ):
163
152
class Foo (object ):
164
153
165
154
def bar (self , arg ):
166
155
return arg * 2
167
156
168
157
foo = Foo ()
169
158
mocker .spy (foo , 'bar' )
170
- return foo , False
159
+ assert foo .bar (arg = 10 ) == 20
160
+ foo .bar .assert_called_once_with (arg = 10 )
171
161
172
162
173
- def create_instance_method_by_class_spy (mocker ):
163
+ def test_instance_method_by_class_spy (mocker ):
174
164
class Foo (object ):
175
165
176
166
def bar (self , arg ):
177
167
return arg * 2
178
168
179
169
mocker .spy (Foo , 'bar' )
180
- return Foo (), True
170
+ foo = Foo ()
171
+ assert foo .bar (arg = 10 ) == 20
172
+ foo .bar .assert_called_once_with (foo , arg = 10 )
181
173
182
174
183
- def create_class_method_spy (mocker ):
175
+ def test_class_method_spy (mocker ):
184
176
class Foo (object ):
185
177
186
178
@classmethod
187
179
def bar (cls , arg ):
188
180
return arg * 2
189
181
190
182
mocker .spy (Foo , 'bar' )
191
- return Foo , False
183
+ assert Foo .bar (arg = 10 ) == 20
184
+ Foo .bar .assert_called_once_with (arg = 10 )
192
185
193
186
194
- def create_class_method_with_metaclass_spy (mocker ):
195
- class MetaFoo (type ): pass
187
+ def test_class_method_with_metaclass_spy (mocker ):
188
+ class MetaFoo (type ):
189
+ pass
196
190
197
191
class Foo (object ):
198
192
@@ -203,33 +197,18 @@ def bar(cls, arg):
203
197
return arg * 2
204
198
205
199
mocker .spy (Foo , 'bar' )
200
+ assert Foo .bar (arg = 10 ) == 20
201
+ Foo .bar .assert_called_once_with (arg = 10 )
206
202
return Foo , False
207
203
208
204
209
- def create_static_method_spy (mocker ):
205
+ def test_static_method_spy (mocker ):
210
206
class Foo (object ):
211
207
212
208
@staticmethod
213
209
def bar (arg ):
214
210
return arg * 2
215
211
216
212
mocker .spy (Foo , 'bar' )
217
- return Foo , False
218
-
219
-
220
- mock_spy_case = namedtuple ('mock_spy_case' , ['do' , 'name' ])
221
-
222
- @pytest .fixture (
223
- params = [
224
- mock_spy_case (do = create_instance_method_spy , name = 'instance_method' ),
225
- mock_spy_case (do = create_instance_method_by_class_spy , name = 'instance_method_by_class' ),
226
- mock_spy_case (do = create_class_method_spy , name = 'classmethod' ),
227
- mock_spy_case (do = create_class_method_with_metaclass_spy , name = 'classmethod_with_metaclass' ),
228
- mock_spy_case (do = create_static_method_spy , name = 'staticmethod' ),
229
- ],
230
- ids = lambda i : i .name ,
231
- )
232
- def mocker_spy_cases (mocker , request ):
233
- case = request .param
234
-
235
- return case .do (mocker )
213
+ assert Foo .bar (arg = 10 ) == 20
214
+ Foo .bar .assert_called_once_with (arg = 10 )
0 commit comments