105 lines
3.4 KiB
Go
105 lines
3.4 KiB
Go
package chart
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/egoist/mygo/ui"
|
|
|
|
"github.com/ZacharyZhang-NY/MujicaUI/core"
|
|
"github.com/ZacharyZhang-NY/MujicaUI/internal"
|
|
"github.com/ZacharyZhang-NY/MujicaUI/theme"
|
|
)
|
|
|
|
func TestDrawingToolsPlaceMoveDeleteUndo(t *testing.T) {
|
|
cs := sampleCandles(60)
|
|
v := MarketView{Count: 60}
|
|
d := DrawingTools{Tool: ToolHorizontal}
|
|
disabled := false
|
|
tt := ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
ui.Column(c).Disabled(disabled).Children(func() {
|
|
CandlestickChart(c, cs, &v, CandlestickOptions{Frame: ChartFrame{Label: "Drawing"}, Drawings: &d})
|
|
})
|
|
}, 640, 340)
|
|
tt.ClickAt(300, 100)
|
|
if len(d.Items) != 1 || d.Items[0].Tool != ToolHorizontal || d.Selected() != 0 {
|
|
t.Fatalf("drawings %+v", d)
|
|
}
|
|
before := d.Items[0].A.Price
|
|
tt.Key(ui.Alt, ui.KeyUp)
|
|
if d.Items[0].A.Price <= before {
|
|
t.Fatal("keyboard move failed")
|
|
}
|
|
tt.Key(0, ui.KeyDelete)
|
|
if len(d.Items) != 0 {
|
|
t.Fatal("delete failed")
|
|
}
|
|
tt.Key(ui.Cmd, ui.KeyZ)
|
|
if len(d.Items) != 1 {
|
|
t.Fatal("undo failed")
|
|
}
|
|
disabled = true
|
|
tt.Frame()
|
|
tt.Key(0, ui.KeyDelete)
|
|
tt.ClickAt(200, 150)
|
|
if len(d.Items) != 1 {
|
|
t.Fatal("disabled drawing changed")
|
|
}
|
|
}
|
|
func TestDrawingToolsSnapshot(t *testing.T) {
|
|
cs := sampleCandles(60)
|
|
v := MarketView{Count: 60}
|
|
d := DrawingTools{Items: []Drawing{{Tool: ToolTrendLine, A: DrawingPoint{cs[10].Time, 98}, B: DrawingPoint{cs[45].Time, 105}}, {Tool: ToolHorizontal, A: DrawingPoint{cs[20].Time, 100}}}, sel: 1}
|
|
snapshot(t, "drawing_tools", 640, 320, func(c *ui.Context) {
|
|
ui.Column(c).Padding(12).Children(func() {
|
|
CandlestickChart(c, cs, &v, CandlestickOptions{Frame: ChartFrame{Label: "Drawings", Height: 260}, Drawings: &d})
|
|
})
|
|
})
|
|
}
|
|
|
|
func TestDrawingToolsTrendLinePaint(t *testing.T) {
|
|
cs := sampleCandles(2)
|
|
img := ui.Render(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{Mode: core.Light})
|
|
m := &mktCore{c: c, k: core.Tokens(c), candles: cs, from: 0, to: 2}
|
|
g := mktGeom{plot: ui.Rect{W: 200, H: 200}, band: internal.Band{N: 2, R0: 0, R1: 200}, ys: internal.Linear{D0: 0, D1: 10, R0: 180, R1: 20}}
|
|
d := DrawingTools{Items: []Drawing{{Tool: ToolTrendLine, A: DrawingPoint{cs[0].Time, 0}, B: DrawingPoint{cs[1].Time, 10}}}}
|
|
ui.Box(c).Size(200, 200).Draw(func(p *ui.Painter, _ ui.Rect) { d.paint(p, g, m, 12) })
|
|
}, 200, 200, 1)
|
|
bg := theme.Light().Background
|
|
pixel := img.RGBAAt(100, 100)
|
|
if int(bg.R)-int(pixel.R) < 40 || int(bg.G)-int(pixel.G) < 20 {
|
|
t.Fatalf("diagonal midpoint lacks the stroke: %v", pixel)
|
|
}
|
|
}
|
|
|
|
func TestDrawingToolsAllTools(t *testing.T) {
|
|
for _, tool := range []DrawingTool{ToolTrendLine, ToolHorizontal, ToolFibonacci, ToolRectangle, ToolText} {
|
|
t.Run(string(rune('0'+tool)), func(t *testing.T) {
|
|
cs := sampleCandles(30)
|
|
v := MarketView{Count: 30}
|
|
d := DrawingTools{Tool: tool, Text: "Note"}
|
|
tt := ui.NewTester(func(c *ui.Context) {
|
|
core.Use(c, core.Settings{})
|
|
CandlestickChart(c, cs, &v, CandlestickOptions{Frame: ChartFrame{Label: "Drawing"}, Drawings: &d})
|
|
}, 640, 340)
|
|
tt.ClickAt(200, 120)
|
|
if tool.twoPoint() {
|
|
if d.pending == nil || len(d.Items) != 0 {
|
|
t.Fatal("missing first anchor")
|
|
}
|
|
tt.ClickAt(400, 160)
|
|
}
|
|
if len(d.Items) != 1 || d.Items[0].Tool != tool || d.Selected() != 0 {
|
|
t.Fatalf("created %+v", d.Items)
|
|
}
|
|
if tool == ToolText && d.Items[0].Text != "Note" {
|
|
t.Fatal("annotation text lost")
|
|
}
|
|
if !d.Clear() || len(d.Items) != 0 || !d.Undo() || len(d.Items) != 1 {
|
|
t.Fatal("clear or undo failed")
|
|
}
|
|
})
|
|
}
|
|
}
|