harness-drone/metric/users_test.go
2019-02-19 15:56:41 -08:00

54 lines
1.3 KiB
Go

// Copyright 2019 Drone.IO Inc. All rights reserved.
// Use of this source code is governed by the Drone Non-Commercial License
// that can be found in the LICENSE file.
package metric
import (
"testing"
"github.com/drone/drone/mock"
"github.com/golang/mock/gomock"
"github.com/prometheus/client_golang/prometheus"
)
func TestUserCount(t *testing.T) {
controller := gomock.NewController(t)
// restore the default prometheus registerer
// when the unit test is complete.
snapshot := prometheus.DefaultRegisterer
defer func() {
prometheus.DefaultRegisterer = snapshot
controller.Finish()
}()
// creates a blank registry
registry := prometheus.NewRegistry()
prometheus.DefaultRegisterer = registry
// x2 repository count
count := int64(5)
store := mock.NewMockUserStore(controller)
store.EXPECT().Count(gomock.Any()).Return(count, nil)
UserCount(store)
metrics, err := registry.Gather()
if err != nil {
t.Error(err)
return
}
if want, got := len(metrics), 1; want != got {
t.Errorf("Expect registered metric")
return
}
metric := metrics[0]
if want, got := metric.GetName(), "drone_user_count"; want != got {
t.Errorf("Expect metric name %s, got %s", want, got)
}
if want, got := metric.Metric[0].Gauge.GetValue(), float64(count); want != got {
t.Errorf("Expect metric value %f, got %f", want, got)
}
}