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)
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()])