def __init__(self, *args, **kwargs):
super(test_SG, self).__init__(*args, **kwargs)
self.sg_parser = SG_Parser()
self.client = Client_Mock()
sg_data = self.client.describe_security_groups()
self.sg_parser.set_data(sg_data)
def test_setNoVpcWithMultiplesVpcs(self):
clientBoto3 = Client_Mock().setMultipleVpcs()
sg_test_name = 'my-sg-name'
self.sg_client.set_client(clientBoto3)
self.sg_client.set_group_name(sg_test_name)
with self.assertRaises(Exception):
self.sg_client.create_default_sg()
def test_getDefaultVpc(self):
clientBoto3 = Client_Mock()
sg_test_name = 'my-sg-name'
self.sg_client.set_client(clientBoto3).set_group_name(sg_test_name)
self.sg_client.create_default_sg()
expectedVpc = "vpc-abcd1234"
choosedVpc = self.sg_client.getVpc()
self.assertEqual(expectedVpc, choosedVpc)
def test_vpcMultiplesGivenInvalidOne(self):
clientBoto3 = Client_Mock().setMultipleVpcs()
sg_test_name = 'my-sg-name'
self.sg_client.set_client(clientBoto3)
self.sg_client.set_group_name(sg_test_name)
self.sg_client.set_vpc("vpc-4919cd1632d1d03b6")
with self.assertRaises(Exception):
self.sg_client.create_default_sg()
def test_is_rds_instance_name_exists_false(self):
client = Client_Mock()
rds_data = client.describe_db_instances()
rds_parser = RDS_Parser()
rds_parser.set_data(rds_data)
rds_name = "non_existent"
exists = self.sg_parser.is_rds_instance_name_exists(rds_name, client)
self.assertFalse(exists)
def test_vpcMultiplesGiven(self):
clientBoto3 = Client_Mock().setMultipleVpcs()
sg_test_name = 'my-sg-name'
self.sg_client\
.set_client(clientBoto3)\
.set_group_name(sg_test_name)\
.set_vpc("vpc-a30ff249b44e63bfe")
self.sg_client.create_default_sg()
expectedVpc = "vpc-a30ff249b44e63bfe"
choosedVpc = self.sg_client.getVpc()
self.assertEqual(expectedVpc, choosedVpc)
def test_checkVpcAmountsSingle(self):
clientBoto3 = Client_Mock()
sg_test_name = 'my-sg-name'
self.sg_client.set_client(clientBoto3).\
set_group_name(sg_test_name)
self.assertFalse(self.sg_client.is_multiples_vpcs())
def test_set_group_name(self):
test_name = 'my-sg-name'
self.sg_client.set_client(Client_Mock()).set_group_name(test_name)
self.assertEqual(test_name, self.sg_client.getGroupName())
def test_fetch_request_right_format(self):
client = Client_Mock()
self.fetcher.set_client(client)
result_fetched = self.fetcher.get_sgs_data()
self.assertTrue(isinstance(result_fetched, dict))
def test_correct_data_fetched_for_regions_list(self):
client = Client_Mock()
regions = get_regions(client)
expected_list = ["sa-north-1", "ju-south-1", "mo-west-3"]
self.assertListEqual(expected_list, regions)
def test_get_sgs_data_by_id(self):
self.fetcher.set_client(Client_Mock())
sg_id = "sg-092498379087908"
result_fetched = self.fetcher.get_sgs_data_by_id(sg_id)
self.assertTrue(isinstance(result_fetched, dict))
def test_get_sgs_data_by_non_existing_name(self):
self.fetcher.set_client(Client_Mock())
with self.assertRaises(Exception):
self.fetcher.get_sgs_data_by_name("non existent")
def test_get_sgs_data_by_name(self):
self.fetcher.set_client(Client_Mock())
sg_name = "allow-http"
result_fetched = self.fetcher.get_sgs_data_by_name(sg_name)
self.assertTrue(isinstance(result_fetched, dict))
def test_fetch_request_right_format_for_regions(self):
self.fetcher.set_client(Client_Mock())
result_fetched = self.fetcher.get_all_regions_data()
self.assertTrue(isinstance(result_fetched, dict))
def test_list_vpcs(self):
clientBoto3 = Client_Mock().setMultipleVpcs()
self.sg_client.set_client(clientBoto3)
vpcs_list_names = self.sg_client.fetch_vpcs_list_names()
self.assertTrue("vpc-71d99e528f6bdc8d2" in vpcs_list_names)
self.assertTrue("vpc-a30ff249b44e63bfe" in vpcs_list_names)
def test_regions_list_correct_type(self):
client = Client_Mock()
regions = get_regions(client)
self.assertTrue(isinstance(regions, list))
def get_client(self) -> Client_Interface:
return Client_Mock()
def __init__(self, *args, **kwargs):
super(test_RDS_Parser, self).__init__(*args, **kwargs)
self.client = Client_Mock()
self.rds_parser = RDS_Parser()
rds_data = self.client.describe_db_instances()
self.rds_parser.set_data(rds_data)