grafana/pkg/infra/log/databaseCounter_test.go
Carl Bergquist 78978048c3
Instrumentation: log the total number of db queries per request (#54647)
Signed-off-by: bergquist <carl.bergquist@gmail.com>
Co-authored-by: Emil Tullstedt <emil.tullstedt@grafana.com>
Co-authored-by: Dave Henderson <dave.henderson@grafana.com>
2022-09-05 07:39:22 +02:00

20 lines
361 B
Go

package log
import (
"context"
"testing"
"github.com/stretchr/testify/assert"
)
func TestCountingDatabaseCalls(t *testing.T) {
ctx := context.Background()
ctx = IncDBCallCounter(ctx)
ctx = IncDBCallCounter(ctx)
ctx = IncDBCallCounter(ctx)
count := TotalDBCallCount(ctx)
assert.Equal(t, int64(3), count, "expect counter to increase three times")
}