[chore] add unit tests for the opampsupervisor

.chloggen/opamp-supervisor-unit-tests.yaml

@@ -0,0 +1,27 @@
+# Use this changelog template to create an entry for release notes.
+
+# One of 'breaking', 'deprecation', 'new_component', 'enhancement', 'bug_fix'
+change_type: enhancement
+
+# The name of the component, or a single word describing the area of concern, (e.g. filelogreceiver)
+component: opemapsupervisor
+
+# A brief description of the change.  Surround your text with quotes ("") if it needs to start with a backtick (`).
+note: Increase unit test coverage.
+
+# Mandatory: One or more tracking issues related to the change. You can use the PR number here if no issue exists.
+issues: [24295]
+
+# (Optional) One or more lines of additional information to render under the primary note.
+# These lines will be padded with 2 spaces and then inserted directly into the document.
+# Use pipe (|) for multiline entries.
+subtext:
+
+# If your change doesn't affect end users or the exported elements of any package,
+# you should instead start your pull request title with [chore] or use the "Skip Changelog" label.
+# Optional: The change log or logs in which this entry should be included.
+# e.g. '[user]' or '[user, api]'
+# Include 'user' if the change is relevant to end users.
+# Include 'api' if there is a change to a library API.
+# Default: '[user]'
+change_logs: []

cmd/opampsupervisor/supervisor/server_test.go

@@ -0,0 +1,85 @@
+// Copyright The OpenTelemetry Authors
+// SPDX-License-Identifier: Apache-2.0
+
+package supervisor
+
+import (
+	"context"
+	"net/http"
+	"testing"
+
+	"github.com/open-telemetry/opamp-go/protobufs"
+	serverTypes "github.com/open-telemetry/opamp-go/server/types"
+	"github.com/stretchr/testify/require"
+)
+
+func Test_flattenedSettings_toServerSettings(t *testing.T) {
+	fs := flattenedSettings{
+		endpoint: "localhost",
+	}
+
+	serverSettings := fs.toServerSettings()
+
+	require.Equal(t, "localhost", serverSettings.ListenEndpoint)
+	require.NotNil(t, serverSettings.Callbacks)
+}
+
+func Test_flattenedSettings_OnConnecting(t *testing.T) {
+	t.Run("accept connection", func(t *testing.T) {
+		onConnectingFuncCalled := false
+		fs := flattenedSettings{
+			onConnectingFunc: func(_ *http.Request) (shouldConnect bool, rejectStatusCode int) {
+				onConnectingFuncCalled = true
+				return true, 0
+			},
+		}
+
+		cr := fs.OnConnecting(&http.Request{})
+
+		require.True(t, onConnectingFuncCalled)
+		require.True(t, cr.Accept)
+		require.NotNil(t, cr.ConnectionCallbacks)
+	})
+	t.Run("do not accept connection", func(t *testing.T) {
+		onConnectingFuncCalled := false
+		fs := flattenedSettings{
+			onConnectingFunc: func(_ *http.Request) (shouldConnect bool, rejectStatusCode int) {
+				onConnectingFuncCalled = true
+				return false, 500
+			},
+		}
+
+		cr := fs.OnConnecting(&http.Request{})
+
+		require.True(t, onConnectingFuncCalled)
+		require.False(t, cr.Accept)
+		require.Equal(t, 500, cr.HTTPStatusCode)
+	})
+}
+
+func Test_flattenedSettings_OnMessage(t *testing.T) {
+	onMessageFuncCalled := false
+	fs := flattenedSettings{
+		onMessageFunc: func(_ serverTypes.Connection, _ *protobufs.AgentToServer) {
+			onMessageFuncCalled = true
+		},
+	}
+
+	sta := fs.OnMessage(context.TODO(), &mockConn{}, &protobufs.AgentToServer{})
+
+	require.True(t, onMessageFuncCalled)
+	require.NotNil(t, sta)
+}
+
+func Test_flattenedSettings_OnConnectionClose(t *testing.T) {
+	onConnectionCloseFuncCalled := false
+	fs := flattenedSettings{
+		onConnectionCloseFunc: func(_ serverTypes.Connection) {
+			onConnectionCloseFuncCalled = true
+		},
+	}
+
+	fs.OnConnectionClose(&mockConn{})
+
+	require.True(t, onConnectionCloseFuncCalled)
+}