def test_success_or_exception_succes(self, logger): input_string = 'Success' succes_msg = 'action succes' fail_msg = 'action fail' Adb.success_or_exception(input_string, succes_msg, fail_msg) logger.assert_called_once_with(succes_msg)
def test_success_or_exception_exception(self, logger): input_string = 'fail' succes_msg = 'action succes' fail_msg = 'action fail' with pytest.raises(Adb.AdbError): Adb.success_or_exception(input_string, succes_msg, fail_msg) logger.assert_called_once_with( fail_msg + '\nMessage returned:\n{}'.format(input_string))