1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
# Copyright 2014 Amazon.com, Inc. or its affiliates. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License"). You
# may not use this file except in compliance with the License. A copy of
# the License is located at
#
# http://aws.amazon.com/apache2.0e
#
# or in the "license" file accompanying this file. This file is
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
# ANY KIND, either express or implied. See the License for the specific
# language governing permissions and limitations under the License.
import os
import sys
import tempfile
from io import StringIO
from awscli.customizations.cloudformation.artifact_exporter import Template
from awscli.customizations.cloudformation.package import PackageCommand
from awscli.customizations.cloudformation.yamlhelper import yaml_dump
from awscli.testutils import mock
from tests.unit.customizations.cloudformation import BaseYAMLTest
class FakeArgs:
def __init__(self, **kwargs):
self.__dict__.update(kwargs)
def __contains__(self, key):
return key in self.__dict__
def get_example_template():
return {"Parameters": {"Key1": "Value1"}, "Resources": {"Resource1": {}}}
class TestPackageCommand(BaseYAMLTest):
def setUp(self):
super(TestPackageCommand, self).setUp()
self.session = mock.Mock()
self.session.get_scoped_config.return_value = {}
self.parsed_args = FakeArgs(
template_file='./foo',
s3_bucket="s3bucket",
s3_prefix="s3prefix",
kms_key_id="kmskeyid",
output_template_file="./oputput",
use_json=False,
force_upload=False,
metadata=None,
)
self.parsed_globals = FakeArgs(
region="us-east-1", endpoint_url=None, verify_ssl=None
)
self.package_command = PackageCommand(self.session)
@mock.patch("awscli.customizations.cloudformation.package.yaml_dump")
def test_main(self, mock_yaml_dump):
exported_template_str = "hello"
self.package_command.write_output = mock.Mock()
self.package_command._export = mock.Mock()
mock_yaml_dump.return_value = exported_template_str
# Create a temporary file and make this my template
with tempfile.NamedTemporaryFile() as handle:
for use_json in (False, True):
filename = handle.name
self.parsed_args.template_file = filename
self.parsed_args.use_json = use_json
rc = self.package_command._run_main(
self.parsed_args, self.parsed_globals
)
self.assertEqual(rc, 0)
self.package_command._export.assert_called_once_with(
filename, use_json
)
self.package_command.write_output.assert_called_once_with(
self.parsed_args.output_template_file, mock.ANY
)
self.package_command._export.reset_mock()
self.package_command.write_output.reset_mock()
def test_main_error(self):
self.package_command._export = mock.Mock()
self.package_command._export.side_effect = RuntimeError()
# Create a temporary file and make this my template
with tempfile.NamedTemporaryFile() as handle:
filename = handle.name
self.parsed_args.template_file = filename
with self.assertRaises(RuntimeError):
self.package_command._run_main(
self.parsed_args, self.parsed_globals
)
@mock.patch("awscli.customizations.cloudformation.package.sys.stdout")
def test_write_output_to_stdout(self, stdoutmock):
data = "some data"
filename = None
self.package_command.write_output(filename, data)
stdoutmock.write.assert_called_once_with(data)
|
