def test_template_write(self): mock_temp = Mock("taskcat._cfn.template.Template", autospec=True)() mock_temp.raw_template = "some-template-data" au_template = Template(underlying=mock_temp) au_template._ls = "some-other-data" au_template.write() mock_temp.write.assert_called_once()
def test_template_set_codename_ami_no_region(self): tc_template = self.create_ephemeral_template_object()["taskcat-json"] au_template = Template(underlying=tc_template, regions_with_creds=[]) actual = au_template.set_codename_ami( "AMZNLINUXHVM", "us-west-1", "slkdfjskldfj" ) self.assertFalse(actual)
def test_template_write(self): mock_temp = unittest.mock.Mock( "taskcat._cfn.template.Template", autospec=True )() mock_temp.raw_template = "some-template-data" au_template = Template(underlying=mock_temp, regions_with_creds=["us-east-1"]) au_template._ls = "some-other-data" au_template.write() mock_temp.write.assert_called_once()
def test_build_codenames(self): example_tc_template = self.create_ephemeral_template_object() au_template = Template(example_tc_template["taskcat-json"], ["us-east-1"]) example_config_obj = AUConfig example_config_obj.raw_dict = { "global": { "AMIs": { "AMZNLINUXHVM": { "name": "amzn-ami-hvm-????.??.?.*-x86_64-gp2", "owner-alias": "amazon", } } } } expected = [ RegionalCodename( region="us-east-1", cn="AMZNLINUXHVM", filters=[ EC2FilterValue("name", ["amzn-ami-hvm-????.??.?.*-x86_64-gp2"]), EC2FilterValue("owner-alias", ["amazon"]), EC2FilterValue("state", ["available"]), ], ) ] actual = build_codenames(au_template, example_config_obj) _mocked_dt = datetime.now() for r1 in expected: r1._creation_dt = _mocked_dt for r2 in actual: r2._creation_dt = _mocked_dt self.assertEqual(expected, actual)
def test_template_set_codename_ami(self): tc_template = self.create_ephemeral_template_object()["taskcat-json"] au_template = Template(underlying=tc_template) actual = au_template.set_codename_ami("AMZNLINUXHVM", "us-east-1", "slkdfjskldfj") self.assertTrue(actual)
def test_invalid_region_exception(self): example_tc_template = self.create_ephemeral_template_object() au_template = Template(example_tc_template["taskcat-json"], ["us-east-1"]) self.assertEqual(au_template.regions_without_creds, {"us-east-2"}) self.assertEqual(au_template.region_names, {"us-east-1"})