fix: load uid key from manager config

server/uid_key_config_test.go

@@ -0,0 +1,69 @@
+package main
+
+import (
+	"encoding/json"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+)
+
+func TestStoreConfigWithAdminUIDKey(t *testing.T) {
+	const managerUIDKey = "MTIzNDU2Nzg5MGFiY2RlZg=="
+	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if r.URL.Path != "/internal/im-config/third-party/im" {
+			t.Errorf("unexpected path: %s", r.URL.Path)
+			return
+		}
+		if r.Header.Get(internalHeaderSignature) == "" {
+			t.Error("internal request must be signed")
+			return
+		}
+		_ = json.NewEncoder(w).Encode(map[string]any{
+			"code": 0,
+			"data": map[string]any{
+				"values": map[string]string{
+					"uid_encryption_key": managerUIDKey,
+				},
+			},
+		})
+	}))
+	defer server.Close()
+
+	client := &adminInternalClient{
+		config: adminInternalAPIConfig{BaseURL: server.URL, ServiceName: "im-server", Secret: "test-secret"},
+		http:   server.Client(),
+	}
+	merged, ok, err := storeConfigWithAdminUIDKey(json.RawMessage(`{"uid_key":"bG9jYWwtZmFsbGJhY2s=","max_results":1024}`), client)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !ok {
+		t.Fatal("expected uid key to be loaded from manager")
+	}
+	var config struct {
+		UIDKey     []byte `json:"uid_key"`
+		MaxResults int    `json:"max_results"`
+	}
+	if err := json.Unmarshal(merged, &config); err != nil {
+		t.Fatal(err)
+	}
+	if string(config.UIDKey) != "1234567890abcdef" {
+		t.Fatalf("unexpected uid key: %q", string(config.UIDKey))
+	}
+	if config.MaxResults != 1024 {
+		t.Fatalf("max_results should be preserved, got %d", config.MaxResults)
+	}
+}
+
+func TestStoreConfigWithAdminUIDKeyFallsBackWhenMissing(t *testing.T) {
+	merged, ok, err := storeConfigWithAdminUIDKey(json.RawMessage(`{"uid_key":"bG9jYWw="}`), nil)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if ok {
+		t.Fatal("nil admin client should not override uid key")
+	}
+	if string(merged) != `{"uid_key":"bG9jYWw="}` {
+		t.Fatalf("unexpected config: %s", string(merged))
+	}
+}