python: specify return value of attribute of mocked module

Question

I have a module ex which has multiple attributes and another module that uses these attributes like this

import ex
def use_ex(attr):
    z = getattr(ex,attr)
    #Do something with z

In my test file, I mock the ex as I cannot import it in the test environment.

sys.modules['ex'] = Mock()

How do I specify the return_value of ex.attr when I write the test for use_ex?

Can you provide more detail? What return value do you need? Your code seems not to call `z`, so there is no chance to provide a `return_value`. You can directly set attributes on Mock objects. `ex = Mock(); ex.x = 3`. There is also `unittest.mock.PropertyMock` if you do just want a plain value. — MisterMiyagi, Nov 21 '18 at 09:39
Works for me. Maybe you want `sys.modules['ex'].attr = desired_value`? But that is just the value, not the return value. — Stop harming Monica, Nov 21 '18 at 09:46
@Goyo yes, it was a data attribute not a function so i just needed to set the value — kishlay kumar, Nov 21 '18 at 10:32

score -1 · Answer 1 · answered Nov 21 '18 at 09:39

-1

You can specify mock return values like this :

return_value = whatyouwant
my_mock = Mock()
my_mock.return_value = return_value
sys.modules['ex'] = my_mock

answered Nov 21 '18 at 09:39

Viper

Actually i just needed to set the attribute value of the mocked module – kishlay kumar Nov 21 '18 at 10:29

1 Answers1