def test_failed_login_does_not_save_config(self): response = { 'success': False, 'body': {}, } self.mock_login.return_value = response login.main() self.assertEqual( 'Authenticating against Ubuntu One SSO.\n' 'Login failed.\n', self.fake_logger.output) self.assertFalse(self.mock_save.called)
def test_successful_login_saves_config(self): response = { 'success': True, 'body': { 'consumer_key': 'consumer_key', 'consumer_secret': 'consumer_secret', 'token_key': 'token_key', 'token_secret': 'token_secret', } } self.mock_login.return_value = response login.main() self.assertEqual( 'Authenticating against Ubuntu One SSO.\n' 'Login successful.\n', self.fake_logger.output) self.mock_save.assert_called_once_with(response['body'])