mirror of
https://github.com/hanzoai/otel-collector.git
synced 2026-10-30 18:26:25 +00:00
zeekayandHanzo Dev
c16ed9fe26
Modules declare go 1.27.1 and the build images follow. go fix applied the modernizers; omitzero is not applied, since it changes what the JSON says. Co-authored-by: Hanzo Dev <[email protected]>
101 lines
2.6 KiB
Go
101 lines
2.6 KiB
Go
package opamp
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"sync/atomic"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"github.com/hanzoai/otel-collector/o11ycol"
|
|
"github.com/open-telemetry/opamp-go/protobufs"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
func TestNewClient(t *testing.T) {
|
|
// FIXME(srikanthccv): this test interferes with other tests that use the same port.
|
|
// Remove the sleep and find a better way to fix this.
|
|
time.Sleep(5 * time.Second)
|
|
srv := StartMockServer(t)
|
|
t.Cleanup(func() { srv.Close() })
|
|
|
|
var conn atomic.Value
|
|
srv.OnWSConnect = func(c *websocket.Conn) {
|
|
conn.Store(c)
|
|
}
|
|
var connected atomic.Int64
|
|
|
|
srv.OnMessage = func(msg *protobufs.AgentToServer) *protobufs.ServerToAgent {
|
|
fileContents, err := os.ReadFile("testdata/coll-config-path-changed.yaml")
|
|
assert.NoError(t, err)
|
|
connected.Add(1)
|
|
var resp *protobufs.ServerToAgent = &protobufs.ServerToAgent{}
|
|
if connected.Load() == 1 {
|
|
resp = &protobufs.ServerToAgent{
|
|
RemoteConfig: &protobufs.AgentRemoteConfig{
|
|
Config: &protobufs.AgentConfigMap{
|
|
ConfigMap: map[string]*protobufs.AgentConfigFile{
|
|
"collector.yaml": {
|
|
Body: fileContents,
|
|
ContentType: "text/yaml",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
return resp
|
|
}
|
|
|
|
logger := zap.NewNop()
|
|
cnt := 0
|
|
reloadFunc := func(contents []byte) error {
|
|
cnt++
|
|
return nil
|
|
}
|
|
|
|
_, err := NewDynamicConfig("./testdata/coll-config-path.yaml", reloadFunc, logger)
|
|
require.NoError(t, err)
|
|
|
|
// maintain a cop of the original config file and restore it after the test
|
|
fileContents, err := os.ReadFile("testdata/coll-config-path.yaml")
|
|
require.NoError(t, err)
|
|
defer func() {
|
|
err := os.WriteFile("testdata/coll-config-path.yaml", fileContents, 0644)
|
|
require.NoError(t, err)
|
|
}()
|
|
|
|
coll := o11ycol.New(o11ycol.WrappedCollectorSettings{
|
|
ConfigPaths: []string{"./testdata/coll-config-path.yaml"},
|
|
Version: "0.0.1-server-client-test",
|
|
})
|
|
|
|
svrClient, err := NewServerClient(&NewServerClientOpts{
|
|
Logger: logger,
|
|
Config: &AgentManagerConfig{
|
|
ServerEndpoint: "ws://" + srv.Endpoint,
|
|
},
|
|
CollectorConfigPath: "./testdata/coll-config-path.yaml",
|
|
WrappedCollector: coll,
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 30*time.Second)
|
|
t.Cleanup(cancel)
|
|
// Start the client.
|
|
err = svrClient.Start(ctx)
|
|
t.Cleanup(func() {
|
|
_ = svrClient.Stop(ctx)
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
// Wait for the client to connect.
|
|
assert.Eventually(t, func() bool {
|
|
return connected.Load() == 1
|
|
}, 15*time.Second, 10*time.Millisecond)
|
|
}
|