def test_get_certificate(self, cfngin_context, patch_time): """Test get_certificate.""" # setup context cfngin_context.add_stubber('acm', 'us-east-1') cfngin_context.add_stubber('route53', 'us-east-1') cfngin_context.config.namespace = 'test' provider = MagicMock(cloudformation=boto3.client('cloudformation')) cfn_stubber = Stubber(provider.cloudformation) cert = Certificate(context=cfngin_context, provider=provider, domain='example.com', hosted_zone_id='test') expected_request = { 'StackName': 'test-example-com', 'LogicalResourceId': 'Certificate' } cfn_stubber.add_response('describe_stack_resources', {'StackResources': []}, expected_request) cfn_stubber.add_response('describe_stack_resources', {'StackResources': [gen_stack_resource()]}, expected_request) cfn_stubber.add_response('describe_stack_resources', { 'StackResources': [gen_stack_resource(PhysicalResourceId='success')] }, expected_request) with cfn_stubber: assert cert.get_certificate() == 'success' cfn_stubber.assert_no_pending_responses()
def test_get_certificate(self, cfngin_context: MockCFNginContext, patch_time: None) -> None: """Test get_certificate.""" # setup context cfngin_context.add_stubber("acm", "us-east-1") cfngin_context.add_stubber("route53", "us-east-1") cfngin_context.config.namespace = "test" provider = MagicMock(cloudformation=boto3.client("cloudformation")) cfn_stubber = Stubber(provider.cloudformation) cert = Certificate( context=cfngin_context, provider=provider, domain="example.com", hosted_zone_id="test", ) expected_request = { "StackName": "test-example-com", "LogicalResourceId": "Certificate", } cfn_stubber.add_response("describe_stack_resources", {"StackResources": []}, expected_request) cfn_stubber.add_response( "describe_stack_resources", {"StackResources": [gen_stack_resource()]}, expected_request, ) cfn_stubber.add_response( "describe_stack_resources", { "StackResources": [gen_stack_resource(PhysicalResourceId="success")] }, expected_request, ) with cfn_stubber: assert cert.get_certificate() == "success" cfn_stubber.assert_no_pending_responses()