mirror of
https://github.com/mum4k/termdash.git
synced 2025-04-28 13:48:51 +08:00
551 lines
14 KiB
Go
551 lines
14 KiB
Go
package sparkline
|
|
|
|
import (
|
|
"image"
|
|
"testing"
|
|
|
|
"github.com/kylelemons/godebug/pretty"
|
|
"github.com/mum4k/termdash/canvas"
|
|
"github.com/mum4k/termdash/canvas/testcanvas"
|
|
"github.com/mum4k/termdash/cell"
|
|
"github.com/mum4k/termdash/draw"
|
|
"github.com/mum4k/termdash/draw/testdraw"
|
|
"github.com/mum4k/termdash/terminal/faketerm"
|
|
"github.com/mum4k/termdash/widgetapi"
|
|
)
|
|
|
|
func TestSparkLine(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
sparkLine *SparkLine
|
|
update func(*SparkLine) error // update gets called before drawing of the widget.
|
|
canvas image.Rectangle
|
|
want func(size image.Point) *faketerm.Terminal
|
|
wantUpdateErr bool // whether to expect an error on a call to the update function
|
|
wantDrawErr bool
|
|
}{
|
|
{
|
|
desc: "draws empty for no data points",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return nil
|
|
},
|
|
canvas: image.Rect(0, 0, 1, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
return faketerm.MustNew(size)
|
|
},
|
|
},
|
|
{
|
|
desc: "fails on negative data points",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 3, -1, 2)
|
|
},
|
|
canvas: image.Rect(0, 0, 1, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
return faketerm.MustNew(size)
|
|
},
|
|
},
|
|
{
|
|
desc: "single height sparkline",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 1, 2, 3, 4, 5, 6, 7)
|
|
},
|
|
canvas: image.Rect(0, 0, 8, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, " ▁▂▃▄▅▆▇", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "sets sparkline color",
|
|
sparkLine: New(
|
|
Color(cell.ColorMagenta),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 1, 2, 3, 4, 5, 6, 7)
|
|
},
|
|
canvas: image.Rect(0, 0, 8, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, " ▁▂▃▄▅▆▇", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(cell.ColorMagenta),
|
|
))
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "full cell has background set",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(8)
|
|
},
|
|
canvas: image.Rect(0, 0, 1, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "█", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "full cell has background set to custom color",
|
|
sparkLine: New(
|
|
Color(cell.ColorMagenta),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(8)
|
|
},
|
|
canvas: image.Rect(0, 0, 1, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "█", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(cell.ColorMagenta),
|
|
cell.BgColor(cell.ColorMagenta),
|
|
))
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "draws data points from the right",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(6, 7)
|
|
},
|
|
canvas: image.Rect(0, 0, 9, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, " ▆▇", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "single height sparkline with label",
|
|
sparkLine: New(
|
|
Label("Hello"),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 1, 2, 3, 7, 3, 2, 0, 1)
|
|
},
|
|
canvas: image.Rect(0, 0, 9, 2),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "Hello", image.Point{0, 0})
|
|
testdraw.MustText(c, " ▁▂▃▇▃▂ ▁", image.Point{0, 1}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "too long label is trimmed",
|
|
sparkLine: New(
|
|
Label("Hello world"),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(7)
|
|
},
|
|
canvas: image.Rect(0, 0, 9, 2),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "Hello wo…", image.Point{0, 0})
|
|
testdraw.MustText(c, " ▇", image.Point{0, 1}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "stretches up to the height of the container",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 100, 50, 85)
|
|
},
|
|
canvas: image.Rect(0, 0, 4, 4),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, " █ ▃", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{1, 0}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testdraw.MustText(c, " █ █", image.Point{0, 1}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{1, 1}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
testcanvas.MustSetCell(c, image.Point{3, 1}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testdraw.MustText(c, " ███", image.Point{0, 2}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testdraw.MustRectangle(c, image.Rect(1, 2, 4, 3),
|
|
draw.RectChar('█'),
|
|
draw.RectCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
),
|
|
)
|
|
testdraw.MustText(c, " ███", image.Point{0, 3}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testdraw.MustRectangle(c, image.Rect(1, 3, 4, 4),
|
|
draw.RectChar('█'),
|
|
draw.RectCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
),
|
|
)
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "stretches up to the height of the container with label",
|
|
sparkLine: New(
|
|
Label("Hello"),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 90, 30, 85)
|
|
},
|
|
canvas: image.Rect(0, 0, 9, 4),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "Hello", image.Point{0, 0})
|
|
testdraw.MustText(c, " █ ▇", image.Point{0, 1}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{1, 1}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testdraw.MustText(c, " █ █", image.Point{0, 2}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{1, 2}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
testcanvas.MustSetCell(c, image.Point{3, 2}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testdraw.MustText(c, " ███", image.Point{0, 3}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testdraw.MustRectangle(c, image.Rect(1, 3, 4, 4),
|
|
draw.RectChar('█'),
|
|
draw.RectCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
),
|
|
)
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "respects fixed height",
|
|
sparkLine: New(
|
|
Height(2),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 100, 50, 85)
|
|
},
|
|
canvas: image.Rect(0, 0, 9, 4),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, " █ ▆", image.Point{0, 2}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{1, 2}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testdraw.MustText(c, " ███", image.Point{0, 3}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testdraw.MustRectangle(c, image.Rect(1, 3, 4, 4),
|
|
draw.RectChar('█'),
|
|
draw.RectCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
),
|
|
)
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "respects fixed height with label",
|
|
sparkLine: New(
|
|
Label("zoo"),
|
|
Height(2),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 100, 50, 85)
|
|
},
|
|
canvas: image.Rect(0, 0, 9, 4),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "zoo", image.Point{0, 1}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testdraw.MustText(c, " █ ▆", image.Point{0, 2}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{1, 2}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testdraw.MustText(c, " ███", image.Point{0, 3}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testdraw.MustRectangle(c, image.Rect(1, 3, 4, 4),
|
|
draw.RectChar('█'),
|
|
draw.RectCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
),
|
|
)
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "sets label color",
|
|
sparkLine: New(
|
|
Label(
|
|
"Hello",
|
|
cell.FgColor(cell.ColorBlue),
|
|
cell.BgColor(cell.ColorYellow),
|
|
),
|
|
),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 1)
|
|
},
|
|
canvas: image.Rect(0, 0, 9, 2),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "Hello", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(cell.ColorBlue),
|
|
cell.BgColor(cell.ColorYellow),
|
|
))
|
|
testdraw.MustText(c, " ▁", image.Point{0, 1}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "displays only data points that fit the width",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(0, 1, 2, 3, 4, 5, 6, 7, 8)
|
|
},
|
|
canvas: image.Rect(0, 0, 3, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "▆▇█", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{2, 0}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
{
|
|
desc: "data points not visible don't affect the determined max data point",
|
|
sparkLine: New(),
|
|
update: func(sl *SparkLine) error {
|
|
return sl.Add(10, 4, 8)
|
|
},
|
|
canvas: image.Rect(0, 0, 2, 1),
|
|
want: func(size image.Point) *faketerm.Terminal {
|
|
ft := faketerm.MustNew(size)
|
|
c := testcanvas.MustNew(ft.Area())
|
|
|
|
testdraw.MustText(c, "▄█", image.Point{0, 0}, draw.TextCellOpts(
|
|
cell.FgColor(DefaultColor),
|
|
))
|
|
testcanvas.MustSetCell(c, image.Point{1, 0}, '█',
|
|
cell.FgColor(DefaultColor),
|
|
cell.BgColor(DefaultColor),
|
|
)
|
|
|
|
testcanvas.MustApply(c, ft)
|
|
return ft
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.desc, func(t *testing.T) {
|
|
c, err := canvas.New(tc.canvas)
|
|
if err != nil {
|
|
t.Fatalf("canvas.New => unexpected error: %v", err)
|
|
}
|
|
|
|
err = tc.update(tc.sparkLine)
|
|
if (err != nil) != tc.wantUpdateErr {
|
|
t.Errorf("update => unexpected error: %v, wantUpdateErr: %v", err, tc.wantUpdateErr)
|
|
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
err = tc.sparkLine.Draw(c)
|
|
if (err != nil) != tc.wantDrawErr {
|
|
t.Errorf("Draw => unexpected error: %v, wantDrawErr: %v", err, tc.wantDrawErr)
|
|
}
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
got, err := faketerm.New(c.Size())
|
|
if err != nil {
|
|
t.Fatalf("faketerm.New => unexpected error: %v", err)
|
|
}
|
|
|
|
if err := c.Apply(got); err != nil {
|
|
t.Fatalf("Apply => unexpected error: %v", err)
|
|
}
|
|
|
|
if diff := faketerm.Diff(tc.want(c.Size()), got); diff != "" {
|
|
t.Errorf("Draw => %v", diff)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestOptions(t *testing.T) {
|
|
tests := []struct {
|
|
desc string
|
|
sparkLine *SparkLine
|
|
want widgetapi.Options
|
|
}{
|
|
{
|
|
desc: "no label and no fixed height",
|
|
sparkLine: New(),
|
|
want: widgetapi.Options{
|
|
MinimumSize: image.Point{1, 1},
|
|
WantKeyboard: false,
|
|
WantMouse: false,
|
|
},
|
|
},
|
|
{
|
|
desc: "label and no fixed height",
|
|
sparkLine: New(
|
|
Label("foo"),
|
|
),
|
|
want: widgetapi.Options{
|
|
MinimumSize: image.Point{1, 2},
|
|
WantKeyboard: false,
|
|
WantMouse: false,
|
|
},
|
|
},
|
|
{
|
|
desc: "no label and fixed height",
|
|
sparkLine: New(
|
|
Height(3),
|
|
),
|
|
want: widgetapi.Options{
|
|
MinimumSize: image.Point{1, 3},
|
|
MaximumSize: image.Point{1, 3},
|
|
WantKeyboard: false,
|
|
WantMouse: false,
|
|
},
|
|
},
|
|
{
|
|
desc: "label and fixed height",
|
|
sparkLine: New(
|
|
Label("foo"),
|
|
Height(3),
|
|
),
|
|
want: widgetapi.Options{
|
|
MinimumSize: image.Point{1, 4},
|
|
MaximumSize: image.Point{1, 4},
|
|
WantKeyboard: false,
|
|
WantMouse: false,
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.desc, func(t *testing.T) {
|
|
got := tc.sparkLine.Options()
|
|
if diff := pretty.Compare(tc.want, got); diff != "" {
|
|
t.Errorf("Options => unexpected diff (-want, +got):\n%s", diff)
|
|
}
|
|
|
|
})
|
|
}
|
|
}
|