52 lines
1.9 KiB
Go
52 lines
1.9 KiB
Go
|
package logging
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
log "github.com/micro/go-micro/v3/logger"
|
||
|
|
"github.com/micro/go-micro/v3/metrics"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestLoggingReporter(t *testing.T) {
|
||
|
|
|
||
|
|
// Make a Reporter:
|
||
|
|
reporter := New(metrics.Path("/prometheus"), metrics.DefaultTags(map[string]string{"service": "prometheus-test"}))
|
||
|
|
assert.NotNil(t, reporter)
|
||
|
|
assert.Equal(t, "prometheus-test", reporter.options.DefaultTags["service"])
|
||
|
|
assert.Equal(t, ":9000", reporter.options.Address)
|
||
|
|
assert.Equal(t, "/prometheus", reporter.options.Path)
|
||
|
|
|
||
|
|
// Make a log buffer and create a new logger to use it:
|
||
|
|
logBuffer := new(bytes.Buffer)
|
||
|
|
reporter.logger = log.NewLogger(log.WithLevel(log.TraceLevel), log.WithOutput(logBuffer), log.WithFields(convertTags(reporter.options.DefaultTags)))
|
||
|
|
|
||
|
|
// Check that our implementation is valid:
|
||
|
|
assert.Implements(t, new(metrics.Reporter), reporter)
|
||
|
|
|
||
|
|
// Test tag conversion:
|
||
|
|
tags := metrics.Tags{
|
||
|
|
"tag1": "false",
|
||
|
|
"tag2": "true",
|
||
|
|
}
|
||
|
|
convertedTags := convertTags(tags)
|
||
|
|
assert.Equal(t, "false", convertedTags["tag1"])
|
||
|
|
assert.Equal(t, "true", convertedTags["tag2"])
|
||
|
|
|
||
|
|
// Test submitting metrics through the interface methods:
|
||
|
|
assert.NoError(t, reporter.Count("test.counter.1", 6, tags))
|
||
|
|
assert.NoError(t, reporter.Count("test.counter.2", 19, tags))
|
||
|
|
assert.NoError(t, reporter.Count("test.counter.1", 5, tags))
|
||
|
|
assert.NoError(t, reporter.Gauge("test.gauge.1", 99, tags))
|
||
|
|
assert.NoError(t, reporter.Gauge("test.gauge.2", 55, tags))
|
||
|
|
assert.NoError(t, reporter.Gauge("test.gauge.1", 98, tags))
|
||
|
|
assert.NoError(t, reporter.Timing("test.timing.1", time.Second, tags))
|
||
|
|
assert.NoError(t, reporter.Timing("test.timing.2", time.Minute, tags))
|
||
|
|
|
||
|
|
// Test reading back the metrics from the logbuffer (doesn't seem to work because the output still goes to StdOut):
|
||
|
|
// assert.Contains(t, logBuffer.String(), "level=debug service=prometheus-test Count metric: map[tag1:false tag2:true]")
|
||
|
|
}
|