fix(client): sync local catalog from client setup

2026-09-30 19:43:51 +08:00
parent 99d553753e
commit d57c90962d
3 changed files with 352 additions and 2 deletions
@@ -0,0 +1,99 @@
+package setup
+
+import (
+	"bytes"
+	"encoding/json"
+	"os"
+	"path/filepath"
+	"testing"
+)
+
+func TestValidateLocalCatalogRequiresEveryRemoteBundle(t *testing.T) {
+	release := t.TempDir()
+	path := filepath.Join("nested", "current.bundle")
+	if err := os.MkdirAll(filepath.Join(release, "nested"), 0o755); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.WriteFile(filepath.Join(release, path), []byte("bundle"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	raw, err := json.Marshal(map[string]any{"m_InternalIds": []string{
+		remoteCatalogPrefix + `StandaloneWindows64\HD\version\nested/current.bundle`,
+	}})
+	if err != nil {
+		t.Fatal(err)
+	}
+	if err := validateLocalCatalog(raw, release); err != nil {
+		t.Fatal(err)
+	}
+	if err := os.Remove(filepath.Join(release, path)); err != nil {
+		t.Fatal(err)
+	}
+	if err := validateLocalCatalog(raw, release); err == nil {
+		t.Fatal("catalog with missing bundle unexpectedly passed")
+	}
+}
+
+func TestCatalogBundlePathRejectsTraversal(t *testing.T) {
+	if _, err := catalogBundlePath(remoteCatalogPrefix + `StandaloneWindows64\HD\version\..\escape.bundle`); err == nil {
+		t.Fatal("catalog traversal path unexpectedly passed")
+	}
+}
+
+func TestReplaceCatalogReplacesExistingFile(t *testing.T) {
+	path := filepath.Join(t.TempDir(), "catalog.json")
+	if err := os.WriteFile(path, []byte("old"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	if err := replaceCatalog(path, []byte("new")); err != nil {
+		t.Fatal(err)
+	}
+	got, err := os.ReadFile(path)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if string(got) != "new" {
+		t.Fatalf("catalog=%q", got)
+	}
+}
+
+func TestLocalizeCatalogPreservesMetadataAndCreatesHardLink(t *testing.T) {
+	release := t.TempDir()
+	aa := t.TempDir()
+	relative := filepath.Join("nested", "current.bundle")
+	if err := os.MkdirAll(filepath.Join(release, "nested"), 0o755); err != nil {
+		t.Fatal(err)
+	}
+	source := filepath.Join(release, relative)
+	if err := os.WriteFile(source, []byte("bundle"), 0o600); err != nil {
+		t.Fatal(err)
+	}
+	raw, err := json.Marshal(map[string]any{
+		"m_BuildResultHash": "current-metadata",
+		"m_InternalIds": []string{
+			remoteCatalogPrefix + `StandaloneWindows64\HD\version\nested/current.bundle`,
+		},
+	})
+	if err != nil {
+		t.Fatal(err)
+	}
+	localized, err := localizeCatalog(raw, release, aa)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !bytes.Contains(localized, []byte(`"m_BuildResultHash":"current-metadata"`)) ||
+		!bytes.Contains(localized, []byte(`Addressables.RuntimePath`)) {
+		t.Fatalf("localized catalog=%s", localized)
+	}
+	sourceInfo, err := os.Stat(source)
+	if err != nil {
+		t.Fatal(err)
+	}
+	destinationInfo, err := os.Stat(filepath.Join(aa, relative))
+	if err != nil {
+		t.Fatal(err)
+	}
+	if !os.SameFile(sourceInfo, destinationInfo) {
+		t.Fatal("localized bundle is not a hard link to the selected release")
+	}
+}