gitee.com/ks-custle/core-gm@v0.0.0-20230922171213-b83bdd97b62c/grpc/trace_test.go (about)

     1  /*
     2   *
     3   * Copyright 2019 gRPC authors.
     4   *
     5   * Licensed under the Apache License, Version 2.0 (the "License");
     6   * you may not use this file except in compliance with the License.
     7   * You may obtain a copy of the License at
     8   *
     9   *     http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the License for the specific language governing permissions and
    15   * limitations under the License.
    16   *
    17   */
    18  
    19  package grpc
    20  
    21  import (
    22  	"testing"
    23  )
    24  
    25  func (s) TestMethodFamily(t *testing.T) {
    26  	cases := []struct {
    27  		desc             string
    28  		method           string
    29  		wantMethodFamily string
    30  	}{
    31  		{
    32  			desc:             "No leading slash",
    33  			method:           "pkg.service/method",
    34  			wantMethodFamily: "pkg.service",
    35  		},
    36  		{
    37  			desc:             "Leading slash",
    38  			method:           "/pkg.service/method",
    39  			wantMethodFamily: "pkg.service",
    40  		},
    41  	}
    42  
    43  	for _, ut := range cases {
    44  		t.Run(ut.desc, func(t *testing.T) {
    45  			if got := methodFamily(ut.method); got != ut.wantMethodFamily {
    46  				t.Fatalf("methodFamily(%s) = %s, want %s", ut.method, got, ut.wantMethodFamily)
    47  			}
    48  		})
    49  	}
    50  }