109 lines
4.3 KiB
Go
109 lines
4.3 KiB
Go
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 TestListEnabledMapTileSources(t *testing.T) {
|
|
st := openTestStore(t)
|
|
defer st.Close()
|
|
|
|
disabled, err := st.CreateMapTileSource(mapTileSourceInput{Name: "Disabled", URLTemplate: "https://disabled.example.com/{z}/{x}/{y}.png", MaxZoom: 18, Enabled: false})
|
|
if err != nil {
|
|
t.Fatalf("CreateMapTileSource(disabled) error = %v", err)
|
|
}
|
|
custom, err := st.CreateMapTileSource(mapTileSourceInput{Name: "Custom", URLTemplate: "https://custom.example.com/{z}/{x}/{y}.png", MaxZoom: 18, Enabled: true})
|
|
if err != nil {
|
|
t.Fatalf("CreateMapTileSource(custom) error = %v", err)
|
|
}
|
|
if _, err := st.SetDefaultMapTileSource(custom.ID); err != nil {
|
|
t.Fatalf("SetDefaultMapTileSource() error = %v", err)
|
|
}
|
|
|
|
rows, err := st.ListEnabledMapTileSources()
|
|
if err != nil {
|
|
t.Fatalf("ListEnabledMapTileSources() error = %v", err)
|
|
}
|
|
if len(rows) < 2 {
|
|
t.Fatalf("ListEnabledMapTileSources() length = %d, want at least 2", len(rows))
|
|
}
|
|
if rows[0].ID != custom.ID {
|
|
t.Fatalf("first enabled source id = %d, want default %d", rows[0].ID, custom.ID)
|
|
}
|
|
for _, row := range rows {
|
|
if row.ID == disabled.ID {
|
|
t.Fatalf("disabled source was returned: %+v", row)
|
|
}
|
|
if !row.Enabled {
|
|
t.Fatalf("disabled row returned: %+v", row)
|
|
}
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|