github.com/lyft/flytestdlib@v0.3.12-0.20210213045714-8cdd111ecda1/promutils/labeled/keys_test.go (about)

     1  package labeled
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/lyft/flytestdlib/contextutils"
     7  	"github.com/stretchr/testify/assert"
     8  )
     9  
    10  func TestMetricKeys(t *testing.T) {
    11  	UnsetMetricKeys()
    12  	input := []contextutils.Key{
    13  		contextutils.ProjectKey, contextutils.DomainKey, contextutils.WorkflowIDKey, contextutils.TaskIDKey, contextutils.LaunchPlanIDKey,
    14  	}
    15  
    16  	assert.NotPanics(t, func() { SetMetricKeys(input...) })
    17  	assert.Equal(t, input, metricKeys)
    18  
    19  	for i, k := range metricKeys {
    20  		assert.Equal(t, k.String(), metricStringKeys[i])
    21  	}
    22  
    23  	assert.NotPanics(t, func() { SetMetricKeys(input...) })
    24  	assert.Panics(t, func() { SetMetricKeys(contextutils.DomainKey) })
    25  }