Python invoke_function示例

示例#1

文件： test_awslambda_actions.py 项目： chaostoolkit-incubator/chaostoolkit-aws

def test_aws_lambda_invoke_with_args(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    mock_payload = {"some": "response"}
    mock_response = {"Payload": MagicMock()}
    mock_response["Payload"].read.return_value = json.dumps(
        mock_payload).encode()
    client.invoke.return_value = mock_response
    lambda_function_name = "my-lambda-function"
    function_arguments = {"some": "argument"}
    invocation_type = "Event"
    client_context = {"some": "context"}
    qualifier = "$LATEST"

    invoke_function(
        function_name=lambda_function_name,
        function_arguments=function_arguments,
        invocation_type=invocation_type,
        client_context=client_context,
        qualifier=qualifier,
    )

    client.invoke.assert_called_with(
        FunctionName=lambda_function_name,
        InvocationType=invocation_type,
        LogType="None",
        Payload=json.dumps(function_arguments),
        ClientContext=b64encode(json.dumps(client_context).encode()).decode(),
        Qualifier=qualifier,
    )

示例#2

def test_aws_lambda_invoke_with_args(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    mock_payload = {'some': 'response'}
    mock_response = {'Payload': MagicMock()}
    mock_response['Payload'].read.return_value = json.dumps(
        mock_payload).encode()
    client.invoke.return_value = mock_response
    lambda_function_name = 'my-lambda-function'
    function_arguments = {'some': 'argument'}
    invocation_type = 'Event'
    client_context = {'some': 'context'}
    qualifier = '$LATEST'

    invoke_function(function_name=lambda_function_name,
                    function_arguments=function_arguments,
                    invocation_type=invocation_type,
                    client_context=client_context,
                    qualifier=qualifier)

    client.invoke.assert_called_with(
        FunctionName=lambda_function_name,
        InvocationType=invocation_type,
        LogType='None',
        Payload=json.dumps(function_arguments),
        ClientContext=b64encode(json.dumps(client_context).encode()).decode(),
        Qualifier=qualifier)

示例#3

def test_aws_lambda_invoke_failed(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    client.invoke.side_effect = Exception('Some Exception')
    lambda_function_name = 'my-lambda-function'
    with pytest.raises(FailedActivity):
        invoke_function(lambda_function_name)
    client.invoke.assert_called_with(FunctionName=lambda_function_name,
                                     InvocationType='RequestResponse',
                                     LogType='None')

示例#4

def test_aws_lambda_invoke_empty_response_payload(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    mock_response = {'Payload': MagicMock()}
    mock_response['Payload'].read.return_value = b''
    client.invoke.return_value = mock_response
    lambda_function_name = 'my-lambda-function'

    response = invoke_function(lambda_function_name, invocation_type='Event')

    assert response['Payload'] == ''
    client.invoke.assert_called_with(FunctionName=lambda_function_name,
                                     InvocationType='Event',
                                     LogType='None')

示例#5

文件： test_awslambda_actions.py 项目： chaostoolkit-incubator/chaostoolkit-aws

def test_aws_lambda_invoke_empty_response_payload(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    mock_response = {"Payload": MagicMock()}
    mock_response["Payload"].read.return_value = b""
    client.invoke.return_value = mock_response
    lambda_function_name = "my-lambda-function"

    response = invoke_function(lambda_function_name, invocation_type="Event")

    assert response["Payload"] == ""
    client.invoke.assert_called_with(FunctionName=lambda_function_name,
                                     InvocationType="Event",
                                     LogType="None")

示例#6

def test_aws_lambda_invoke_no_args(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    mock_payload = {'some': 'response'}
    mock_response = {'Payload': MagicMock()}
    mock_response['Payload'].read.return_value = json.dumps(
        mock_payload).encode()
    client.invoke.return_value = mock_response
    lambda_function_name = 'my-lambda-function'

    response = invoke_function(lambda_function_name)

    assert response['Payload'] == mock_payload
    client.invoke.assert_called_with(FunctionName=lambda_function_name,
                                     InvocationType='RequestResponse',
                                     LogType='None')

示例#7

文件： test_awslambda_actions.py 项目： chaostoolkit-incubator/chaostoolkit-aws

def test_aws_lambda_invoke_no_args(aws_client):
    client = MagicMock()
    aws_client.return_value = client
    mock_payload = {"some": "response"}
    mock_response = {"Payload": MagicMock()}
    mock_response["Payload"].read.return_value = json.dumps(
        mock_payload).encode()
    client.invoke.return_value = mock_response
    lambda_function_name = "my-lambda-function"

    response = invoke_function(lambda_function_name)

    assert response["Payload"] == mock_payload
    client.invoke.assert_called_with(
        FunctionName=lambda_function_name,
        InvocationType="RequestResponse",
        LogType="None",
    )