add server tests

Signed-off-by: Florian Bacher <florian.bacher@dynatrace.com>

cmd/opampsupervisor/supervisor/server_test.go

@@ -0,0 +1,81 @@
+package supervisor
+
+import (
+	"context"
+	"github.com/open-telemetry/opamp-go/protobufs"
+	serverTypes "github.com/open-telemetry/opamp-go/server/types"
+	"github.com/stretchr/testify/require"
+	"net/http"
+	"testing"
+)
+
+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(request *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(request *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(conn serverTypes.Connection, message *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(conn serverTypes.Connection) {
+			onConnectionCloseFuncCalled = true
+		},
+	}
+
+	fs.OnConnectionClose(&mockConn{})
+
+	require.True(t, onConnectionCloseFuncCalled)
+}