79 lines
2.6 KiB
Go
79 lines
2.6 KiB
Go
package mujica
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
)
|
|
|
|
var waterfallSample = []WaterfallSpan{
|
|
{ID: "db", Parent: "api", Name: "SELECT guests", Service: "postgres", Start: 40 * time.Millisecond, Duration: 90 * time.Millisecond},
|
|
{ID: "root", Name: "GET /ball", Service: "gateway", Duration: 320 * time.Millisecond},
|
|
{ID: "api", Parent: "root", Name: "ListGuests", Service: "api", Start: 20 * time.Millisecond, Duration: 200 * time.Millisecond},
|
|
{ID: "pay", Parent: "root", Name: "Charge", Service: "payments", Start: 230 * time.Millisecond, Duration: 80 * time.Millisecond, Error: true},
|
|
}
|
|
|
|
func traceTester(spans []WaterfallSpan, changes *int) *ui.Tester {
|
|
return ui.NewTester(func(c *ui.Context) {
|
|
Use(c, Settings{})
|
|
if TraceWaterfall(c, spans, TraceWaterfallOptions{}).Changed() {
|
|
*changes++
|
|
}
|
|
}, 700, 460)
|
|
}
|
|
|
|
func TestTraceWaterfallOrderAndDetail(t *testing.T) {
|
|
n := 0
|
|
tt := traceTester(waterfallSample, &n)
|
|
if !tt.HasText("4 spans · 320 ms") {
|
|
t.Fatalf("summary: %q", tt.Texts())
|
|
}
|
|
root, _ := tt.Find("GET /ball")
|
|
api, _ := tt.Find("ListGuests")
|
|
db, _ := tt.Find("SELECT guests")
|
|
pay, _ := tt.Find("Charge")
|
|
if !(root.Y < api.Y && api.Y < db.Y && db.Y < pay.Y && db.X > api.X && api.X > root.X) {
|
|
t.Fatalf("tree order: root %v api %v db %v pay %v", root, api, db, pay)
|
|
}
|
|
_ = tt.Click("Charge")
|
|
if n != 1 || !tt.HasText("payments · starts at 230 ms · takes 80 ms") {
|
|
t.Fatalf("detail: %q", tt.Texts())
|
|
}
|
|
tt.Key(0, ui.KeyUp)
|
|
tt.Key(0, ui.KeyEnter)
|
|
if !tt.HasText("postgres · starts at 40 ms · takes 90 ms") {
|
|
t.Fatalf("keyboard detail: %q", tt.Texts())
|
|
}
|
|
}
|
|
|
|
func TestTraceWaterfallRulesAndMany(t *testing.T) {
|
|
n := 0
|
|
tt := traceTester(nil, &n)
|
|
if !tt.HasText("No spans") {
|
|
t.Fatal("empty")
|
|
}
|
|
many := make([]WaterfallSpan, 100000)
|
|
many[0] = WaterfallSpan{ID: "0", Name: "root", Duration: time.Second}
|
|
for i := 1; i < len(many); i++ {
|
|
many[i] = WaterfallSpan{ID: fmt.Sprint(i), Parent: "0", Name: fmt.Sprint("span ", i), Start: time.Duration(i) * time.Microsecond, Duration: time.Microsecond}
|
|
}
|
|
tt = traceTester(many, &n)
|
|
if got := len(tt.Texts()); got > 400 {
|
|
t.Fatalf("%d texts for 100000 spans", got)
|
|
}
|
|
defer func() {
|
|
if recover() == nil {
|
|
t.Fatal("cycle did not panic")
|
|
}
|
|
}()
|
|
traceTester([]WaterfallSpan{{ID: "a", Parent: "b", Name: "a"}, {ID: "b", Parent: "a", Name: "b"}}, &n)
|
|
}
|
|
|
|
func TestTraceWaterfallSnapshot(t *testing.T) {
|
|
snapshot(t, "trace_waterfall", 680, 280, func(c *ui.Context) {
|
|
ui.Column(c).Padding(16).Children(func() { TraceWaterfall(c, waterfallSample, TraceWaterfallOptions{Height: 180}) })
|
|
})
|
|
}
|