def test_no_attach_history_handler_when_history_not_configured(
self, mock_recorder, mock_db_sqlite3, mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
attach_history_handler(session=mock_session, parsed_args=parsed_args)
self.assertFalse(mock_recorder.add_handler.called)
self.assertFalse(mock_db_sqlite3.connect.called)
def test_no_attach_history_handler_when_history_not_configured(
self, mock_recorder, mock_db_sqlite3, mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
attach_history_handler(session=mock_session, parsed_args=parsed_args)
self.assertFalse(mock_recorder.add_handler.called)
self.assertFalse(mock_db_sqlite3.connect.called)
def test_profile_does_not_exist(self, mock_recorder, mock_db_sqlite3,
mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.side_effect = ProfileNotFound(
profile='no-exist')
parsed_args = argparse.Namespace()
parsed_args.command = 'configure'
attach_history_handler(session=mock_session, parsed_args=parsed_args)
self.assertFalse(mock_recorder.add_handler.called)
self.assertFalse(mock_db_sqlite3.connect.called)
def test_profile_does_not_exist(self, mock_recorder, mock_db_sqlite3,
mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.side_effect = ProfileNotFound(
profile='no-exist')
parsed_args = argparse.Namespace()
parsed_args.command = 'configure'
attach_history_handler(session=mock_session, parsed_args=parsed_args)
self.assertFalse(mock_recorder.add_handler.called)
self.assertFalse(mock_db_sqlite3.connect.called)
def test_attach_history_handler(self, mock_recorder, mock_db_sqlite3, mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {
'cli_history': 'enabled'
}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
attach_history_handler(session=mock_session, parsed_args=parsed_args)
self.assertEqual(mock_recorder.add_handler.call_count, 1)
self.assertIsInstance(
mock_recorder.add_handler.call_args[0][0], DatabaseHistoryHandler)
self.assertTrue(mock_db_sqlite3.connect.called)
def test_attach_history_handler(self, mock_recorder, mock_db_sqlite3,
mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {
'cli_history': 'enabled'
}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
attach_history_handler(session=mock_session, parsed_args=parsed_args)
self.assertEqual(mock_recorder.add_handler.call_count, 1)
self.assertIsInstance(mock_recorder.add_handler.call_args[0][0],
DatabaseHistoryHandler)
self.assertTrue(mock_db_sqlite3.connect.called)
def test_no_attach_history_handler_when_no_sqlite3(self, mock_recorder,
mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {
'cli_history': 'enabled'
}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
with mock.patch('sys.stderr', StringIO()) as mock_stderr:
attach_history_handler(session=mock_session,
parsed_args=parsed_args)
self.assertIn('enabled but sqlite3 is unavailable',
mock_stderr.getvalue())
self.assertFalse(mock_recorder.add_handler.called)
self.assertFalse(mock_sqlite3.connect.called)
def test_no_attach_history_handler_when_no_sqlite3(
self, mock_recorder, mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {
'cli_history': 'enabled'
}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
with mock.patch('sys.stderr', StringIO()) as mock_stderr:
attach_history_handler(
session=mock_session, parsed_args=parsed_args)
self.assertIn(
'enabled but sqlite3 is unavailable', mock_stderr.getvalue())
self.assertFalse(mock_recorder.add_handler.called)
self.assertFalse(mock_sqlite3.connect.called)
def test_create_directory_no_exists(self, mock_recorder, mock_db_sqlite3,
mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {
'cli_history': 'enabled'
}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
directory_to_create = os.path.join(self.files.rootdir, 'create-dir')
db_filename = os.path.join(directory_to_create, 'name.db')
with mock.patch('os.environ', {'AWS_CLI_HISTORY_FILE': db_filename}):
attach_history_handler(session=mock_session,
parsed_args=parsed_args)
self.assertEqual(mock_recorder.add_handler.call_count, 1)
# Is should create any missing parent directories of the
# file as well.
self.assertTrue(os.path.exists(directory_to_create))
self.assertTrue(mock_db_sqlite3.connect.called)
def test_create_directory_no_exists(self, mock_recorder,
mock_db_sqlite3, mock_sqlite3):
mock_session = mock.Mock(Session)
mock_session.get_scoped_config.return_value = {
'cli_history': 'enabled'
}
parsed_args = argparse.Namespace()
parsed_args.command = 's3'
directory_to_create = os.path.join(self.files.rootdir, 'create-dir')
db_filename = os.path.join(directory_to_create, 'name.db')
with mock.patch('os.environ', {'AWS_CLI_HISTORY_FILE': db_filename}):
attach_history_handler(
session=mock_session, parsed_args=parsed_args)
self.assertEqual(mock_recorder.add_handler.call_count, 1)
# Is should create any missing parent directories of the
# file as well.
self.assertTrue(os.path.exists(directory_to_create))
self.assertTrue(mock_db_sqlite3.connect.called)