github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/network/armnetwork/v6@v6.2.0/fake/applicationgatewaywafdynamicmanifestsdefault_server.go (about)

     1  //go:build go1.18
     2  // +build go1.18
     3  
     4  // Copyright (c) Microsoft Corporation. All rights reserved.
     5  // Licensed under the MIT License. See License.txt in the project root for license information.
     6  // Code generated by Microsoft (R) AutoRest Code Generator. DO NOT EDIT.
     7  // Changes may cause incorrect behavior and will be lost if the code is regenerated.
     8  
     9  package fake
    10  
    11  import (
    12  	"context"
    13  	"errors"
    14  	"fmt"
    15  	azfake "github.com/Azure/azure-sdk-for-go/sdk/azcore/fake"
    16  	"github.com/Azure/azure-sdk-for-go/sdk/azcore/fake/server"
    17  	"github.com/Azure/azure-sdk-for-go/sdk/azcore/runtime"
    18  	"github.com/Azure/azure-sdk-for-go/sdk/resourcemanager/network/armnetwork/v6"
    19  	"net/http"
    20  	"net/url"
    21  	"regexp"
    22  )
    23  
    24  // ApplicationGatewayWafDynamicManifestsDefaultServer is a fake server for instances of the armnetwork.ApplicationGatewayWafDynamicManifestsDefaultClient type.
    25  type ApplicationGatewayWafDynamicManifestsDefaultServer struct {
    26  	// Get is the fake for method ApplicationGatewayWafDynamicManifestsDefaultClient.Get
    27  	// HTTP status codes to indicate success: http.StatusOK
    28  	Get func(ctx context.Context, location string, options *armnetwork.ApplicationGatewayWafDynamicManifestsDefaultClientGetOptions) (resp azfake.Responder[armnetwork.ApplicationGatewayWafDynamicManifestsDefaultClientGetResponse], errResp azfake.ErrorResponder)
    29  }
    30  
    31  // NewApplicationGatewayWafDynamicManifestsDefaultServerTransport creates a new instance of ApplicationGatewayWafDynamicManifestsDefaultServerTransport with the provided implementation.
    32  // The returned ApplicationGatewayWafDynamicManifestsDefaultServerTransport instance is connected to an instance of armnetwork.ApplicationGatewayWafDynamicManifestsDefaultClient via the
    33  // azcore.ClientOptions.Transporter field in the client's constructor parameters.
    34  func NewApplicationGatewayWafDynamicManifestsDefaultServerTransport(srv *ApplicationGatewayWafDynamicManifestsDefaultServer) *ApplicationGatewayWafDynamicManifestsDefaultServerTransport {
    35  	return &ApplicationGatewayWafDynamicManifestsDefaultServerTransport{srv: srv}
    36  }
    37  
    38  // ApplicationGatewayWafDynamicManifestsDefaultServerTransport connects instances of armnetwork.ApplicationGatewayWafDynamicManifestsDefaultClient to instances of ApplicationGatewayWafDynamicManifestsDefaultServer.
    39  // Don't use this type directly, use NewApplicationGatewayWafDynamicManifestsDefaultServerTransport instead.
    40  type ApplicationGatewayWafDynamicManifestsDefaultServerTransport struct {
    41  	srv *ApplicationGatewayWafDynamicManifestsDefaultServer
    42  }
    43  
    44  // Do implements the policy.Transporter interface for ApplicationGatewayWafDynamicManifestsDefaultServerTransport.
    45  func (a *ApplicationGatewayWafDynamicManifestsDefaultServerTransport) Do(req *http.Request) (*http.Response, error) {
    46  	rawMethod := req.Context().Value(runtime.CtxAPINameKey{})
    47  	method, ok := rawMethod.(string)
    48  	if !ok {
    49  		return nil, nonRetriableError{errors.New("unable to dispatch request, missing value for CtxAPINameKey")}
    50  	}
    51  
    52  	var resp *http.Response
    53  	var err error
    54  
    55  	switch method {
    56  	case "ApplicationGatewayWafDynamicManifestsDefaultClient.Get":
    57  		resp, err = a.dispatchGet(req)
    58  	default:
    59  		err = fmt.Errorf("unhandled API %s", method)
    60  	}
    61  
    62  	if err != nil {
    63  		return nil, err
    64  	}
    65  
    66  	return resp, nil
    67  }
    68  
    69  func (a *ApplicationGatewayWafDynamicManifestsDefaultServerTransport) dispatchGet(req *http.Request) (*http.Response, error) {
    70  	if a.srv.Get == nil {
    71  		return nil, &nonRetriableError{errors.New("fake for method Get not implemented")}
    72  	}
    73  	const regexStr = `/subscriptions/(?P<subscriptionId>[!#&$-;=?-\[\]_a-zA-Z0-9~%@]+)/providers/Microsoft\.Network/locations/(?P<location>[!#&$-;=?-\[\]_a-zA-Z0-9~%@]+)/applicationGatewayWafDynamicManifests/dafault`
    74  	regex := regexp.MustCompile(regexStr)
    75  	matches := regex.FindStringSubmatch(req.URL.EscapedPath())
    76  	if matches == nil || len(matches) < 2 {
    77  		return nil, fmt.Errorf("failed to parse path %s", req.URL.Path)
    78  	}
    79  	locationParam, err := url.PathUnescape(matches[regex.SubexpIndex("location")])
    80  	if err != nil {
    81  		return nil, err
    82  	}
    83  	respr, errRespr := a.srv.Get(req.Context(), locationParam, nil)
    84  	if respErr := server.GetError(errRespr, req); respErr != nil {
    85  		return nil, respErr
    86  	}
    87  	respContent := server.GetResponseContent(respr)
    88  	if !contains([]int{http.StatusOK}, respContent.HTTPStatus) {
    89  		return nil, &nonRetriableError{fmt.Errorf("unexpected status code %d. acceptable values are http.StatusOK", respContent.HTTPStatus)}
    90  	}
    91  	resp, err := server.MarshalResponseAsJSON(respContent, server.GetResponse(respr).ApplicationGatewayWafDynamicManifestResult, req)
    92  	if err != nil {
    93  		return nil, err
    94  	}
    95  	return resp, nil
    96  }