def test_s3_get_file_raises_exception_on_error(self, s3_mock):
s3_mock.return_value.get_contents_from_url.side_effect = CfnSphereBotoError
with self.assertRaises(CfnSphereException):
FileLoader._s3_get_file("s3://foo/foo.yml")
def test_s3_get_file_parses_yaml_for_yml_suffix(self, s3_mock, json_mock, yaml_mock):
s3_mock.return_value.get_contents_from_url.return_value = "{}"
FileLoader._s3_get_file('s3://foo/baa.yml')
json_mock.assert_not_called()
yaml_mock.assert_called_once_with("{}")
def test_s3_get_file__raises_exception_for_unknown_suffix(self, s3_mock, json_mock, yaml_mock):
s3_mock.return_value.get_contents_from_url.return_value = "{}"
with self.assertRaises(TemplateErrorException):
FileLoader._s3_get_file('s3://foo/baa.foo')
def test_s3_get_file_raises_exception_on_error(self, s3_mock):
s3_mock.return_value.get_contents_from_url.side_effect = CfnSphereBotoError
with self.assertRaises(CfnSphereException):
FileLoader._s3_get_file("s3://foo/foo.yml")
