|
1 | 1 | from __future__ import print_function |
2 | 2 | from __future__ import division |
3 | 3 | from __future__ import absolute_import |
| 4 | +from future import standard_library |
| 5 | +standard_library.install_aliases() |
| 6 | + |
4 | 7 | import unittest |
5 | 8 |
|
6 | | -from botocore.stub import Stubber, ANY |
| 9 | +import mock |
| 10 | + |
7 | 11 | import botocore.exceptions |
| 12 | +from botocore.stub import Stubber, ANY |
8 | 13 |
|
9 | 14 | from stacker.actions.base import ( |
10 | 15 | BaseAction |
11 | 16 | ) |
12 | | - |
13 | | -from stacker.providers.aws.default import Provider |
14 | 17 | from stacker.blueprints.base import Blueprint |
| 18 | +from stacker.providers.aws.default import Provider |
15 | 19 | from stacker.session_cache import get_session |
16 | 20 |
|
17 | 21 | from stacker.tests.factories import ( |
@@ -121,30 +125,27 @@ def test_ensure_cfn_forbidden(self): |
121 | 125 | action.ensure_cfn_bucket() |
122 | 126 |
|
123 | 127 | def test_stack_template_url(self): |
124 | | - test_cases = ( |
125 | | - ("us-east-1", "s3.amazonaws.com"), |
126 | | - ("us-west-1", "s3.us-west-1.amazonaws.com"), |
127 | | - ("eu-west-1", "s3.eu-west-1.amazonaws.com"), |
128 | | - ("sa-east-1", "s3.sa-east-1.amazonaws.com"), |
129 | | - ) |
130 | 128 | context = mock_context("mynamespace") |
131 | 129 | blueprint = TestBlueprint(name="myblueprint", context=context) |
132 | 130 |
|
133 | | - for region, endpoint in test_cases: |
134 | | - session = get_session(region) |
135 | | - provider = Provider(session) |
136 | | - action = BaseAction( |
137 | | - context=context, |
138 | | - provider_builder=MockProviderBuilder(provider, region=region) |
139 | | - ) |
| 131 | + region = "us-east-1" |
| 132 | + endpoint = "https://example.com" |
| 133 | + session = get_session(region) |
| 134 | + provider = Provider(session) |
| 135 | + action = BaseAction( |
| 136 | + context=context, |
| 137 | + provider_builder=MockProviderBuilder(provider, region=region) |
| 138 | + ) |
| 139 | + |
| 140 | + with mock.patch('stacker.actions.base.get_s3_endpoint', autospec=True, |
| 141 | + return_value=endpoint): |
140 | 142 | self.assertEqual( |
141 | 143 | action.stack_template_url(blueprint), |
142 | | - "https://%s/%s/stack_templates/%s/%s-%s.json" % ( |
| 144 | + "%s/%s/stack_templates/%s/%s-%s.json" % ( |
143 | 145 | endpoint, |
144 | 146 | "stacker-mynamespace", |
145 | 147 | "mynamespace-myblueprint", |
146 | 148 | "myblueprint", |
147 | 149 | MOCK_VERSION |
148 | | - |
149 | 150 | ) |
150 | 151 | ) |
0 commit comments