lerko
d982359f25
Press < / > to cycle sort column (Status, Name, Latency). Press r to reverse direction. Sorted column shows ▲/▼ arrow in the header. Groups always float to top. Sort applies to ungrouped monitors and group children independently. Default: Status descending (DOWN first).
96 lines
3.3 KiB
Go
96 lines
3.3 KiB
Go
package tui
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"gitea.lerkolabs.com/lerkolabs/uptop/internal/models"
|
|
)
|
|
|
|
func TestSortSitesForDisplay_GroupsFirst(t *testing.T) {
|
|
sites := []models.Site{
|
|
{SiteConfig: models.SiteConfig{ID: 3, Name: "ungrouped", Type: "http"}, SiteState: models.SiteState{Status: "UP"}},
|
|
{SiteConfig: models.SiteConfig{ID: 1, Name: "group-a", Type: "group"}, SiteState: models.SiteState{Status: "UP"}},
|
|
{SiteConfig: models.SiteConfig{ID: 2, Name: "child", Type: "http", ParentID: 1}, SiteState: models.SiteState{Status: "UP"}},
|
|
}
|
|
result := sortSitesForDisplay(sites, nil, sortStatus, false)
|
|
if len(result) != 3 {
|
|
t.Fatalf("expected 3 sites, got %d", len(result))
|
|
}
|
|
if result[0].Name != "group-a" {
|
|
t.Errorf("first should be group, got %s", result[0].Name)
|
|
}
|
|
if result[1].Name != "child" {
|
|
t.Errorf("second should be child, got %s", result[1].Name)
|
|
}
|
|
if result[2].Name != "ungrouped" {
|
|
t.Errorf("third should be ungrouped, got %s", result[2].Name)
|
|
}
|
|
}
|
|
|
|
func TestSortSitesForDisplay_CollapsedHidesChildren(t *testing.T) {
|
|
sites := []models.Site{
|
|
{SiteConfig: models.SiteConfig{ID: 1, Name: "group-a", Type: "group"}, SiteState: models.SiteState{Status: "UP"}},
|
|
{SiteConfig: models.SiteConfig{ID: 2, Name: "child-1", Type: "http", ParentID: 1}, SiteState: models.SiteState{Status: "UP"}},
|
|
{SiteConfig: models.SiteConfig{ID: 3, Name: "child-2", Type: "http", ParentID: 1}, SiteState: models.SiteState{Status: "UP"}},
|
|
}
|
|
collapsed := map[int]bool{1: true}
|
|
result := sortSitesForDisplay(sites, collapsed, sortStatus, false)
|
|
if len(result) != 1 {
|
|
t.Fatalf("collapsed group should hide children, got %d items", len(result))
|
|
}
|
|
if result[0].Name != "group-a" {
|
|
t.Errorf("only group should remain, got %s", result[0].Name)
|
|
}
|
|
}
|
|
|
|
func TestSortSitesForDisplay_StatusOrdering(t *testing.T) {
|
|
sites := []models.Site{
|
|
{SiteConfig: models.SiteConfig{ID: 1, Name: "up-site", Type: "http"}, SiteState: models.SiteState{Status: "UP"}},
|
|
{SiteConfig: models.SiteConfig{ID: 2, Name: "down-site", Type: "http"}, SiteState: models.SiteState{Status: "DOWN"}},
|
|
{SiteConfig: models.SiteConfig{ID: 3, Name: "late-site", Type: "http"}, SiteState: models.SiteState{Status: "LATE"}},
|
|
}
|
|
result := sortSitesForDisplay(sites, nil, sortStatus, false)
|
|
if result[0].Status != "DOWN" {
|
|
t.Errorf("DOWN should sort first, got %s", result[0].Status)
|
|
}
|
|
if result[1].Status != "LATE" {
|
|
t.Errorf("LATE should sort second, got %s", result[1].Status)
|
|
}
|
|
if result[2].Status != "UP" {
|
|
t.Errorf("UP should sort third, got %s", result[2].Status)
|
|
}
|
|
}
|
|
|
|
func TestFilterSites(t *testing.T) {
|
|
sites := []models.Site{
|
|
{SiteConfig: models.SiteConfig{Name: "Production API"}},
|
|
{SiteConfig: models.SiteConfig{Name: "Staging API"}},
|
|
{SiteConfig: models.SiteConfig{Name: "Database"}},
|
|
}
|
|
|
|
tests := []struct {
|
|
needle string
|
|
want int
|
|
}{
|
|
{"api", 2},
|
|
{"API", 2},
|
|
{"database", 1},
|
|
{"nonexistent", 0},
|
|
{"", 3},
|
|
}
|
|
for _, tt := range tests {
|
|
got := filterSites(sites, tt.needle)
|
|
if len(got) != tt.want {
|
|
t.Errorf("filterSites(%q) returned %d, want %d", tt.needle, len(got), tt.want)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestFilterSites_EmptyNeedle(t *testing.T) {
|
|
sites := []models.Site{{SiteConfig: models.SiteConfig{Name: "a"}}, {SiteConfig: models.SiteConfig{Name: "b"}}}
|
|
got := filterSites(sites, "")
|
|
if len(got) != 2 {
|
|
t.Errorf("empty needle should return all, got %d", len(got))
|
|
}
|
|
}
|