nvim/plugin: add CommandEvalHandler testcases

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

Author: zchee

nvim/plugin/plugin_test.go

@@ -13,6 +13,10 @@ import (
 func TestRegister(t *testing.T) {
 	t.Parallel()
 
+	type testEval struct {
+		BaseDir string `eval:"fnamemodify(getcwd(), ':t')"`
+	}
+
 	p := plugin.New(nvimtest.NewChildProcess(t))
 
 	// SimpleHandler
@@ -31,16 +35,14 @@ func TestRegister(t *testing.T) {
 	)
 
 	// FunctionEvalHandler
-	type testEval struct {
-		BaseDir string `eval:"fnamemodify(getcwd(), ':t')"`
-	}
 	p.HandleFunction(
 		&plugin.FunctionOptions{Name: "TestEval", Eval: "*"},
 		func(_ []string, eval *testEval) (string, error) {
 			return eval.BaseDir, nil
 		},
 	)
 
+	// CommandHandler
 	p.HandleCommand(
 		&plugin.CommandOptions{
 			Name:     "Hello",
@@ -66,6 +68,23 @@ func TestRegister(t *testing.T) {
 		},
 	)
 
+	// CommandEvalHandler
+	p.HandleCommand(
+		&plugin.CommandOptions{
+			Name: "HelloEval",
+			Eval: "*",
+		},
+		func(n *nvim.Nvim, eval *testEval) error {
+			chunks := []nvim.TextChunk{
+				{
+					Text: eval.BaseDir,
+				},
+			}
+
+			return n.Echo(chunks, true, make(map[string]interface{}))
+		},
+	)
+
 	if err := p.RegisterForTests(); err != nil {
 		t.Fatalf("register for test: %v", err)
 	}
@@ -118,6 +137,18 @@ func TestRegister(t *testing.T) {
 			t.Fatalf("Hello returned %q, want %q", result, expected)
 		}
 	})
+
+	t.Run("CommandEvalHandler", func(t *testing.T) {
+		result, err := p.Nvim.Exec(`HelloEval`, true)
+		if err != nil {
+			t.Fatalf("exec 'Hello' command: %v", err)
+		}
+
+		expected := `plugin`
+		if result != expected {
+			t.Fatalf("Hello returned %q, want %q", result, expected)
+		}
+	})
 }
 
 func TestSubscribe(t *testing.T) {