Trading: components 161-180

level2_quotes_test.go

@@ -0,0 +1,63 @@
+package mujica
+
+import (
+	"bytes"
+	"testing"
+
+	"github.com/egoist/mygo/ui"
+)
+
+func level2Sample() Level2Data {
+	return Level2Data{
+		Bids: []L2Level{{100.00, 500, 12}, {99.99, 300, 4}},
+		Asks: []L2Level{{100.01, 200, 3}, {100.02, 900, 21}, {100.03, 50, 1}},
+	}
+}
+
+func TestLevel2QuotesLevels(t *testing.T) {
+	tt := tradingTester(GreenUp, 600, 300, func(c *ui.Context) {
+		ui.Box(c).Width(560).Children(func() { Level2Quotes(c, level2Sample(), Level2QuotesOptions{Levels: 3}) })
+	})
+	for _, name := range []string{"B1 100.00 500 12 orders", "S2 100.02 900 21 orders"} {
+		if _, ok := tt.Find(name); !ok {
+			t.Fatalf("no row %q in %q", name, tt.Texts())
+		}
+	}
+	if !tt.HasText("B3") || !tt.HasText("—") {
+		t.Fatal("missing third bid not shown as empty")
+	}
+}
+
+// A changed level marks its size move and flashes.
+func TestLevel2QuotesHighlightsChangedLevel(t *testing.T) {
+	data := level2Sample()
+	var marks *map[string]*level2Mark
+	tt := tradingTester(GreenUp, 600, 300, func(c *ui.Context) {
+		e := Level2Quotes(c, data, Level2QuotesOptions{Levels: 2})
+		marks = ui.Local(e, "marks", func() map[string]*level2Mark { return nil })
+	})
+	before := bytes.Clone(tt.Image().Pix)
+	data.Bids = []L2Level{{100.00, 650, 13}, {99.99, 300, 4}}
+	tt.Frame()
+	m := (*marks)["1 0"]
+	if m.dir != 1 || m.at.IsZero() {
+		t.Fatalf("bid 1 mark %+v", m)
+	}
+	if (*marks)["1 1"].dir != 0 || !(*marks)["1 1"].at.IsZero() {
+		t.Fatal("unchanged level marked")
+	}
+	if bytes.Equal(before, tt.Image().Pix) {
+		t.Fatal("nothing drawn for the change")
+	}
+	data.Bids = []L2Level{{100.01, 650, 13}, {99.99, 300, 4}}
+	tt.Frame()
+	if (*marks)["1 0"].dir != 0 {
+		t.Fatal("new price read as a size move")
+	}
+}
+
+func TestLevel2QuotesSnapshot(t *testing.T) {
+	snapshot(t, "level2_quotes", 600, 230, func(c *ui.Context) {
+		ui.Column(c).Padding(16).Children(func() { Level2Quotes(c, level2Sample(), Level2QuotesOptions{}) })
+	})
+}