weave-scope/report/table_test.go

package report_test

import (
	"fmt"
	"reflect"
	"testing"

	"github.com/weaveworks/scope/report"
	"github.com/weaveworks/scope/test"
)

func TestPrefixTables(t *testing.T) {
	want := map[string]string{
		"foo1": "bar1",
		"foo2": "bar2",
	}
	nmd := report.MakeNode("foo1")

	nmd = nmd.AddPrefixTable("foo_", want)
	have, truncationCount := nmd.ExtractTable(report.TableTemplate{Prefix: "foo_"})

	if truncationCount != 0 {
		t.Error("Table shouldn't had been truncated")
	}

	if !reflect.DeepEqual(want, have) {
		t.Error(test.Diff(want, have))
	}
}

func TestFixedTables(t *testing.T) {
	want := map[string]string{
		"foo1": "bar1",
		"foo2": "bar2",
	}
	nmd := report.MakeNodeWith("foo1", map[string]string{
		"foo1key": "bar1",
		"foo2key": "bar2",
	})

	template := report.TableTemplate{FixedRows: map[string]string{
		"foo1key": "foo1",
		"foo2key": "foo2",
	},
	}

	have, _ := nmd.ExtractTable(template)

	if !reflect.DeepEqual(want, have) {
		t.Error(test.Diff(want, have))
	}
}

func TestTruncation(t *testing.T) {
	wantTruncationCount := 1
	want := map[string]string{}
	for i := 0; i < report.MaxTableRows+wantTruncationCount; i++ {
		key := fmt.Sprintf("key%d", i)
		value := fmt.Sprintf("value%d", i)
		want[key] = value
	}

	nmd := report.MakeNode("foo1")

	nmd = nmd.AddPrefixTable("foo_", want)
	_, truncationCount := nmd.ExtractTable(report.TableTemplate{Prefix: "foo_"})

	if truncationCount != wantTruncationCount {
		t.Error(
			"Table should had been truncated by",
			wantTruncationCount,
			"and not",
			truncationCount,
		)
	}
}