65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package jitsiexporter
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/client_golang/prometheus/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestUpdate(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
s := make(map[string]interface{})
|
|
s["foo"] = "foo"
|
|
s["bar"] = 1 // nolint:gomnd
|
|
s["zonk"] = float64(1) // nolint:gomnd
|
|
mockStater := &MockStater{}
|
|
mockStater.On("Now", "http://foo.tld").Return(s, nil)
|
|
|
|
m := &Metrics{
|
|
URL: "http://foo.tld",
|
|
Metrics: make(map[string]Metric),
|
|
Stater: mockStater,
|
|
}
|
|
|
|
err := m.Update()
|
|
assert.Empty(err)
|
|
|
|
assert.Equal(testutil.ToFloat64(m.Metrics["jitsi_zonk"].Gauge), float64(1))
|
|
assert.Equal(m.Metrics["jitsi_foo"], Metric{Name: "", Gauge: prometheus.Gauge(nil)})
|
|
assert.Equal(m.Metrics["jitsi_bar"], Metric{Name: "", Gauge: prometheus.Gauge(nil)})
|
|
assert.Equal(len(m.Metrics), 1)
|
|
}
|
|
|
|
func TestUpdateOnError(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
mockStater := &MockStater{}
|
|
mockStater.On("Now", "http://foo.tld").Return(nil, errors.New("something went foo"))
|
|
|
|
e := prometheus.NewCounter(prometheus.CounterOpts{Name: "jitsi_fetch_errors"})
|
|
metricsMap := make(map[string]Metric)
|
|
metricsMap["testmetric"] = Metric{
|
|
Name: "testmetric",
|
|
Gauge: prometheus.NewGauge(prometheus.GaugeOpts{Name: "jitsi_testmetric"}),
|
|
}
|
|
m := &Metrics{
|
|
URL: "http://foo.tld",
|
|
Metrics: metricsMap,
|
|
Stater: mockStater,
|
|
Errors: e,
|
|
}
|
|
|
|
assert.Equal(1, len(m.Metrics))
|
|
|
|
err := m.Update()
|
|
fmt.Println(err)
|
|
assert.NotEmpty(err)
|
|
|
|
assert.Equal(0, len(m.Metrics))
|
|
}
|