def test_main_should_capture_guerrillamail_exception(self, **kwargs):
self.setup_mocks(**kwargs)
def raise_exception(*args):
raise GuerrillaMailException()
self.mock_command.invoke.side_effect = raise_exception
main()
def test_main_should_create_session_using_settings(self,
GuerrillaMailSession,
load_settings,
**kwargs):
self.setup_mocks(GuerrillaMailSession=GuerrillaMailSession,
load_settings=load_settings,
**kwargs)
load_settings.return_value = {'arg1': 1, 'arg2': 'cheese'}
main()
GuerrillaMailSession.assert_called_with(arg1=1, arg2='cheese')
def test_main_should_save_settings_with_updated_session_id(self, save_settings, **kwargs):
self.setup_mocks(**kwargs)
self.mock_args.command = 'cheese'
def set_session_state(*args):
self.mock_session.session_id = 123
self.mock_session.email_timestamp = 4321
self.mock_session.email_address = '*****@*****.**'
self.mock_command.invoke.side_effect = set_session_state
main()
expected_settings = {'session_id': 123, 'email_timestamp': 4321, 'email_address': '*****@*****.**'}
save_settings.assert_called_with(expected_settings)
def test_main_should_save_settings_with_updated_session_id(
self, save_settings, **kwargs):
self.setup_mocks(**kwargs)
self.mock_args.command = 'cheese'
def set_session_state(*args):
self.mock_session.session_id = 123
self.mock_session.email_timestamp = 4321
self.mock_session.email_address = '*****@*****.**'
self.mock_command.invoke.side_effect = set_session_state
main()
expected_settings = {
'session_id': 123,
'email_timestamp': 4321,
'email_address': '*****@*****.**'
}
save_settings.assert_called_with(expected_settings)
def test_main_should_capture_guerrillamail_exception(self, **kwargs):
self.setup_mocks(**kwargs)
def raise_exception(*args):
raise GuerrillaMailException()
self.mock_command.invoke.side_effect = raise_exception
main()
def test_main_should_invoke_command(self, **kwargs):
self.setup_mocks(**kwargs)
main()
self.mock_command.invoke.assert_called_once_with(self.mock_session, self.mock_args)
def test_main_should_get_command_by_command_name_arg(self, get_command, **kwargs):
self.setup_mocks(get_command=get_command, **kwargs)
self.mock_args.command = 'cheese'
main()
get_command.assert_called_with('cheese')
def test_main_should_create_session_using_settings(self, GuerrillaMailSession, load_settings, **kwargs):
self.setup_mocks(GuerrillaMailSession=GuerrillaMailSession, load_settings=load_settings, **kwargs)
load_settings.return_value = {'arg1': 1, 'arg2': 'cheese'}
main()
GuerrillaMailSession.assert_called_with(arg1=1, arg2='cheese')
def test_main_should_invoke_command(self, **kwargs):
self.setup_mocks(**kwargs)
main()
self.mock_command.invoke.assert_called_once_with(
self.mock_session, self.mock_args)
def test_main_should_get_command_by_command_name_arg(
self, get_command, **kwargs):
self.setup_mocks(get_command=get_command, **kwargs)
self.mock_args.command = 'cheese'
main()
get_command.assert_called_with('cheese')
