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_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_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')