trpc.group/trpc-go/trpc-go@v1.0.3/naming/selector/passthrough_test.go (about)

     1  //
     2  //
     3  // Tencent is pleased to support the open source community by making tRPC available.
     4  //
     5  // Copyright (C) 2023 THL A29 Limited, a Tencent company.
     6  // All rights reserved.
     7  //
     8  // If you have downloaded a copy of the tRPC source code from Tencent,
     9  // please note that tRPC source code is licensed under the  Apache 2.0 License,
    10  // A copy of the Apache 2.0 License is included in this file.
    11  //
    12  //
    13  
    14  package selector
    15  
    16  import (
    17  	"testing"
    18  
    19  	"github.com/stretchr/testify/assert"
    20  )
    21  
    22  func TestPassthroughSelectorSelect(t *testing.T) {
    23  	selector := &passthroughSelector{}
    24  	n, err := selector.Select("passthrough")
    25  	assert.Nil(t, err)
    26  	assert.Equal(t, n.Address, "passthrough")
    27  	assert.Equal(t, n.ServiceName, "passthrough")
    28  }
    29  
    30  func TestPassthroughSelectorReport(t *testing.T) {
    31  	selector := &passthroughSelector{}
    32  	n, err := selector.Select("passthrough")
    33  	assert.Nil(t, err)
    34  	assert.Equal(t, n.Address, "passthrough")
    35  	assert.Equal(t, n.ServiceName, "passthrough")
    36  	assert.Nil(t, selector.Report(n, 0, nil))
    37  }