def test_context_get_current_template_with_client_error(self): # GIVEN} client = _get_cf_stubbed_client_with_error( method='get_template', service_error_code='ValidationError') context = _set_up_context(mock_client=client) stack_context = StackContext(context) # WHEN with self.assertRaises(HandledError): stack_context.get_current_template(stack_id=TEST_STACK_ID)
def test_context_get_current_template(self): # GIVEN expected_params = {'StackName': TEST_STACK_ID} expected_body = str({'AWSTemplateFormatVersion': '2010-09-09'}) response = {"TemplateBody": expected_body} client = _get_cf_stubbed_client(method='get_template', response=response, expected_params=expected_params) context = _set_up_context(mock_client=client) stack_context = StackContext(context) # WHEN response = stack_context.get_current_template(stack_id=TEST_STACK_ID) # THEN self.assertEqual(response, expected_body)