KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use _botocore_request_to_localstack_request method in localstack

Best Python code snippet using localstack_python

test_service_router.py

Source:test_service_router.py

...87        parameter_validation=False,88        user_agent="aws-cli/1.33.7",89    )90    return aws_stack.create_external_boto_client(service, config=config)91def _botocore_request_to_localstack_request(request_object: AWSRequest) -> Request:92    """Converts a botocore request (AWSRequest) to our HTTP framework's Request object based on Werkzeug."""93    split_url = urlsplit(request_object.url)94    path = split_url.path95    query_string = split_url.query96    body = request_object.body97    headers = request_object.headers98    return Request(99        method=request_object.method or "GET",100        path=path,101        query_string=to_str(query_string),102        headers=dict(headers),103        body=body,104        raw_path=path,105    )106# Simple dummy value mapping for the different shape types107_dummy_values = {108    "string": "dummy-value",109    "list": [],110    "integer": 0,111    "long": 0,112    "timestamp": datetime.now(),113}114def _create_dummy_request_args(operation_model: OperationModel) -> Dict:115    """Creates a dummy request param dict for the given operation."""116    input_shape: StructureShape = operation_model.input_shape117    if not input_shape:118        return {}119    result = {}120    for required_member in input_shape.required_members:121        required_shape: Shape = input_shape.members[required_member]122        location = required_shape.serialization.get("location")123        if location in ["uri", "querystring", "header", "headers"]:124            result[required_member] = _dummy_values[required_shape.type_name]125    return result126def _generate_test_name(param: Any):127    """Simple helper function to generate readable test names."""128    if isinstance(param, ServiceModel):129        return param.service_name130    elif isinstance(param, OperationModel):131        return param.name132    return param133@pytest.mark.parametrize(134    "service, operation",135    _collect_operations(),136    ids=_generate_test_name,137)138def test_service_router_works_for_every_service(139    service: ServiceModel, operation: OperationModel, caplog140):141    caplog.set_level("CRITICAL", "botocore")142    # Create a dummy request for the service router143    client = _client(service.service_name)144    request_context = {145        "client_region": client.meta.region_name,146        "client_config": client.meta.config,147        "has_streaming_input": operation.has_streaming_input,148        "auth_type": operation.auth_type,149    }150    request_args = _create_dummy_request_args(operation)151    request_dict = client._convert_to_request_dict(request_args, operation, request_context)152    request_object = create_request_object(request_dict)153    client._request_signer.sign(operation.name, request_object)154    request: Request = _botocore_request_to_localstack_request(request_object)155    # Execute the service router156    detected_service_name = determine_aws_service_name(request)157    # Make sure the detected service is the same as the one we generated the request for158    assert service.service_name == detected_service_name159def test_endpoint_prefix_based_routing():160    # TODO could be generalized using endpoint resolvers and replacing "amazonaws.com" with "localhost.localstack.cloud"161    detected_service_name = determine_aws_service_name(162        Request(method="GET", path="/", headers={"Host": "sqs.localhost.localstack.cloud"})163    )164    assert detected_service_name == "sqs"165    detected_service_name = determine_aws_service_name(166        Request(167            method="POST",168            path="/app-instances",...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.