You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
|
|
|
package trace
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
|
|
)
|
|
|
|
|
|
|
|
func TestStartAgent(t *testing.T) {
|
|
|
|
logx.Disable()
|
|
|
|
|
|
|
|
const (
|
|
|
|
endpoint1 = "localhost:1234"
|
|
|
|
endpoint2 = "remotehost:1234"
|
|
|
|
)
|
|
|
|
c1 := Config{
|
|
|
|
Name: "foo",
|
|
|
|
}
|
|
|
|
c2 := Config{
|
|
|
|
Name: "bar",
|
|
|
|
Endpoint: endpoint1,
|
|
|
|
Batcher: kindJaeger,
|
|
|
|
}
|
|
|
|
c3 := Config{
|
|
|
|
Name: "any",
|
|
|
|
Endpoint: endpoint2,
|
|
|
|
Batcher: kindZipkin,
|
|
|
|
}
|
|
|
|
c4 := Config{
|
|
|
|
Name: "bla",
|
|
|
|
Endpoint: endpoint1,
|
|
|
|
Batcher: "otlp",
|
|
|
|
}
|
|
|
|
c5 := Config{
|
|
|
|
Name: "grpc",
|
|
|
|
Endpoint: endpoint1,
|
|
|
|
Batcher: "grpc",
|
|
|
|
}
|
|
|
|
|
|
|
|
StartAgent(c1)
|
|
|
|
StartAgent(c1)
|
|
|
|
StartAgent(c2)
|
|
|
|
StartAgent(c3)
|
|
|
|
StartAgent(c4)
|
|
|
|
StartAgent(c5)
|
|
|
|
|
|
|
|
lock.Lock()
|
|
|
|
defer lock.Unlock()
|
|
|
|
|
|
|
|
// because remotehost cannot be resolved
|
|
|
|
assert.Equal(t, 2, len(agents))
|
|
|
|
_, ok := agents[""]
|
|
|
|
assert.True(t, ok)
|
|
|
|
_, ok = agents[endpoint1]
|
|
|
|
assert.True(t, ok)
|
|
|
|
_, ok = agents[endpoint2]
|
|
|
|
assert.False(t, ok)
|
|
|
|
}
|