python mock functions with parameters

Question

I want to mock a function that calls an external function with parameters. I know how to mock a function, but I can't give parameters. I tried with @patch, side_effects, but no success.

def functionToTest(self, ip):
    var1 = self.config.get(self.section, 'externalValue1')
    var2 = self.config.get(self.section, 'externalValue2')
    var3 = self.config.get(self.section, 'externalValue3')

    if var1 == "xxx":
        return False
    if var2 == "yyy":
        return False

    [...]

In my test I can do this:

    def test_functionToTest(self):
    [...]
    c.config = Mock()
    c.config.get.return_value = 'xxx'

So both var1, var2 and var3 take "xxx" same value, but I don't know how to mock every single instruction and give var1, var2 and var3 values I want

(python version 2.7.3)

chepner · Accepted Answer · 2016-02-12 14:15:25Z

9

Use side_effect to queue up a series of return values.

c.config = Mock()
c.config.get.side_effect = ['xxx', 'yyy', 'zzz']

The first time c.config.get is called, it will return 'xxx'; the second time, 'yyy'; and the third time, 'zzz'. (If it is called a fourth time, it will raise a StopIteration error.)

answered Feb 12, 2016 at 14:15

chepner

538k77 gold badges594 silver badges746 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

myg Over a year ago

Hey chepner, it works! I was trying change parameters (self.section, 'externalValue') but your solution was better. Thanks!

Collectives™ on Stack Overflow

python mock functions with parameters

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related