def test_create_repository_repo_already_in_results(self): client = Mock() client.snapshot.get_repository.return_value = {'repo':{'foo':'bar'}} with self.assertRaises(SystemExit) as cm: curator.create_repository(client, repository="repo", repo_type="fs") self.assertEqual(cm.exception.code, 1)
def test_create_repository_repo_already_in_results(self): client = Mock() client.snapshot.get_repository.return_value = {'repo': {'foo': 'bar'}} with self.assertRaises(SystemExit) as cm: curator.create_repository(client, repository="repo", repo_type="fs") self.assertEqual(cm.exception.code, 1)
def test_empty_result_call(self): client = Mock() client.snapshot.get_repository.return_value = None self.assertTrue( curator.create_repository(client, repository="repo", repo_type="fs"))
def test_create_repository_exception(self): client = Mock() client.snapshot.get_repository.return_value = {"not_your_repo": {"foo": "bar"}} client.snapshot.create_repository.side_effect = elasticsearch.TransportError( 500, "Error message", {"message": "Error"} ) self.assertRaises(Exception, curator.create_repository(client, repository="repo", repo_type="fs"))
def lambda_handler(event, context): if not SNAPSHOT_INDEX_FILTERS and not DELETE_INDEX_FILTERS: raise ValueError( 'No value for delete_index_filters or snapshot_index_filters found - aborting' ) if SNAPSHOT_INDEX_FILTERS: if not SNAPSHOT_BUCKET or not SNAPSHOT_BUCKET_REGION or not SNAPSHOT_NAME: raise ValueError( 'Some required snapshot parameters have no values - aborting') es = elasticsearch_client() result = {} if SNAPSHOT_INDEX_FILTERS: snapshot_name = datetime.now().strftime(SNAPSHOT_NAME) indices = filter_indices(es, SNAPSHOT_INDEX_FILTERS) if TEST_MODE: result['snapshots'] = indices.working_list() result['test_mode'] = True else: if indices.working_list(): if not curator.repository_exists(es, repository=SNAPSHOT_BUCKET): print('Registering snapshot repository in s3://{}'.format( SNAPSHOT_BUCKET)) response = curator.create_repository( client=es, repository=SNAPSHOT_BUCKET, repo_type='s3', bucket=SNAPSHOT_BUCKET, region=SNAPSHOT_BUCKET_REGION) print('Response: ' + str(response)) print('Creating a snapshot of indices') snapshot_indices = curator.Snapshot(indices, repository=SNAPSHOT_BUCKET, name=snapshot_name) snapshot_indices.do_action() result['snapshot'] = indices.working_list() if DELETE_INDEX_FILTERS: indices = filter_indices(es, DELETE_INDEX_FILTERS) if TEST_MODE: result['deleted'] = indices.working_list() result['test_mode'] = True else: if indices.working_list(): print('Deleting indices') delete_indices = curator.DeleteIndices(indices) delete_indices.do_action() result['deleted'] = indices.working_list() return result
def test_repo_not_in_results(self): client = Mock() client.snapshot.get_repository.return_value = { 'not_your_repo': { 'foo': 'bar' } } self.assertTrue( curator.create_repository(client, repository="repo", repo_type="fs"))
def test_create_repository_exception(self): client = Mock() client.snapshot.get_repository.return_value = { 'not_your_repo': { 'foo': 'bar' } } client.snapshot.create_repository.side_effect = fake_fail self.assertRaises( Exception, curator.create_repository(client, repository="repo", repo_type="fs"))
def test_create_repository_exception(self): client = Mock() client.snapshot.get_repository.return_value = {'not_your_repo':{'foo':'bar'}} client.snapshot.create_repository.side_effect = fake_fail self.assertRaises(Exception, curator.create_repository(client, repository="repo", repo_type="fs"))
def test_create_repository_repo_not_in_results(self): client = Mock() client.snapshot.get_repository.return_value = {'not_your_repo':{'foo':'bar'}} self.assertTrue(curator.create_repository(client, repository="repo", repo_type="fs"))
def test_create_repository_empty_result_call(self): client = Mock() client.snapshot.get_repository.return_value = None self.assertTrue(curator.create_repository(client, repository="repo", repo_type="fs"))
def test_create_repository_missing_arg(self): client = Mock() with self.assertRaises(SystemExit) as cm: curator.create_repository(client) self.assertEqual(cm.exception.code, 1)
def test_create_repository_exception(self): client = Mock() client.snapshot.get_repository.return_value = {'not_your_repo':{'foo':'bar'}} client.snapshot.create_repository.side_effect = elasticsearch.TransportError(500, "Error message", {"message":"Error"}) self.assertRaises(Exception, curator.create_repository(client, repository="repo", repo_type="fs"))