🎉 add unit tests

Signed-off-by: Salim Afiune Maya <afiune@mondoo.com>

Author: afiune

providers/os/connection/ssh/ssh_test.go

@@ -4,9 +4,11 @@
 package ssh
 
 import (
+	"net"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 	"go.mondoo.com/cnquery/v11/providers-sdk/v1/inventory"
 	"go.mondoo.com/cnquery/v11/providers/os/connection/shared"
 )
@@ -37,3 +39,66 @@ func TestSSHAuthError(t *testing.T) {
 			// local testing without ssh agent
 			err.Error() == "no authentication method defined")
 }
+
+// helper to start a fake SSH server with a custom banner
+func startMockSSHServer(t *testing.T, banner string) (addr string, closeFn func()) {
+	ln, err := net.Listen("tcp", "127.0.0.1:0")
+	require.Nil(t, err)
+
+	go func() {
+		conn, err := ln.Accept()
+		if err != nil {
+			return
+		}
+		defer conn.Close()
+		// simulate SSH banner
+		conn.Write([]byte(banner + "\r\n"))
+	}()
+
+	return ln.Addr().String(), func() { ln.Close() }
+}
+
+func TestServerSupportsHybridKEX(t *testing.T) {
+	tests := []struct {
+		name         string
+		banner       string
+		expectHybrid bool
+	}{
+		{
+			name:         "OpenSSH 9.9 detected",
+			banner:       "SSH-2.0-OpenSSH_9.9",
+			expectHybrid: true,
+		},
+		{
+			name:         "OpenSSH 9.7 (no hybrid)",
+			banner:       "SSH-2.0-OpenSSH_9.7",
+			expectHybrid: false,
+		},
+		{
+			name:         "Non-OpenSSH server",
+			banner:       "SSH-2.0-CustomSSH_1.0",
+			expectHybrid: false,
+		},
+		{
+			name:         "Malformed banner",
+			banner:       "garbage",
+			expectHybrid: false,
+		},
+	}
+
+	for _, tt := range tests {
+		t.Run(tt.name, func(t *testing.T) {
+			addr, shutdown := startMockSSHServer(t, tt.banner)
+			defer shutdown()
+
+			got, err := serverSupportsHybridKEX(addr)
+			require.Nil(t, err)
+			assert.Equal(t, tt.expectHybrid, got)
+		})
+	}
+}
+
+func TestServerSupportsHybridKEX_ServerUnreachable(t *testing.T) {
+	_, err := serverSupportsHybridKEX("127.0.0.1:9")
+	require.NotNil(t, err)
+}