nvim/plugin: add CommandRangeDotHandler and CommandCountHandler tests

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

Author: zchee

nvim/plugin/plugin_test.go

@@ -56,9 +56,7 @@ func TestRegister(t *testing.T) {
 		},
 		func(n *nvim.Nvim, args []string) error {
 			chunks := []nvim.TextChunk{
-				{
-					Text: `Hello`,
-				},
+				{Text: `Hello`},
 			}
 			for _, arg := range args {
 				chunks = append(chunks, nvim.TextChunk{Text: arg})
@@ -68,6 +66,36 @@ func TestRegister(t *testing.T) {
 		},
 	)
 
+	// CommandRangeDotHandler
+	p.HandleCommand(
+		&plugin.CommandOptions{
+			Name:  "HelloRangeDot",
+			Range: ".",
+		},
+		func(n *nvim.Nvim) error {
+			chunks := []nvim.TextChunk{
+				{Text: `Hello`},
+			}
+
+			return n.Echo(chunks, true, make(map[string]interface{}))
+		},
+	)
+
+	// CommandCountHandler
+	p.HandleCommand(
+		&plugin.CommandOptions{
+			Name:  "HelloCount",
+			Count: "0",
+		},
+		func(n *nvim.Nvim) error {
+			chunks := []nvim.TextChunk{
+				{Text: `Hello`},
+			}
+
+			return n.Echo(chunks, true, make(map[string]interface{}))
+		},
+	)
+
 	// CommandEvalHandler
 	p.HandleCommand(
 		&plugin.CommandOptions{
@@ -86,7 +114,7 @@ func TestRegister(t *testing.T) {
 	)
 
 	if err := p.RegisterForTests(); err != nil {
-		t.Fatalf("register for test: %v", err)
+		t.Fatalf("register handlers for test: %v", err)
 	}
 
 	t.Run("SimpleHandler", func(t *testing.T) {
@@ -138,6 +166,30 @@ func TestRegister(t *testing.T) {
 		}
 	})
 
+	t.Run("CommandRangeDotHandler", func(t *testing.T) {
+		result, err := p.Nvim.Exec(`HelloRangeDot`, true)
+		if err != nil {
+			t.Fatalf("exec 'Hello' command: %v", err)
+		}
+
+		expected := `Hello`
+		if result != expected {
+			t.Fatalf("Hello returned %q, want %q", result, expected)
+		}
+	})
+
+	t.Run("CommandCountHandler", func(t *testing.T) {
+		result, err := p.Nvim.Exec(`HelloCount`, true)
+		if err != nil {
+			t.Fatalf("exec 'Hello' command: %v", err)
+		}
+
+		expected := `Hello`
+		if result != expected {
+			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 {