Add aliases and conversion from the kftray.

2026-07-12 07:41:42 +00:00 · 2025-11-23 15:42:42 +00:00
parent f50f0a9b49
commit 555f21c6f3
6 changed files with 532 additions and 8 deletions
@@ -0,0 +1,212 @@
+package converter
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestConvertToKPortal_SingleContext(t *testing.T) {
+	kftrayConfigs := []KFTrayConfig{
+		{
+			Service:      "postgres",
+			Namespace:    "default",
+			LocalPort:    5432,
+			RemotePort:   5432,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "prod-db",
+		},
+	}
+
+	result := convertToKPortal(kftrayConfigs)
+
+	assert.Len(t, result.Contexts, 1)
+	assert.Equal(t, "production", result.Contexts[0].Name)
+	assert.Len(t, result.Contexts[0].Namespaces, 1)
+	assert.Equal(t, "default", result.Contexts[0].Namespaces[0].Name)
+	assert.Len(t, result.Contexts[0].Namespaces[0].Forwards, 1)
+
+	forward := result.Contexts[0].Namespaces[0].Forwards[0]
+	assert.Equal(t, "service/postgres", forward.Resource)
+	assert.Equal(t, "tcp", forward.Protocol)
+	assert.Equal(t, 5432, forward.Port)
+	assert.Equal(t, 5432, forward.LocalPort)
+	assert.Equal(t, "prod-db", forward.Alias)
+}
+
+func TestConvertToKPortal_MultipleContexts(t *testing.T) {
+	kftrayConfigs := []KFTrayConfig{
+		{
+			Service:      "api",
+			Namespace:    "default",
+			LocalPort:    8080,
+			RemotePort:   8080,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "prod-api",
+		},
+		{
+			Service:      "api",
+			Namespace:    "default",
+			LocalPort:    8081,
+			RemotePort:   8080,
+			Context:      "staging",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "staging-api",
+		},
+	}
+
+	result := convertToKPortal(kftrayConfigs)
+
+	assert.Len(t, result.Contexts, 2)
+
+	// Contexts should be sorted alphabetically
+	assert.Equal(t, "production", result.Contexts[0].Name)
+	assert.Equal(t, "staging", result.Contexts[1].Name)
+}
+
+func TestConvertToKPortal_MultipleNamespaces(t *testing.T) {
+	kftrayConfigs := []KFTrayConfig{
+		{
+			Service:      "api",
+			Namespace:    "default",
+			LocalPort:    8080,
+			RemotePort:   8080,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "api",
+		},
+		{
+			Service:      "postgres",
+			Namespace:    "database",
+			LocalPort:    5432,
+			RemotePort:   5432,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "db",
+		},
+	}
+
+	result := convertToKPortal(kftrayConfigs)
+
+	assert.Len(t, result.Contexts, 1)
+	assert.Len(t, result.Contexts[0].Namespaces, 2)
+
+	// Namespaces should be sorted alphabetically
+	assert.Equal(t, "database", result.Contexts[0].Namespaces[0].Name)
+	assert.Equal(t, "default", result.Contexts[0].Namespaces[1].Name)
+}
+
+func TestConvertToKPortal_MultipleForwardsInNamespace(t *testing.T) {
+	kftrayConfigs := []KFTrayConfig{
+		{
+			Service:      "api",
+			Namespace:    "default",
+			LocalPort:    8080,
+			RemotePort:   8080,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "api",
+		},
+		{
+			Service:      "postgres",
+			Namespace:    "default",
+			LocalPort:    5432,
+			RemotePort:   5432,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "db",
+		},
+		{
+			Service:      "redis",
+			Namespace:    "default",
+			LocalPort:    6379,
+			RemotePort:   6379,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "redis",
+		},
+	}
+
+	result := convertToKPortal(kftrayConfigs)
+
+	assert.Len(t, result.Contexts, 1)
+	assert.Len(t, result.Contexts[0].Namespaces, 1)
+	assert.Len(t, result.Contexts[0].Namespaces[0].Forwards, 3)
+
+	// Forwards should be sorted by local port
+	forwards := result.Contexts[0].Namespaces[0].Forwards
+	assert.Equal(t, 5432, forwards[0].LocalPort) // postgres
+	assert.Equal(t, 6379, forwards[1].LocalPort) // redis
+	assert.Equal(t, 8080, forwards[2].LocalPort) // api
+}
+
+func TestConvertToKPortal_PodWorkloadType(t *testing.T) {
+	kftrayConfigs := []KFTrayConfig{
+		{
+			Service:      "my-app",
+			Namespace:    "default",
+			LocalPort:    8080,
+			RemotePort:   8080,
+			Context:      "production",
+			WorkloadType: "pod",
+			Protocol:     "tcp",
+			Alias:        "app",
+		},
+	}
+
+	result := convertToKPortal(kftrayConfigs)
+
+	forward := result.Contexts[0].Namespaces[0].Forwards[0]
+	assert.Equal(t, "pod/my-app", forward.Resource)
+}
+
+func TestConvertToKPortal_WithoutAlias(t *testing.T) {
+	kftrayConfigs := []KFTrayConfig{
+		{
+			Service:      "postgres",
+			Namespace:    "default",
+			LocalPort:    5432,
+			RemotePort:   5432,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "", // No alias
+		},
+	}
+
+	result := convertToKPortal(kftrayConfigs)
+
+	forward := result.Contexts[0].Namespaces[0].Forwards[0]
+	assert.Equal(t, "", forward.Alias)
+}
+
+func TestConvertToKPortal_DifferentPorts(t *testing.T) {
+	kftrayConfigs := []KFTrayConfig{
+		{
+			Service:      "api",
+			Namespace:    "default",
+			LocalPort:    8080,
+			RemotePort:   3000,
+			Context:      "production",
+			WorkloadType: "service",
+			Protocol:     "tcp",
+			Alias:        "api",
+		},
+	}
+
+	result := convertToKPortal(kftrayConfigs)
+
+	forward := result.Contexts[0].Namespaces[0].Forwards[0]
+	assert.Equal(t, 3000, forward.Port, "Remote port should be 3000")
+	assert.Equal(t, 8080, forward.LocalPort, "Local port should be 8080")
+}