nvim/plugin: add Subscribe and Unsubscribe testcase

Signed-off-by: Koichi Shiraishi <[email protected]>

Author: zchee

nvim/plugin/register_test.go

@@ -120,3 +120,95 @@ func TestRegister(t *testing.T) {
 		}
 	}
 }
+
+func TestSubscribe(t *testing.T) {
+	p, cleanup := newChildProcess(t)
+	t.Cleanup(func() {
+		cleanup()
+	})
+
+	const event1 = "event1"
+	eventFn1 := func(t *testing.T, v *nvim.Nvim) error {
+		return v.RegisterHandler(event1, func(event ...interface{}) {
+			if event[0] != int64(1) {
+				t.Fatalf("expected event[0] is 1 but got %d", event[0])
+			}
+			if event[1] != int64(2) {
+				t.Fatalf("expected event[1] is 2 but got %d", event[1])
+			}
+			if event[2] != int64(3) {
+				t.Fatalf("expected event[2] is 3 but got %d", event[2])
+			}
+		})
+	}
+	const event2 = "event2"
+	eventFn2 := func(t *testing.T, v *nvim.Nvim) error {
+		return v.RegisterHandler(event1, func(event ...interface{}) {
+			if event[0] != int64(4) {
+				t.Fatalf("expected event[0] is 4 but got %d", event[0])
+			}
+			if event[1] != int64(5) {
+				t.Fatalf("expected event[1] is 5 but got %d", event[1])
+			}
+			if event[2] != int64(6) {
+				t.Fatalf("expected event[2] is 6 but got %d", event[2])
+			}
+		})
+	}
+	p.Handle(event1, func() error { return eventFn1(t, p.Nvim) })
+	p.Handle(event2, func() error { return eventFn2(t, p.Nvim) })
+
+	if err := p.RegisterForTests(); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := p.Nvim.Subscribe(event1); err != nil {
+		t.Fatal(err)
+	}
+	b := p.Nvim.NewBatch()
+	b.Subscribe(event2)
+	if err := b.Execute(); err != nil {
+		t.Fatal(err)
+	}
+
+	// warm-up
+	var result int
+	if err := p.Nvim.Eval(fmt.Sprintf(`rpcnotify(0, %q)`, event1), &result); err != nil {
+		t.Fatal(err)
+	}
+	if result != 1 {
+		t.Fatalf("expect 1 but got %d", result)
+	}
+
+	var result2 int
+	if err := p.Nvim.Eval(fmt.Sprintf(`rpcnotify(0, %q, 1, 2, 3)`, event1), &result2); err != nil {
+		t.Fatal(err)
+	}
+	if result2 != 1 {
+		t.Fatalf("expect 1 but got %d", result2)
+	}
+
+	var result3 int
+	if err := p.Nvim.Eval(fmt.Sprintf(`rpcnotify(0, %q, 4, 5, 6)`, event2), &result3); err != nil {
+		t.Fatal(err)
+	}
+	if result3 != 1 {
+		t.Fatalf("expect 1 but got %d", result3)
+	}
+
+	if err := p.Nvim.Unsubscribe(event1); err != nil {
+		t.Fatal(err)
+	}
+	b.Unsubscribe(event2)
+	if err := b.Execute(); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := p.Nvim.Eval(fmt.Sprintf(`rpcnotify(0, %q, 7, 8, 9)`, event1), nil); err != nil {
+		t.Fatal(err)
+	}
+
+	if err := p.Nvim.Eval(fmt.Sprintf(`rpcnotify(0, %q, 10, 11, 12)`, event2), nil); err != nil {
+		t.Fatal(err)
+	}
+}