How to test exception part in unnitest python

Question

I want to test the exception part of my code.

def main():
    try:
        logger.info("Ejecución planificada proceso Mae_CIL")
        a = ObjectA()
        a.method()
        logger.info("Success")
    except Exception as e:
        logger.error("Error: %s" % e)


@mock.patch.object(ObjectA, 'method')
def test_main(self, mock_method):
    # Case 1: Success
    mock_method.return_value = None
    main()
    self.assertEqual(1, 1)

    # Case 2: Test Exception
    mock_method.return_value = Exception
    self.assertRaises(Exception, main)

The Case 1 test the main method. And the Case 2 test the exception but the test fail with this message 'Exception: Exception not raised by main'

How can i test the exception part??

This might be a duplicate of stackoverflow.com/questions/129507/…. — Oscar
– Oscar, Commented Mar 24, 2022 at 9:01

Iguananaut · Accepted Answer · 2022-03-24 08:57:55Z

1

As you found, just setting return_value to an exception doesn't cause that exception to be raised. That's because an exception is not a return--it's a completely different execution context that almost by definition means your function does not return. It's also possible to write a function that returns an exception object (or class as in your example).

You want Mock.side_effect.

answered Mar 24, 2022 at 8:57

Iguananaut

23.8k6 gold badges54 silver badges65 bronze badges

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

Collectives™ on Stack Overflow

How to test exception part in unnitest python

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related