101 lines
3.1 KiB
Go
101 lines
3.1 KiB
Go
package agent
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
"git.zacharyzhang.com/ZacharyZhang-NY/MujicaUI/core"
|
|
)
|
|
|
|
func cmdTester(run *CommandRun, opts CommandExecutionCardOptions, aborts *int) *ui.Tester {
|
|
opts.Still = true
|
|
return ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
if CommandExecutionCard(c, *run, opts).Changed() {
|
|
*aborts++
|
|
}
|
|
}, 600, 420)
|
|
}
|
|
|
|
func TestCommandExecutionCardRunningAbortAndFinish(t *testing.T) {
|
|
run := CommandRun{Command: "go test ./...", Dir: "~/court", Running: true, Output: "\x1b[32mok\x1b[0m court\n"}
|
|
aborts := 0
|
|
tt := cmdTester(&run, CommandExecutionCardOptions{}, &aborts)
|
|
for _, s := range []string{"go test ./...", "~/court", "ok", " court", "Running"} {
|
|
if !tt.HasText(s) {
|
|
t.Fatalf("missing %q in %q", s, tt.Texts())
|
|
}
|
|
}
|
|
if tt.HasText("\x1b[32mok") {
|
|
t.Fatal("escape codes shown as text")
|
|
}
|
|
_ = tt.Click("Abort")
|
|
if aborts != 1 {
|
|
t.Fatalf("aborts %d", aborts)
|
|
}
|
|
run.Output += "FAIL ledger\n"
|
|
run.Running, run.ExitCode, run.Duration = false, 1, 3200*time.Millisecond
|
|
tt.Frame()
|
|
if !tt.HasText("FAIL ledger") || !tt.HasText("Exit 1") || !tt.HasText("3.2 s") || tt.HasText("Abort") {
|
|
t.Fatalf("finished: %q", tt.Texts())
|
|
}
|
|
}
|
|
|
|
func TestCommandExecutionCardAnsiAcrossLinesAndReset(t *testing.T) {
|
|
var o commandOutput
|
|
o.update("\x1b[31mred\nstill red\x1b[0m plain\n")
|
|
if len(o.lines) != 2 || o.lines[1][0].Style.FG.Index != 1 || o.lines[1][1].Style.FG.Index != 0 {
|
|
t.Fatalf("lines %+v", o.lines)
|
|
}
|
|
o.update("\x1b[31mred\nstill red\x1b[0m plain\nnext\n")
|
|
if len(o.lines) != 3 || o.lines[2][0].Text != "next" {
|
|
t.Fatalf("append: %+v", o.lines)
|
|
}
|
|
o.update("fresh\n")
|
|
if len(o.lines) != 1 || o.lines[0][0].Text != "fresh" {
|
|
t.Fatalf("replace: %+v", o.lines)
|
|
}
|
|
}
|
|
|
|
func TestCommandExecutionCardDisabledAndEmpty(t *testing.T) {
|
|
run := CommandRun{Command: "sleep 10", Running: true}
|
|
aborts := 0
|
|
tt := cmdTester(&run, CommandExecutionCardOptions{Disabled: true}, &aborts)
|
|
_ = tt.Click("Abort")
|
|
tt.Key(0, ui.KeyTab)
|
|
tt.Key(0, ui.KeyEnter)
|
|
if aborts != 0 || !tt.HasText("No output") {
|
|
t.Fatalf("aborts %d texts %q", aborts, tt.Texts())
|
|
}
|
|
}
|
|
|
|
func TestCommandExecutionCardLargeOutputVirtualized(t *testing.T) {
|
|
var b strings.Builder
|
|
for i := range 100000 {
|
|
fmt.Fprintf(&b, "\x1b[2mline %d\x1b[0m\n", i)
|
|
}
|
|
run := CommandRun{Command: "yes", Output: b.String()}
|
|
n := 0
|
|
tt := cmdTester(&run, CommandExecutionCardOptions{}, &n)
|
|
if got := len(tt.Texts()); got > 200 {
|
|
t.Fatalf("%d texts built for 100000 lines", got)
|
|
}
|
|
if !tt.HasText("line 99999") {
|
|
t.Fatal("output does not follow its end")
|
|
}
|
|
}
|
|
|
|
func TestCommandExecutionCardSnapshot(t *testing.T) {
|
|
snapshot(t, "command_execution_card", 560, 300, func(c *ui.Context) {
|
|
run := CommandRun{Command: "go test ./court/...", Dir: "~/projects/court", ExitCode: 1, Duration: 4200 * time.Millisecond,
|
|
Output: "\x1b[32mok\x1b[0m court/guests 0.21s\n\x1b[1;31mFAIL\x1b[0m court/ledger 0.08s\n ledger_test.go:14: want 42 guests, got 40\n\x1b[33mwarning:\x1b[0m 1 test skipped\n"}
|
|
ui.Column(c).Padding(16).Children(func() {
|
|
CommandExecutionCard(c, run, CommandExecutionCardOptions{Height: 140})
|
|
})
|
|
})
|
|
}
|