def create_deletion_deployer(client, ui):
# type: (TypedAWSClient, UI) -> Deployer
return Deployer(
application_builder=ApplicationGraphBuilder(),
deps_builder=DependencyBuilder(),
build_stage=BuildStage(steps=[]),
plan_stage=NoopPlanner(),
sweeper=ResourceSweeper(),
executor=Executor(client, ui),
recorder=ResultsRecorder(osutils=OSUtils()),
)
def create_default_deployer(session, config, ui):
# type: (Session, Config, UI) -> Deployer
client = TypedAWSClient(session)
osutils = OSUtils()
return Deployer(
application_builder=ApplicationGraphBuilder(),
deps_builder=DependencyBuilder(),
build_stage=create_build_stage(
osutils, UI(), TemplatedSwaggerGenerator(),
),
plan_stage=PlanStage(
osutils=osutils, remote_state=RemoteState(
client, config.deployed_resources(config.chalice_stage)),
),
sweeper=ResourceSweeper(),
executor=Executor(client, ui),
recorder=ResultsRecorder(osutils=osutils),
)
def setup_method(self):
self.mock_client = mock.Mock(spec=TypedAWSClient)
self.ui = mock.Mock(spec=UI)
self.executor = Executor(self.mock_client, self.ui)
class TestExecutor(object):
def setup_method(self):
self.mock_client = mock.Mock(spec=TypedAWSClient)
self.ui = mock.Mock(spec=UI)
self.executor = Executor(self.mock_client, self.ui)
def execute(self, instructions, messages=None):
if messages is None:
messages = {}
self.executor.execute(models.Plan(instructions, messages))
def test_can_invoke_api_call_with_no_output(self):
params = {
'name': 'foo',
'trust_policy': {
'trust': 'policy'
},
'policy': {
'iam': 'policy'
}
}
call = APICall('create_role', params)
self.execute([call])
self.mock_client.create_role.assert_called_with(**params)
def test_can_store_api_result(self):
params = {
'name': 'foo',
'trust_policy': {
'trust': 'policy'
},
'policy': {
'iam': 'policy'
}
}
apicall = APICall('create_role', params, output_var='my_variable_name')
self.mock_client.create_role.return_value = 'myrole:arn'
self.execute([apicall])
assert self.executor.variables['my_variable_name'] == 'myrole:arn'
def test_can_store_multiple_value(self):
instruction = models.StoreMultipleValue(name='list_data',
value=['first_elem'])
self.execute([instruction])
assert self.executor.variables['list_data'] == ['first_elem']
instruction = models.StoreMultipleValue(name='list_data',
value=['second_elem'])
self.execute([instruction])
assert self.executor.variables['list_data'] == [
'first_elem', 'second_elem'
]
def test_can_reference_stored_results_in_api_calls(self):
params = {
'name': Variable('role_name'),
'trust_policy': {
'trust': 'policy'
},
'policy': {
'iam': 'policy'
}
}
call = APICall('create_role', params)
self.mock_client.create_role.return_value = 'myrole:arn'
self.executor.variables['role_name'] = 'myrole-name'
self.execute([call])
self.mock_client.create_role.assert_called_with(
name='myrole-name',
trust_policy={'trust': 'policy'},
policy={'iam': 'policy'},
)
def test_can_return_created_resources(self):
params = {}
call = APICall('create_function', params, output_var='myfunction_arn')
self.mock_client.create_function.return_value = 'function:arn'
record_instruction = RecordResourceVariable(
resource_type='lambda_function',
resource_name='myfunction',
name='myfunction_arn',
variable_name='myfunction_arn',
)
self.execute([call, record_instruction])
assert self.executor.resource_values == [{
'name':
'myfunction',
'myfunction_arn':
'function:arn',
'resource_type':
'lambda_function',
}]
def test_can_reference_varname(self):
self.mock_client.create_function.return_value = 'function:arn'
self.execute([
APICall('create_function', {}, output_var='myvarname'),
RecordResourceVariable(
resource_type='lambda_function',
resource_name='myfunction',
name='myfunction_arn',
variable_name='myvarname',
),
])
assert self.executor.resource_values == [{
'name':
'myfunction',
'resource_type':
'lambda_function',
'myfunction_arn':
'function:arn',
}]
def test_can_record_value_directly(self):
self.execute([
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='myfunction_arn',
value='arn:foo',
)
])
assert self.executor.resource_values == [{
'name': 'myfunction',
'resource_type': 'lambda_function',
'myfunction_arn': 'arn:foo',
}]
def test_can_aggregate_multiple_resource_values(self):
self.execute([
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key1',
value='value1',
),
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key2',
value='value2',
)
])
assert self.executor.resource_values == [{
'name': 'myfunction',
'resource_type': 'lambda_function',
'key1': 'value1',
'key2': 'value2',
}]
def test_new_keys_override_old_keys(self):
self.execute([
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key1',
value='OLD',
),
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key1',
value='NEW',
)
])
assert self.executor.resource_values == [{
'name': 'myfunction',
'resource_type': 'lambda_function',
'key1': 'NEW',
}]
def test_validates_no_unresolved_deploy_vars(self):
params = {'zip_contents': models.Placeholder.BUILD_STAGE}
call = APICall('create_function', params)
self.mock_client.create_function.return_value = 'function:arn'
# We should raise an exception because a param has
# a models.Placeholder.BUILD_STAGE value which should have
# been handled in an earlier stage.
with pytest.raises(UnresolvedValueError):
self.execute([call])
def test_can_jp_search(self):
self.execute([
StoreValue(name='searchval', value={'foo': {
'bar': 'baz'
}}),
JPSearch('foo.bar', input_var='searchval', output_var='result'),
])
assert self.executor.variables['result'] == 'baz'
def test_can_copy_variable(self):
self.execute([
StoreValue(name='foo', value='bar'),
CopyVariable(from_var='foo', to_var='baz'),
])
assert self.executor.variables['baz'] == 'bar'
def test_can_call_builtin_function(self):
self.execute([
StoreValue(name='my_arn',
value='arn:aws:lambda:us-west-2:123:function:name'),
BuiltinFunction(
function_name='parse_arn',
args=[Variable('my_arn')],
output_var='result',
)
])
assert self.executor.variables['result'] == {
'account_id': '123',
'region': 'us-west-2',
'service': 'lambda'
}
def test_errors_out_on_unknown_function(self):
with pytest.raises(ValueError):
self.execute([
BuiltinFunction(
function_name='unknown_foo',
args=[],
output_var=None,
)
])
def test_can_print_ui_messages(self):
params = {
'name': 'foo',
'trust_policy': {
'trust': 'policy'
},
'policy': {
'iam': 'policy'
}
}
call = APICall('create_role', params)
messages = {id(call): 'Creating role'}
self.execute([call], messages)
self.mock_client.create_role.assert_called_with(**params)
self.ui.write.assert_called_with('Creating role')
def test_error_out_on_unknown_instruction(self):
class CustomInstruction(Instruction):
pass
with pytest.raises(RuntimeError):
self.execute([CustomInstruction()])
def setup_method(self):
self.mock_client = mock.Mock(spec=TypedAWSClient)
self.mock_client.endpoint_dns_suffix.return_value = 'amazonaws.com'
self.ui = mock.Mock(spec=UI)
self.executor = Executor(self.mock_client, self.ui)
def setup_method(self):
self.mock_client = mock.Mock(spec=TypedAWSClient)
self.ui = mock.Mock(spec=UI)
self.executor = Executor(self.mock_client, self.ui)
class TestExecutor(object):
def setup_method(self):
self.mock_client = mock.Mock(spec=TypedAWSClient)
self.ui = mock.Mock(spec=UI)
self.executor = Executor(self.mock_client, self.ui)
def execute(self, instructions, messages=None):
if messages is None:
messages = {}
self.executor.execute(models.Plan(instructions, messages))
def test_can_invoke_api_call_with_no_output(self):
params = {'name': 'foo', 'trust_policy': {'trust': 'policy'},
'policy': {'iam': 'policy'}}
call = APICall('create_role', params)
self.execute([call])
self.mock_client.create_role.assert_called_with(**params)
def test_can_store_api_result(self):
params = {'name': 'foo', 'trust_policy': {'trust': 'policy'},
'policy': {'iam': 'policy'}}
apicall = APICall('create_role', params, output_var='my_variable_name')
self.mock_client.create_role.return_value = 'myrole:arn'
self.execute([apicall])
assert self.executor.variables['my_variable_name'] == 'myrole:arn'
def test_can_reference_stored_results_in_api_calls(self):
params = {
'name': Variable('role_name'),
'trust_policy': {'trust': 'policy'},
'policy': {'iam': 'policy'}
}
call = APICall('create_role', params)
self.mock_client.create_role.return_value = 'myrole:arn'
self.executor.variables['role_name'] = 'myrole-name'
self.execute([call])
self.mock_client.create_role.assert_called_with(
name='myrole-name',
trust_policy={'trust': 'policy'},
policy={'iam': 'policy'},
)
def test_can_return_created_resources(self):
params = {}
call = APICall('create_function', params,
output_var='myfunction_arn')
self.mock_client.create_function.return_value = 'function:arn'
record_instruction = RecordResourceVariable(
resource_type='lambda_function',
resource_name='myfunction',
name='myfunction_arn',
variable_name='myfunction_arn',
)
self.execute([call, record_instruction])
assert self.executor.resource_values == [{
'name': 'myfunction',
'myfunction_arn': 'function:arn',
'resource_type': 'lambda_function',
}]
def test_can_reference_varname(self):
self.mock_client.create_function.return_value = 'function:arn'
self.execute([
APICall('create_function', {}, output_var='myvarname'),
RecordResourceVariable(
resource_type='lambda_function',
resource_name='myfunction',
name='myfunction_arn',
variable_name='myvarname',
),
])
assert self.executor.resource_values == [{
'name': 'myfunction',
'resource_type': 'lambda_function',
'myfunction_arn': 'function:arn',
}]
def test_can_record_value_directly(self):
self.execute([
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='myfunction_arn',
value='arn:foo',
)
])
assert self.executor.resource_values == [{
'name': 'myfunction',
'resource_type': 'lambda_function',
'myfunction_arn': 'arn:foo',
}]
def test_can_aggregate_multiple_resource_values(self):
self.execute([
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key1',
value='value1',
),
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key2',
value='value2',
)
])
assert self.executor.resource_values == [{
'name': 'myfunction',
'resource_type': 'lambda_function',
'key1': 'value1',
'key2': 'value2',
}]
def test_new_keys_override_old_keys(self):
self.execute([
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key1',
value='OLD',
),
RecordResourceValue(
resource_type='lambda_function',
resource_name='myfunction',
name='key1',
value='NEW',
)
])
assert self.executor.resource_values == [{
'name': 'myfunction',
'resource_type': 'lambda_function',
'key1': 'NEW',
}]
def test_validates_no_unresolved_deploy_vars(self):
params = {'zip_contents': models.Placeholder.BUILD_STAGE}
call = APICall('create_function', params)
self.mock_client.create_function.return_value = 'function:arn'
# We should raise an exception because a param has
# a models.Placeholder.BUILD_STAGE value which should have
# been handled in an earlier stage.
with pytest.raises(UnresolvedValueError):
self.execute([call])
def test_can_jp_search(self):
self.execute([
StoreValue(name='searchval', value={'foo': {'bar': 'baz'}}),
JPSearch('foo.bar', input_var='searchval', output_var='result'),
])
assert self.executor.variables['result'] == 'baz'
def test_can_call_builtin_function(self):
self.execute([
StoreValue(
name='my_arn',
value='arn:aws:lambda:us-west-2:123:function:name'),
BuiltinFunction(
function_name='parse_arn',
args=[Variable('my_arn')],
output_var='result',
)
])
assert self.executor.variables['result'] == {
'account_id': '123',
'region': 'us-west-2',
'service': 'lambda'
}
def test_errors_out_on_unknown_function(self):
with pytest.raises(ValueError):
self.execute([
BuiltinFunction(
function_name='unknown_foo',
args=[],
output_var=None,
)
])
def test_can_print_ui_messages(self):
params = {'name': 'foo', 'trust_policy': {'trust': 'policy'},
'policy': {'iam': 'policy'}}
call = APICall('create_role', params)
messages = {id(call): 'Creating role'}
self.execute([call], messages)
self.mock_client.create_role.assert_called_with(**params)
self.ui.write.assert_called_with('Creating role')
def test_error_out_on_unknown_instruction(self):
class CustomInstruction(Instruction):
pass
with pytest.raises(RuntimeError):
self.execute([CustomInstruction()])
