Skip to content

Commit 07eee17

Browse files
author
Yuri Shkuro
committed
Add tests
1 parent 0022dc7 commit 07eee17

File tree

1 file changed

+18
-2
lines changed

1 file changed

+18
-2
lines changed

cmd/agent/app/builder_test.go

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ import (
3030
"go.uber.org/zap"
3131
"gopkg.in/yaml.v2"
3232

33+
"github.com/uber/jaeger-lib/metrics"
3334
"github.com/uber/jaeger/thrift-gen/jaeger"
3435
"github.com/uber/jaeger/thrift-gen/zipkincore"
3536
)
@@ -123,11 +124,26 @@ func TestBuilderWithExtraReporter(t *testing.T) {
123124
assert.NotNil(t, agent)
124125
}
125126

126-
func TestBuilderWithError(t *testing.T) {
127+
func TestBuilderMetrics(t *testing.T) {
128+
mf := metrics.NullFactory
129+
b := new(Builder).WithMetricsFactory(mf)
130+
mf2, err := b.getMetricsFactory()
131+
assert.NoError(t, err)
132+
assert.Equal(t, mf, mf2)
133+
}
134+
135+
func TestBuilderMetricsError(t *testing.T) {
136+
b := &Builder{}
137+
b.Metrics.Backend = "invalid"
138+
_, err := b.CreateAgent(zap.NewNop())
139+
assert.EqualError(t, err, "cannot create metrics factory: unknown metrics backend specified")
140+
}
141+
142+
func TestBuilderWithDiscoveryError(t *testing.T) {
127143
cfg := &Builder{}
128144
cfg.WithDiscoverer(fakeDiscoverer{})
129145
agent, err := cfg.CreateAgent(zap.NewNop())
130-
assert.Error(t, err)
146+
assert.EqualError(t, err, "cannot create main Reporter: cannot enable service discovery: both discovery.Discoverer and discovery.Notifier must be specified")
131147
assert.Nil(t, agent)
132148
}
133149

0 commit comments

Comments
 (0)