def test_reduce_without_snapshot_id(): # creating block device first_block_device = AWSBlockDevice() first_block_device.snapshot_id = None # creating tests objects first_ami = AMI() first_ami.id = 'ami-28c2b348' first_ami.name = "ubuntu-20160102" first_ami.block_device_mappings.append(first_block_device) # creating amis to drop dict candidates = [first_ami] assert AMICleaner().reduce_candidates(candidates) == candidates
def test_deletion_ami_min_days(): """ Test deletion methods """ # creating tests objects first_ami = AMI() first_ami.name = "test-ami" first_ami.id = 'ami-28c2b348' first_ami.creation_date = "2017-11-04T01:35:31.000Z" second_ami = AMI() second_ami.name = "test-ami" second_ami.id = 'ami-28c2b349' second_ami.creation_date = "2017-11-04T01:35:31.000Z" # constructing dicts amis_dict = dict() amis_dict[first_ami.id] = first_ami amis_dict[second_ami.id] = second_ami parser = parse_args( [ '--keep-previous', '0', '--ami-min-days', '1', '--mapping-key', 'name', '--mapping-values', 'test-ami'] ) app = App(parser) # testing filter candidates = app.fetch_candidates(amis_dict) candidates_tobedeleted = app.prepare_candidates(candidates) assert len(candidates) == 2 assert len(candidates_tobedeleted) == 2 parser = parse_args( [ '--keep-previous', '0', '--ami-min-days', '10000', '--mapping-key', 'name', '--mapping-values', 'test-ami'] ) app = App(parser) candidates_tobedeleted2 = app.prepare_candidates(candidates) assert len(candidates) == 2 assert len(candidates_tobedeleted2) == 0
def test_print_report(): assert Printer.print_report({}) is None with open("tests/mocks/ami.json") as mock_file: json_to_parse = json.load(mock_file) ami = AMI.object_with_json(json_to_parse) candidates = {'test': [ami]} assert Printer.print_report(candidates) is None assert Printer.print_report(candidates, full_report=True) is None
def test_get_ami_from_json(): with open("tests/mocks/ami.json") as mock_file: json_to_parse = json.load(mock_file) ami = AMI.object_with_json(json_to_parse) assert ami.id == "ami-02197662" assert ami.virtualization_type == "hvm" assert ami.name == "custom-debian-201511040131" assert repr(ami) == "AMI: ami-02197662 2015-11-04T01:35:31.000Z" assert ami.tags[0].value is not None assert ami.tags[0].value is not None assert len(ami.tags) == 2 assert len(ami.block_device_mappings) == 2
def test_fetch_candidates(): # creating tests objects first_ami = AMI() first_ami.id = 'ami-28c2b348' first_ami.creation_date = datetime.now() first_instance = AWSEC2Instance() first_instance.id = 'i-9f9f6a2a' first_instance.name = "first-instance" first_instance.image_id = first_ami.id first_instance.launch_time = datetime.now() second_ami = AMI() second_ami.id = 'unused-ami' second_ami.creation_date = datetime.now() second_instance = AWSEC2Instance() second_instance.id = 'i-9f9f6a2b' second_instance.name = "second-instance" second_instance.image_id = first_ami.id second_instance.launch_time = datetime.now() # constructing dicts amis_dict = dict() amis_dict[first_ami.id] = first_ami amis_dict[second_ami.id] = second_ami instances_dict = dict() instances_dict[first_instance.image_id] = instances_dict instances_dict[second_instance.image_id] = second_instance # testing filter unused_ami_dict = App(parse_args([])).fetch_candidates( amis_dict, list(instances_dict) ) assert len(unused_ami_dict) == 1 assert amis_dict.get('unused-ami') is not None
def test_map_with_tags(): # tags stack_tag = AWSTag() stack_tag.key = "stack" stack_tag.value = "web-server" env_tag = AWSTag() env_tag.key = "env" env_tag.value = "prod" # creating tests objects # prod and web-server first_ami = AMI() first_ami.id = 'ami-28c2b348' first_ami.name = "ubuntu-20160102" first_ami.tags.append(stack_tag) first_ami.tags.append(env_tag) first_ami.creation_date = datetime.now() # just prod second_ami = AMI() second_ami.id = 'ami-28c2b349' second_ami.name = "ubuntu-20160103" second_ami.tags.append(env_tag) second_ami.creation_date = datetime.now() # prod and web-server third_ami = AMI() third_ami.id = 'ami-28c2b350' third_ami.name = "debian-20160104" third_ami.tags.append(stack_tag) third_ami.tags.append(env_tag) third_ami.creation_date = datetime.now() # creating amis to drop dict candidates = [first_ami, second_ami, third_ami] # grouping strategy grouping_strategy = {"key": "tags", "values": ["stack", "env"]} grouped_amis = AMICleaner().map_candidates(candidates, grouping_strategy) assert grouped_amis is not None assert len(grouped_amis.get("prod")) == 1 assert len(grouped_amis.get("prod.web-server")) == 2
def test_map_with_names(): # creating tests objects first_ami = AMI() first_ami.id = 'ami-28c2b348' first_ami.name = "ubuntu-20160102" first_ami.creation_date = datetime.now() second_ami = AMI() second_ami.id = 'ami-28c2b349' second_ami.name = "ubuntu-20160103" second_ami.creation_date = datetime.now() third_ami = AMI() third_ami.id = 'ami-28c2b350' third_ami.name = "debian-20160104" third_ami.creation_date = datetime.now() # creating amis to drop dict candidates = [first_ami, second_ami, third_ami] # grouping strategy grouping_strategy = {"key": "name", "values": ["ubuntu", "debian"]} grouped_amis = AMICleaner().map_candidates(candidates, grouping_strategy) assert grouped_amis is not None assert len(grouped_amis.get('ubuntu')) == 2 assert len(grouped_amis.get('debian')) == 1
def test_reduce(): # creating tests objects first_ami = AMI() first_ami.id = 'ami-28c2b348' first_ami.name = "ubuntu-20160102" first_ami.creation_date = datetime(2016, 1, 10) # just prod second_ami = AMI() second_ami.id = 'ami-28c2b349' second_ami.name = "ubuntu-20160103" second_ami.creation_date = datetime(2016, 1, 11) # prod and web-server third_ami = AMI() third_ami.id = 'ami-28c2b350' third_ami.name = "debian-20160104" third_ami.creation_date = datetime(2016, 1, 12) # keep 2 recent amis candidates = [second_ami, third_ami, first_ami] rotation_number = 2 cleaner = AMICleaner() left = cleaner.reduce_candidates(candidates, rotation_number) assert len(left) == 1 assert left[0].id == first_ami.id # keep 1 recent ami rotation_number = 1 left = cleaner.reduce_candidates(candidates, rotation_number) assert len(left) == 2 assert left[0].id == second_ami.id # keep 5 recent amis rotation_number = 5 left = cleaner.reduce_candidates(candidates, rotation_number) assert len(left) == 0
def test_reduce_without_rotation_number(): # creating tests objects first_ami = AMI() first_ami.id = 'ami-28c2b348' first_ami.name = "ubuntu-20160102" first_ami.creation_date = datetime(2016, 1, 10) # just prod second_ami = AMI() second_ami.id = 'ami-28c2b349' second_ami.name = "ubuntu-20160103" second_ami.creation_date = datetime(2016, 1, 11) # prod and web-server third_ami = AMI() third_ami.id = 'ami-28c2b350' third_ami.name = "debian-20160104" third_ami.creation_date = datetime(2016, 1, 12) # creating amis to drop dict candidates = [second_ami, third_ami, first_ami] assert AMICleaner().reduce_candidates(candidates) == candidates
def test_models_to_tring(): assert str(AMI()) is not None assert str(AWSBlockDevice()) is not None assert str(AWSEC2Instance()) is not None assert str(AWSTag()) is not None
def test_get_ami_from_none(): ami = AMI.object_with_json(None) assert ami is None