chore: refresh gramsrv public release

internal/rpc/callback_registry_test.go

@@ -0,0 +1,77 @@
+package rpc
+
+import (
+	"testing"
+
+	"telesrv/internal/domain"
+)
+
+func TestCallbackRegistryResolveDelivers(t *testing.T) {
+	reg := newCallbackRegistry()
+	queryID, p := reg.register(100, 200)
+	if queryID == 0 {
+		t.Fatal("query id must be non-zero")
+	}
+	want := domain.BotCallbackAnswer{Alert: true, Message: "hi"}
+	if !reg.resolve(100, queryID, want) {
+		t.Fatal("resolve by owner bot must succeed")
+	}
+	select {
+	case got := <-p.ch:
+		if got != want {
+			t.Fatalf("delivered %+v, want %+v", got, want)
+		}
+	default:
+		t.Fatal("answer not delivered to channel")
+	}
+	// 解挂后条目已删除：再次 resolve 失败、registry 计数归零。
+	if reg.resolve(100, queryID, want) {
+		t.Fatal("second resolve must fail (already deregistered)")
+	}
+	if n := reg.size(); n != 0 {
+		t.Fatalf("registry size = %d, want 0 after resolve", n)
+	}
+}
+
+func TestCallbackRegistryNonOwnerRejected(t *testing.T) {
+	reg := newCallbackRegistry()
+	queryID, p := reg.register(100, 200)
+	// 非属主 bot（999）不得解挂（I6）。
+	if reg.resolve(999, queryID, domain.BotCallbackAnswer{Message: "evil"}) {
+		t.Fatal("non-owner resolve must be rejected")
+	}
+	select {
+	case <-p.ch:
+		t.Fatal("non-owner answer must not be delivered")
+	default:
+	}
+	if n := reg.size(); n != 1 {
+		t.Fatalf("registry size = %d, want 1 (entry retained after rejected resolve)", n)
+	}
+	reg.deregister(queryID)
+	if n := reg.size(); n != 0 {
+		t.Fatalf("registry size = %d, want 0 after deregister", n)
+	}
+}
+
+func TestCallbackRegistryUnknownQuery(t *testing.T) {
+	reg := newCallbackRegistry()
+	if reg.resolve(100, 12345, domain.BotCallbackAnswer{}) {
+		t.Fatal("resolve of unregistered query must fail")
+	}
+}
+
+func TestCallbackRegistryUniqueQueryIDs(t *testing.T) {
+	reg := newCallbackRegistry()
+	seen := make(map[int64]struct{})
+	for i := 0; i < 1000; i++ {
+		q, _ := reg.register(1, 2)
+		if _, dup := seen[q]; dup {
+			t.Fatalf("duplicate query id %d", q)
+		}
+		seen[q] = struct{}{}
+	}
+	if n := reg.size(); n != 1000 {
+		t.Fatalf("registry size = %d, want 1000", n)
+	}
+}