对接

2026-09-23 17:42:55 +08:00
parent 53e8599bdd
commit 0a8ecbe09b
75 changed files with 2299 additions and 276 deletions
@@ -0,0 +1,71 @@
+package live
+
+import (
+	"testing"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/gorm"
+)
+
+func TestLiveClientPolicyMigrationAddsNewPrivateRoutes(t *testing.T) {
+	db, err := gorm.Open(sqlite.Open("file:"+t.Name()+"?mode=memory&cache=shared"), &gorm.Config{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err = db.Exec(`CREATE TABLE casbin_rule (
+		id integer primary key autoincrement,
+		ptype text,
+		v0 text,
+		v1 text,
+		v2 text,
+		v3 text,
+		v4 text,
+		v5 text
+	)`).Error; err != nil {
+		t.Fatal(err)
+	}
+
+	var migrate func(*gorm.DB) error
+	for _, step := range Definition().Migrations {
+		if step.ID == "202609220004_live_client_policies" {
+			migrate = step.Migrate
+			break
+		}
+	}
+	if migrate == nil {
+		t.Fatal("live client policy migration is missing")
+	}
+	if err = migrate(db); err != nil {
+		t.Fatal(err)
+	}
+	if err = migrate(db); err != nil {
+		t.Fatalf("policy migration is not idempotent: %v", err)
+	}
+
+	want := []struct{ method, path string }{
+		{method: "POST", path: "/live/client/auth/logout"},
+		{method: "GET", path: "/live/client/balance"},
+		{method: "GET", path: "/live/rooms/:id/follow"},
+		{method: "POST", path: "/live/rooms/:id/follow"},
+		{method: "DELETE", path: "/live/rooms/:id/follow"},
+		{method: "GET", path: "/live/recharge/orders/:orderNo"},
+		{method: "GET", path: "/live/recharge/orders/:orderNo/status"},
+	}
+	for _, route := range want {
+		var count int64
+		if err = db.Table("casbin_rule").Where("ptype = ? AND v0 = ? AND v1 = ? AND v2 = ?", "p", "1000", route.path, route.method).Count(&count).Error; err != nil {
+			t.Fatal(err)
+		}
+		if count != 1 {
+			t.Errorf("policy %s %s count = %d, want 1", route.method, route.path, count)
+		}
+	}
+
+	var publicProviderPolicies int64
+	if err = db.Table("casbin_rule").Where("v0 = ? AND v1 = ?", "1000", "/live/recharge/providers").Count(&publicProviderPolicies).Error; err != nil {
+		t.Fatal(err)
+	}
+	if publicProviderPolicies != 0 {
+		t.Fatalf("public provider discovery unexpectedly received a client policy: %d", publicProviderPolicies)
+	}
+}