Python AwsLambdaDeploymentOperator.describe Examples

Example #1

File: test_aws_lambda_deployment_operator.py Project: vedashree29296/BentoML

def test_aws_lambda_describe_success():
    def mock_cf_response(_self, op_name, _kwarg):
        if op_name == 'DescribeStacks':
            return {
                'Stacks': [{
                    'StackStatus':
                    'CREATE_COMPLETE',
                    'Outputs': [{
                        'OutputKey':
                        'EndpointUrl',
                        'OutputValue':
                        'https://somefakelink.amazonaws.com'
                        '/prod/predict',
                    }],
                }]
            }
        else:
            raise Exception(
                'This test does not handle operation {}'.format(op_name))

    yatai_service_mock = create_yatai_service_mock()
    test_deployment_pb = generate_lambda_deployment_pb()
    with patch(
            'bentoml.yatai.deployment.aws_lambda.operator.get_default_aws_region',
            MagicMock(return_value='mock_region'),
    ):
        with patch('botocore.client.BaseClient._make_api_call',
                   new=mock_cf_response):
            deployment_operator = AwsLambdaDeploymentOperator(
                yatai_service_mock)
            result_pb = deployment_operator.describe(test_deployment_pb)
            assert result_pb.status.status_code == status_pb2.Status.OK
            assert result_pb.state.state == DeploymentState.RUNNING

Example #2

File: test_aws_lambda_deployment_operator.py Project: subhayuroy/BentoML

def test_aws_lambda_describe_still_in_progress():
    def mock_cf_response(_self, op_name, _kwarg):
        if op_name == 'DescribeStacks':
            return {'Stacks': [{'StackStatus': 'CREATE_IN_PROGRESS'}]}
        else:
            raise Exception('This test does not handle operation {}'.format(op_name))

    yatai_service_mock = create_yatai_service_mock()
    test_deployment_pb = generate_lambda_deployment_pb()
    with patch(
        'bentoml.yatai.deployment.aws_lambda.operator.get_default_aws_region',
        MagicMock(return_value='mock_region'),
    ):
        with patch('botocore.client.BaseClient._make_api_call', new=mock_cf_response):
            deployment_operator = AwsLambdaDeploymentOperator(yatai_service_mock)
            result_pb = deployment_operator.describe(test_deployment_pb)
            assert result_pb.status.status_code == status_pb2.Status.OK
            assert result_pb.state.state == DeploymentState.PENDING