mirror of
https://github.com/weaveworks/scope.git
synced 2026-03-03 02:00:43 +00:00
3be8cf71dd
* Pass Go context down to Renderers This is useful for cancellation or tracing. * Add tracing spans to app Also log things like number of nodes in Map, total number of reports.
69 lines
2.3 KiB
Go
69 lines
2.3 KiB
Go
package render_test
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/weaveworks/common/test"
|
|
"github.com/weaveworks/scope/probe/kubernetes"
|
|
"github.com/weaveworks/scope/render"
|
|
"github.com/weaveworks/scope/render/expected"
|
|
"github.com/weaveworks/scope/test/fixture"
|
|
"github.com/weaveworks/scope/test/reflect"
|
|
"github.com/weaveworks/scope/test/utils"
|
|
)
|
|
|
|
func TestPodRenderer(t *testing.T) {
|
|
have := utils.Prune(render.PodRenderer.Render(context.Background(), fixture.Report).Nodes)
|
|
want := utils.Prune(expected.RenderedPods)
|
|
if !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
}
|
|
|
|
var filterNonKubeSystem = render.Transformers([]render.Transformer{
|
|
render.Complement(render.IsNamespace("kube-system")),
|
|
render.FilterUnconnectedPseudo,
|
|
})
|
|
|
|
func TestPodFilterRenderer(t *testing.T) {
|
|
// tag on containers or pod namespace in the topology and ensure
|
|
// it is filtered out correctly.
|
|
input := fixture.Report.Copy()
|
|
input.Pod.Nodes[fixture.ClientPodNodeID] = input.Pod.Nodes[fixture.ClientPodNodeID].WithLatests(map[string]string{
|
|
kubernetes.Namespace: "kube-system",
|
|
})
|
|
|
|
have := utils.Prune(render.Render(context.Background(), input, render.PodRenderer, filterNonKubeSystem).Nodes)
|
|
want := utils.Prune(expected.RenderedPods.Copy())
|
|
delete(want, fixture.ClientPodNodeID)
|
|
if !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
}
|
|
|
|
func TestPodServiceRenderer(t *testing.T) {
|
|
have := utils.Prune(render.PodServiceRenderer.Render(context.Background(), fixture.Report).Nodes)
|
|
want := utils.Prune(expected.RenderedPodServices)
|
|
if !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
}
|
|
|
|
func TestPodServiceFilterRenderer(t *testing.T) {
|
|
// tag on containers or pod namespace in the topology and ensure
|
|
// it is filtered out correctly.
|
|
input := fixture.Report.Copy()
|
|
input.Service.Nodes[fixture.ServiceNodeID] = input.Service.Nodes[fixture.ServiceNodeID].WithLatests(map[string]string{
|
|
kubernetes.Namespace: "kube-system",
|
|
})
|
|
|
|
have := utils.Prune(render.Render(context.Background(), input, render.PodServiceRenderer, filterNonKubeSystem).Nodes)
|
|
want := utils.Prune(expected.RenderedPodServices.Copy())
|
|
delete(want, fixture.ServiceNodeID)
|
|
delete(want, render.IncomingInternetID)
|
|
if !reflect.DeepEqual(want, have) {
|
|
t.Error(test.Diff(want, have))
|
|
}
|
|
}
|