github.com/blend/go-sdk@v1.20220411.3/stats/grpcstats/listeners_test.go (about)

     1  /*
     2  
     3  Copyright (c) 2022 - Present. Blend Labs, Inc. All rights reserved
     4  Use of this source code is governed by a MIT license that can be found in the LICENSE file.
     5  
     6  */
     7  
     8  package grpcstats
     9  
    10  import (
    11  	"testing"
    12  
    13  	"github.com/blend/go-sdk/assert"
    14  	"github.com/blend/go-sdk/grpcutil"
    15  	"github.com/blend/go-sdk/logger"
    16  	"github.com/blend/go-sdk/stats"
    17  )
    18  
    19  func TestAddListeners(t *testing.T) {
    20  	assert := assert.New(t)
    21  
    22  	log := logger.None()
    23  	AddListeners(nil, nil)
    24  	assert.False(log.HasListener(grpcutil.FlagRPC, stats.ListenerNameStats))
    25  	AddListeners(log, stats.NewMockCollector(32))
    26  	assert.True(log.HasListener(grpcutil.FlagRPC, stats.ListenerNameStats))
    27  }