新增图源修改功能

2026-06-06 01:16:03 +08:00
parent 91267eb99c
commit 0f9cb3eae5
15 changed files with 1267 additions and 24 deletions
@@ -0,0 +1,72 @@
+package main
+
+import (
+	"errors"
+	"testing"
+)
+
+func TestMapTileSourceDefaultSeeded(t *testing.T) {
+	st := openTestStore(t)
+	defer st.Close()
+
+	row, err := st.GetDefaultMapTileSource()
+	if err != nil {
+		t.Fatalf("GetDefaultMapTileSource() error = %v", err)
+	}
+	if row.Name != defaultMapTileSourceName || row.URLTemplate != defaultMapTileSourceURLTemplate || !row.Enabled || !row.IsDefault {
+		t.Fatalf("default map source = %+v, want built-in default", row)
+	}
+}
+
+func TestCreateMapTileSourceValidation(t *testing.T) {
+	st := openTestStore(t)
+	defer st.Close()
+
+	if _, err := st.CreateMapTileSource(mapTileSourceInput{Name: "bad", URLTemplate: "https://tiles.example.com/{z}/{x}.png", MaxZoom: 19, Enabled: true}); err == nil {
+		t.Fatal("CreateMapTileSource() missing placeholder error = nil, want error")
+	}
+	if _, err := st.CreateMapTileSource(mapTileSourceInput{Name: "bad", URLTemplate: "javascript:alert(1)/{z}/{x}/{y}", MaxZoom: 19, Enabled: true}); err == nil {
+		t.Fatal("CreateMapTileSource() invalid scheme error = nil, want error")
+	}
+	if _, err := st.CreateMapTileSource(mapTileSourceInput{Name: "bad", URLTemplate: "https://user:pass@tiles.example.com/{z}/{x}/{y}.png", MaxZoom: 19, Enabled: true}); err == nil {
+		t.Fatal("CreateMapTileSource() credentials error = nil, want error")
+	}
+}
+
+func TestMapTileSourceDuplicateAndDefaultRules(t *testing.T) {
+	st := openTestStore(t)
+	defer st.Close()
+
+	first, err := st.CreateMapTileSource(mapTileSourceInput{Name: "Custom", URLTemplate: "https://tiles.example.com/{z}/{x}/{y}.png", MaxZoom: 18, Enabled: true})
+	if err != nil {
+		t.Fatalf("CreateMapTileSource() error = %v", err)
+	}
+	if _, err := st.CreateMapTileSource(mapTileSourceInput{Name: "Custom", URLTemplate: "https://tiles2.example.com/{z}/{x}/{y}.png", MaxZoom: 18, Enabled: true}); !errors.Is(err, errMapTileSourceAlreadyExists) {
+		t.Fatalf("duplicate name error = %v, want errMapTileSourceAlreadyExists", err)
+	}
+	if _, err := st.CreateMapTileSource(mapTileSourceInput{Name: "Custom 2", URLTemplate: first.URLTemplate, MaxZoom: 18, Enabled: true}); !errors.Is(err, errMapTileSourceAlreadyExists) {
+		t.Fatalf("duplicate url error = %v, want errMapTileSourceAlreadyExists", err)
+	}
+
+	updated, err := st.SetDefaultMapTileSource(first.ID)
+	if err != nil {
+		t.Fatalf("SetDefaultMapTileSource() error = %v", err)
+	}
+	if !updated.IsDefault {
+		t.Fatalf("updated default = %+v, want is_default", updated)
+	}
+
+	oldDefault, err := st.GetDefaultMapTileSource()
+	if err != nil {
+		t.Fatalf("GetDefaultMapTileSource() error = %v", err)
+	}
+	if oldDefault.ID != first.ID {
+		t.Fatalf("default id = %d, want %d", oldDefault.ID, first.ID)
+	}
+	if _, err := st.UpdateMapTileSource(first.ID, mapTileSourceInput{Name: first.Name, URLTemplate: first.URLTemplate, Attribution: first.Attribution, MaxZoom: first.MaxZoom, Enabled: false, IsDefault: true}); !errors.Is(err, errMapTileSourceCannotDisableDefault) {
+		t.Fatalf("disable default error = %v, want errMapTileSourceCannotDisableDefault", err)
+	}
+	if err := st.DeleteMapTileSource(first.ID); !errors.Is(err, errMapTileSourceCannotDeleteDefault) {
+		t.Fatalf("delete default error = %v, want errMapTileSourceCannotDeleteDefault", err)
+	}
+}