Python Mock object z metodą o nazwie multiple time

Question

Mam klasę, którą testuję, która ma jako zależność inną klasę (której wystąpienie zostaje przekazane do metody init CUT). Chcę wyśmiewać tę klasę przy użyciu biblioteki Python Mock.

Co mam coś takiego jak:

mockobj = Mock(spec=MyDependencyClass) 
mockobj.methodfromdepclass.return_value = "the value I want the mock to return" 
assertTrue(mockobj.methodfromdepclass(42), "the value I want the mock to return") 

cutobj = ClassUnderTest(mockobj) 

co jest w porządku, ale „methodfromdepclass” jest metodą parametryzowane i jako takie Chcę utworzyć jeden atrapa obiektu, gdzie w zależności od tego, co argumenty są przekazywane do methodfromdepclass zwraca różne wartości.

Powodem, dla którego chcę tego sparametryzowanego zachowania, chcę utworzyć wiele wystąpień ClassUnderTest, które zawierają różne wartości (wartości których są wytwarzane przez to, co jest zwracane z mockobj).

Kinda, co mam na myśli (to oczywiście nie działa):

mockobj = Mock(spec=MyDependencyClass) 
mockobj.methodfromdepclass.ifcalledwith(42).return_value = "you called me with arg 42" 
mockobj.methodfromdepclass.ifcalledwith(99).return_value = "you called me with arg 99" 

assertTrue(mockobj.methodfromdepclass(42), "you called me with arg 42") 
assertTrue(mockobj.methodfromdepclass(99), "you called me with arg 99") 

cutinst1 = ClassUnderTest(mockobj, 42) 
cutinst2 = ClassUnderTest(mockobj, 99) 

# now cutinst1 & cutinst2 contain different values 

Jak mogę osiągnąć ten „ifcalledwith” rodzaj semantyki?

Answer 1

Spróbuj side_effect

def my_side_effect(*args, **kwargs): 
    if args[0] == 42: 
     return "Called with 42" 
    elif args[0] == 43: 
     return "Called with 43" 
    elif kwarg['foo'] == 7: 
     return "Foo is seven" 

mockobj.mockmethod.side_effect = my_side_effect 

Answer 2

ja wpadłem na to kiedy robiłam moje własne badania. Jeśli nie dbają o przechwytywanie połączeń do methodfromdepclass(), ale wystarczy to, aby powrócić coś, po czym dodaje się może wystarczyć:

def makeFakeMethod(mapping={}): 
    def fakeMethod(inputParam): 
     return mapping[inputParam] if inputParam in mapping else MagicMock() 
    return fakeMethod 

mapping = {42:"Called with 42", 59:"Called with 59"} 
mockobj.methodfromdepclass = makeFakeMethod(mapping) 

Oto parametryzowane wersja:

def makeFakeMethod(): 
    def fakeMethod(param): 
     return "Called with " + str(param) 
    return fakeMethod 

Answer 3

Trochę słodsze :

mockobj.method.side_effect = lambda x: {123: 100, 234: 10000}[x] 

lub wieloma argumentami:

mockobj.method.side_effect = lambda *x: {(123, 234): 100, (234, 345): 10000}[x] 

lub o wartości domyślnej:

mockobj.method.side_effect = lambda x: {123: 100, 234: 10000}.get(x, 20000) 

lub kombinacja obu:

mockobj.method.side_effect = lambda *x: {(123, 234): 100, (234, 345): 10000}.get(x, 20000) 

i wesoło na wysoki idziemy.