def test_get_available_azs(self): pg = ParamGen(**self.class_kwargs) acceptable_azs = ['us-east-1a','us-east-1b','us-east-1c','us-east-1d','us-east-1e','us-east-1f'] pg._boto_client = MockClientFactory() returned_azs = pg.get_available_azs(2) returned_az_list = returned_azs.split(',') test_criteria = [ # tuple (first_param, second_param, test_description) (len(returned_az_list), 2, "Verifying we return 2 AZs"), (len(set(returned_az_list)), 2, "Verifying we return 2 *unique* AZs") ] for first_param, second_param, test_desc in test_criteria: with self.subTest(test_desc): self.assertEqual(first_param, second_param)
def test_get_content(self): pg = ParamGen(**self.class_kwargs) pg._boto_client = MockClientFactory() self.assertEqual(pg.get_content(bucket='unit-test-bucket', object_key='unit-test-key'), 'unicorns')
def test_genaz_raises_taskcat_exception(self): pg = ParamGen(**self.class_kwargs) pg._boto_client = MockClientFactory(ec2_single_az=True) with self.assertRaises(TaskCatException): pg.get_available_azs(2)