timeline_test.go

package tachymeter_test

import (
	"io/ioutil"
	"os"
	"testing"
	"time"

	"github.com/develersrl/tachymeter"
)

func newTestTachy() *tachymeter.Tachymeter {
	ta := tachymeter.New(&tachymeter.Config{Size: 4})

	for i := 12; i < 16; i++ {
		ta.AddTime(time.Duration(i) * time.Millisecond)
	}
	return ta
}

func BenchmarkWriteHTMLTo(b *testing.B) {
	ta := newTestTachy()
	tl := tachymeter.Timeline{}
	tl.AddEvent(ta.Calc())

	b.ReportAllocs()
	b.ResetTimer()
	w := ioutil.Discard

	for i := 0; i < b.N; i++ {
		err := tl.WriteHTMLTo(w)
		if err != nil {
			b.Fatalf("can't write html: %s", err)
		}
	}
}

func TestTimelineEmptyMetrics(t *testing.T) {
	tmp, err := ioutil.TempDir("", t.Name())
	if err != nil {
		t.Fatal("TempDir failed: ", err)
	}
	defer os.RemoveAll(tmp)

	tline := &tachymeter.Timeline{}

	ta := tachymeter.New(&tachymeter.Config{Size: 30})
	m := ta.Calc()

	tline.AddEvent(m)

	defer func() {
		if r := recover(); r != nil {
			t.Errorf("WriteHtml panicked: %v", r)
		}
	}()

	tline.WriteHTML(tmp)
}