The view shown in the zone list and on the zone page now links to that
view. The zone page and changeset page get breadcrumbs, and the zone list
keeps its filters in the address so going back restores them. Views can
be given a colour (Views > Edit, `jarl view color`, PUT /views/{id}/color)
so zones of the same view are easy to spot; the colour is UI-only, kept
outside changesets, and needs no deploy.
Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
82 lines
3.6 KiB
Go
82 lines
3.6 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
)
|
|
|
|
func TestViewColors(t *testing.T) {
|
|
e := newAPIEnv(t)
|
|
in := e.do("POST", "/views", e.token, map[string]any{"name": "internal", "match_clients": []string{"10.0.0.0/8"}}, 201)["object"].(map[string]any)
|
|
ex := e.do("POST", "/views", e.token, map[string]any{"name": "external"}, 201)["object"].(map[string]any)
|
|
inID, exID := int64(in["id"].(float64)), int64(ex["id"].(float64))
|
|
applied := len(e.list("/changesets?status=applied", e.token))
|
|
|
|
v := e.do("PUT", "/views/internal/color", e.token, map[string]any{"color": "green"}, 200)
|
|
if v["color"] != "green" || v["name"] != "internal" {
|
|
t.Errorf("set colour = %v", v)
|
|
}
|
|
e.do("PUT", "/views/"+itoa(exID)+"/color", e.token, map[string]any{"color": "purple"}, 200)
|
|
got := map[string]any{}
|
|
for _, v := range e.list("/views", e.token) {
|
|
got[v["name"].(string)] = v["color"]
|
|
}
|
|
if got["internal"] != "green" || got["external"] != "purple" {
|
|
t.Errorf("listed colours = %v", got)
|
|
}
|
|
if v := e.do("GET", "/views/internal", e.token, nil, 200); v["color"] != "green" {
|
|
t.Errorf("get view colour = %v", v["color"])
|
|
}
|
|
// A colour is not configuration: no changeset.
|
|
if n := len(e.list("/changesets?status=applied", e.token)); n != applied {
|
|
t.Errorf("colour changes made changesets: %d -> %d", applied, n)
|
|
}
|
|
if v := e.do("PATCH", "/views/internal", e.token, map[string]any{"comment": "LAN"}, 200)["object"].(map[string]any); v["color"] != "green" {
|
|
t.Errorf("colour after update = %v", v["color"])
|
|
}
|
|
if _, status := e.req("PUT", "/views/internal/color", e.token, map[string]any{"color": "#ff0000"}); status != 400 {
|
|
t.Errorf("bad colour = %d", status)
|
|
}
|
|
if _, status := e.req("PUT", "/views/nope/color", e.token, map[string]any{"color": "red"}); status != 404 {
|
|
t.Errorf("unknown view = %d", status)
|
|
}
|
|
|
|
// Readers of views can't change colours.
|
|
e.do("POST", "/users", e.token, map[string]any{"username": "reader", "password": "reader-password-123", "grant_strings": []string{"views:read"}}, 201)
|
|
var readerID int64
|
|
for _, u := range e.list("/users", e.token) {
|
|
if u["username"] == "reader" {
|
|
readerID = int64(u["id"].(float64))
|
|
}
|
|
}
|
|
rt := e.do("POST", "/users/"+itoa(readerID)+"/tokens", e.token, map[string]any{"name": "t"}, 201)["token"].(string)
|
|
if _, status := e.req("PUT", "/views/internal/color", rt, map[string]any{"color": "red"}); status != 403 {
|
|
t.Errorf("reader set a colour: %d", status)
|
|
}
|
|
|
|
// A view created in a draft can be given a colour before it is applied.
|
|
cs := itoa(int64(e.do("POST", "/changesets", e.token, map[string]any{"description": "dmz"}, 201)["id"].(float64)))
|
|
e.do("POST", "/views?changeset="+cs, e.token, map[string]any{"name": "dmz"}, 202)
|
|
if _, status := e.req("PUT", "/views/dmz/color", e.token, map[string]any{"color": "orange"}); status != 404 {
|
|
t.Errorf("draft view found outside its changeset: %d", status)
|
|
}
|
|
e.do("PUT", "/views/dmz/color?changeset="+cs, e.token, map[string]any{"color": "orange"}, 200)
|
|
e.do("PUT", "/views/internal/color", e.token, map[string]any{"color": "teal"}, 200)
|
|
e.do("POST", "/changesets/"+cs+"/apply", e.token, nil, 200)
|
|
if v := e.do("GET", "/views/dmz", e.token, nil, 200); v["color"] != "orange" {
|
|
t.Errorf("draft view colour after apply = %v", v["color"])
|
|
}
|
|
|
|
// "" goes back to the default.
|
|
if v := e.do("PUT", "/views/internal/color", e.token, map[string]any{"color": ""}, 200); v["color"] != nil {
|
|
t.Errorf("reset colour = %v", v["color"])
|
|
}
|
|
colors, err := e.st.ViewColors(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, ok := colors[inID]; ok {
|
|
t.Errorf("stored colours after reset = %v", colors)
|
|
}
|
|
}
|