Yasuke/moto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added region parameter to boto3 calls

Browse Source
This commit is contained in:
Theodore Wong 2018-05-31 16:31:04 -07:00
parent da8bd545bf
commit 76c69c0dc5
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_cloudformation/test_import_value.py
View File

@ -11,6 +11,8 @@ from botocore.exceptions import ClientError
# Package modules # Package modules
from moto import mock_cloudformation from moto import mock_cloudformation
AWS_REGION = 'us-west-1'
SG_STACK_NAME = 'simple-sg-stack' SG_STACK_NAME = 'simple-sg-stack'
SG_TEMPLATE = """ SG_TEMPLATE = """
AWSTemplateFormatVersion: 2010-09-09 AWSTemplateFormatVersion: 2010-09-09
@ -63,7 +65,7 @@ class TestSimpleInstance(unittest.TestCase):
def test_simple_instance(self): def test_simple_instance(self):
"""Test that we can create a simple CloudFormation stack that imports values from an existing CloudFormation stack""" """Test that we can create a simple CloudFormation stack that imports values from an existing CloudFormation stack"""
with mock_cloudformation(): with mock_cloudformation():
client = boto3.client('cloudformation') client = boto3.client('cloudformation', region_name=AWS_REGION)
client.create_stack(StackName=SG_STACK_NAME, TemplateBody=SG_TEMPLATE) client.create_stack(StackName=SG_STACK_NAME, TemplateBody=SG_TEMPLATE)
response = client.create_stack(StackName=EC2_STACK_NAME, TemplateBody=EC2_TEMPLATE) response = client.create_stack(StackName=EC2_STACK_NAME, TemplateBody=EC2_TEMPLATE)
self.assertIn('StackId', response) self.assertIn('StackId', response)
@ -77,7 +79,7 @@ class TestSimpleInstance(unittest.TestCase):
def test_simple_instance_missing_export(self): def test_simple_instance_missing_export(self):
"""Test that we get an exception if a CloudFormation stack tries to imports a non-existent export value""" """Test that we get an exception if a CloudFormation stack tries to imports a non-existent export value"""
with mock_cloudformation(): with mock_cloudformation():
client = boto3.client('cloudformation') client = boto3.client('cloudformation', region_name=AWS_REGION)
with self.assertRaises(ClientError) as e: with self.assertRaises(ClientError) as e:
client.create_stack(StackName=EC2_STACK_NAME, TemplateBody=EC2_TEMPLATE) client.create_stack(StackName=EC2_STACK_NAME, TemplateBody=EC2_TEMPLATE)
self.assertIn('Error', e.exception.response) self.assertIn('Error', e.exception.response)