实现 gost 网关

web/services/channel_gost_test.go

@@ -0,0 +1,74 @@
+package services
+
+import (
+	"testing"
+
+	m "platform/web/models"
+)
+
+func TestExpandGostEdgesRejectsEmpty(t *testing.T) {
+	_, err := expandGostEdges(nil, 1)
+	if err == nil {
+		t.Fatal("expected error, got nil")
+	}
+}
+
+func TestExpandGostEdgesReusesWhenInsufficient(t *testing.T) {
+	edges := []*m.Edge{
+		{Mac: "chain-a"},
+		{Mac: "chain-b"},
+	}
+
+	result, err := expandGostEdges(edges, 5)
+	if err != nil {
+		t.Fatalf("expandGostEdges returned error: %v", err)
+	}
+	if len(result) != 5 {
+		t.Fatalf("unexpected edge count: %d", len(result))
+	}
+
+	expected := []string{"chain-a", "chain-b", "chain-a", "chain-b", "chain-a"}
+	for i, edge := range result {
+		if edge.Mac != expected[i] {
+			t.Fatalf("unexpected edge at %d: %s", i, edge.Mac)
+		}
+	}
+}
+
+func TestEdgeFilterIsEmpty(t *testing.T) {
+	if !(*EdgeFilter)(nil).IsEmpty() {
+		t.Fatal("nil filter should be empty")
+	}
+	if (&EdgeFilter{}).IsEmpty() != true {
+		t.Fatal("empty filter should be empty")
+	}
+	if (&EdgeFilter{Prov: strPtr("")}).IsEmpty() != true {
+		t.Fatal("filter with empty province should be empty")
+	}
+	if (&EdgeFilter{City: strPtr("")}).IsEmpty() != true {
+		t.Fatal("filter with empty city should be empty")
+	}
+	if (&EdgeFilter{Isp: ispPtr(m.ToEdgeISP(0))}).IsEmpty() != true {
+		t.Fatal("filter with zero ISP should be empty")
+	}
+	if (&EdgeFilter{Isp: ispPtr(m.ToEdgeISP(99))}).IsEmpty() != true {
+		t.Fatal("filter with invalid ISP should be empty")
+	}
+
+	prov := "江苏"
+	if (&EdgeFilter{Prov: &prov}).IsEmpty() {
+		t.Fatal("filter with province should not be empty")
+	}
+	isp := m.EdgeISPTelecom
+	if (&EdgeFilter{Isp: &isp}).IsEmpty() {
+		t.Fatal("filter with valid ISP should not be empty")
+	}
+}
+
+func strPtr(v string) *string {
+	return &v
+}
+
+func ispPtr(v m.EdgeISP) *m.EdgeISP {
+	return &v
+}