CODEWITHSUNDEEP
pythonunit-testingmocking
Rastikan
Rastikan

Reputation: 532

Python unittest mock: Is it possible to mock the value of the __init__'s default arguments at test time?

I want to test a class created with default parameters by replacing the value of the default parameter during unit testing.

For example, I want the following line (throughout the code) obj = SomeClass()

To look like it was called with a parameter

obj = SomeClass(overiden_parameter)

One solution might be to create simple subclass:

``` def OriginalClass(object):

def __init_(some_param="default_value"):
    ...

```

``` def MockedOriginalClass(OriginalClass):

def __init_():
    super(MockedOriginalClass, self).__init__("some_other_value)
    ...

```

How to a mock/patch OriginalClass to be MockedOriginalClass thoughout the code? Keep in mind that I do want to keep functionality of the original class, the only one thing I want to change is it's default __init__ parameter.

I feel this is a very simple thing to do with Mocking, I just didn't quite figure how to do it.

I found out about this question: Python unittest mock: Is it possible to mock the value of a method's default arguments at test time?

It's very close but I don't think the same trick can be applied to the __init__ method.

Upvotes: 1

Views: 1238

Answers (2)

Irtiza
Irtiza

Reputation: 333

One way to do this is by mocking the whole class for you specific tests like this:

Example: I have a class SomeClass that I want to mock. My mocking class name is MockSomeClass that will mock class SomeClass.

class MockSomeClass(SomeClass):
    '''
    Mock Class
    '''
    def __init__(overiden_parameter):
        self.overiden_parameter = overiden_parameter

So during the test, you will use the mock class which has overridden functionality and while the other functions behavior will remain same(inheritance).

Patching

mock_some_class_obj = MockSomeClass()

@mock.patch('SomeClass', return_value=mock_some_class_obj)
def test1(self, mock_some_class_obj):
'''
Test 1
'''
    obj = SomeClass()

catch so in the code whenever you will create the object of SomeClass the object of the mock class will be returned. in the mock class, you can add your own functionality.

Upvotes: 1

Josep Valls
Josep Valls

Reputation: 5560

Look at @Martinj Pieters comment, but alternatively, you could use monkey patching https://en.wikipedia.org/wiki/Monkey_patch which is supported in pytest https://docs.pytest.org/en/documentation-restructure/how-to/monkeypatch.html to override the __init__ method.

Upvotes: 0

Related Questions