How to assert multiple method calls with different parameters each time in python unittest?

Question

I am using python unittest to test my code. As part of my code, I am using these

boto3.client('sts')
boto3.client('ec2')
boto3.client('ssm', arg1, arg2)

So I mocked boto3 before writing my test case, and taking it as argument. Now I can assert whether boto3.client called or not.

But I want to check whehter boto3.client called with sts and boto3.client called wit ec2 and with ssm,arg1,arg2.

When there is only one call I was able to do using boto3.client.assert_called_with('my parameters'). But facing issue to check multiple calls with different parameters each time.

@patch('createCustomer.src.main.boto3')
    def test_my_code(self, mock_boto3):
        # Executing main class
        mainclass(arg1, arg2)
        # Verifing
        mock_boto3.client.assert_called()

I want to achieve like

mock_boto3.client.assert_called_once_with('sts')
mock_boto3.client.assert_called_once_with('ec2')
mock_boto3.client.assert_called_once_with('ssm',arg1,arg2)

But this is giving error in first assert only, saying boto3.client called 3 times and parameters it is showing of last call i.e. 'ssm', arg1, arg2

Samuel Dauzon · Accepted Answer · 2024-04-25 17:09:47Z

8

If you want to verify multiple calls to the same mock, you can use assert_has_calls with call. In your case:

from unittest.mock import call

mock_boto3.client.assert_has_calls([
    call('sts'),
    call('ec2'),
    call('ssm', arg1, arg2)
])

edited Apr 25, 2024 at 17:09

Samuel Dauzon

11.4k15 gold badges66 silver badges102 bronze badges

answered Oct 15, 2019 at 20:03

jonrsharpe

123k31 gold badges277 silver badges488 bronze badges

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

3 Comments

venkat Over a year ago

is this call is part of syntax? If I am writing call('sts'), it is giving syntax error, unresolved reference to call / name call is not defined. But when I check has_calls expected parameters it contains call('st's)

jonrsharpe Over a year ago

@venkat do you mean a NameError? Like patch, call is part of unittest.mock; you'll have to import it from that module similarly.

venkat Over a year ago

ohh! got it. I did, from unittest.mock import call and it worked. Previously I am just importing patch and MagicMock

Collectives™ on Stack Overflow

How to assert multiple method calls with different parameters each time in python unittest?

1 Answer 1

3 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

3 Comments

Your Answer

Sign up or log in

Post as a guest

Related