open-consul/agent/consul/server_overview_test.go

170 lines
5.3 KiB
Go

// Copyright (c) HashiCorp, Inc.
// SPDX-License-Identifier: MPL-2.0
package consul
import (
"testing"
"github.com/hashicorp/consul/agent/structs"
"github.com/hashicorp/consul/api"
"github.com/stretchr/testify/require"
)
func TestCatalogOverview(t *testing.T) {
cases := []struct {
name string
nodes []*structs.Node
services []*structs.ServiceNode
checks []*structs.HealthCheck
expected structs.CatalogSummary
}{
{
name: "empty",
expected: structs.CatalogSummary{},
},
{
name: "one node with no checks",
nodes: []*structs.Node{
{Node: "node1"},
},
expected: structs.CatalogSummary{
Nodes: []structs.HealthSummary{
{Total: 1, Passing: 1},
},
},
},
{
name: "one service with no checks",
services: []*structs.ServiceNode{
{Node: "node1", ServiceName: "service1"},
},
expected: structs.CatalogSummary{
Services: []structs.HealthSummary{
{Name: "service1", Total: 1, Passing: 1},
},
},
},
{
name: "three nodes with node checks",
nodes: []*structs.Node{
{Node: "node1"},
{Node: "node2"},
{Node: "node3"},
},
checks: []*structs.HealthCheck{
{Node: "node1", Name: "check1", CheckID: "check1", Status: api.HealthPassing},
{Node: "node2", Name: "check1", CheckID: "check1", Status: api.HealthWarning},
{Node: "node3", Name: "check1", CheckID: "check1", Status: api.HealthCritical},
},
expected: structs.CatalogSummary{
Nodes: []structs.HealthSummary{
{Total: 3, Passing: 1, Warning: 1, Critical: 1},
},
Checks: []structs.HealthSummary{
{Name: "check1", Total: 3, Passing: 1, Warning: 1, Critical: 1},
},
},
},
{
name: "three instances of one service with checks",
nodes: []*structs.Node{
{Node: "node1"},
},
services: []*structs.ServiceNode{
{Node: "node1", ServiceName: "service1", ServiceID: "id1"},
{Node: "node1", ServiceName: "service1", ServiceID: "id2"},
{Node: "node1", ServiceName: "service1", ServiceID: "id3"},
},
checks: []*structs.HealthCheck{
{Node: "node1", Name: "check1", CheckID: "check1", ServiceID: "id1", Status: api.HealthPassing},
{Node: "node1", Name: "check1", CheckID: "check2", ServiceID: "id2", Status: api.HealthWarning},
{Node: "node1", Name: "check1", CheckID: "check3", ServiceID: "id3", Status: api.HealthCritical},
},
expected: structs.CatalogSummary{
Nodes: []structs.HealthSummary{
{Total: 1, Passing: 1},
},
Services: []structs.HealthSummary{
{Name: "service1", Total: 3, Passing: 1, Warning: 1, Critical: 1},
},
Checks: []structs.HealthSummary{
{Name: "check1", Total: 3, Passing: 1, Warning: 1, Critical: 1},
},
},
},
{
name: "three instances of different services with checks",
nodes: []*structs.Node{
{Node: "node1"},
},
services: []*structs.ServiceNode{
{Node: "node1", ServiceName: "service1", ServiceID: "id1"},
{Node: "node1", ServiceName: "service2", ServiceID: "id2"},
{Node: "node1", ServiceName: "service3", ServiceID: "id3"},
},
checks: []*structs.HealthCheck{
{Node: "node1", Name: "check1", CheckID: "check1", ServiceID: "id1", Status: api.HealthPassing},
{Node: "node1", Name: "check1", CheckID: "check2", ServiceID: "id2", Status: api.HealthWarning},
{Node: "node1", Name: "check1", CheckID: "check3", ServiceID: "id3", Status: api.HealthCritical},
},
expected: structs.CatalogSummary{
Nodes: []structs.HealthSummary{
{Total: 1, Passing: 1},
},
Services: []structs.HealthSummary{
{Name: "service1", Total: 1, Passing: 1},
{Name: "service2", Total: 1, Warning: 1},
{Name: "service3", Total: 1, Critical: 1},
},
Checks: []structs.HealthSummary{
{Name: "check1", Total: 3, Passing: 1, Warning: 1, Critical: 1},
},
},
},
{
name: "many instances of the same check",
checks: []*structs.HealthCheck{
{Name: "check1", CheckID: "check1", Status: api.HealthPassing},
{Name: "check1", CheckID: "check2", Status: api.HealthWarning},
{Name: "check1", CheckID: "check3", Status: api.HealthCritical},
{Name: "check1", CheckID: "check4", Status: api.HealthPassing},
{Name: "check1", CheckID: "check5", Status: api.HealthCritical},
},
expected: structs.CatalogSummary{
Checks: []structs.HealthSummary{
{Name: "check1", Total: 5, Passing: 2, Warning: 1, Critical: 2},
},
},
},
{
name: "three different checks",
checks: []*structs.HealthCheck{
{Name: "check1", CheckID: "check1", Status: api.HealthPassing},
{Name: "check2", CheckID: "check2", Status: api.HealthWarning},
{Name: "check3", CheckID: "check3", Status: api.HealthCritical},
},
expected: structs.CatalogSummary{
Checks: []structs.HealthSummary{
{Name: "check1", Total: 1, Passing: 1},
{Name: "check2", Total: 1, Warning: 1},
{Name: "check3", Total: 1, Critical: 1},
},
},
},
}
for _, tc := range cases {
t.Run(tc.name, func(t *testing.T) {
summary := getCatalogOverview(&structs.CatalogContents{
Nodes: tc.nodes,
Services: tc.services,
Checks: tc.checks,
})
require.ElementsMatch(t, tc.expected.Nodes, summary.Nodes)
require.ElementsMatch(t, tc.expected.Services, summary.Services)
require.ElementsMatch(t, tc.expected.Checks, summary.Checks)
})
}
}