Fix: Make colon optional for reset (#559)

microsoft · Dec 12, 2024 · abdb3bf · abdb3bf
1 parent 58bc354
commit abdb3bf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/pkg/sqlcmd/commands.go b/pkg/sqlcmd/commands.go
@@ -77,7 +77,7 @@ func newCommands() Commands {
 			name:   "LISTVAR",
 		},
 		"RESET": {
-			regex:  regexp.MustCompile(`(?im)^[ \t]*:RESET(?:[ \t]+(.*$)|$)`),
+			regex:  regexp.MustCompile(`(?im)^[ \t]*?:?RESET(?:[ \t]+(.*$)|$)`),
 			action: resetCommand,
 			name:   "RESET",
 		},

diff --git a/pkg/sqlcmd/commands_test.go b/pkg/sqlcmd/commands_test.go
@@ -53,6 +53,8 @@ func TestCommandParsing(t *testing.T) {
 		{` !! dir c:\`, "EXEC", []string{` dir c:\`}},
 		{`!!dir c:\`, "EXEC", []string{`dir c:\`}},
 		{`:XML ON `, "XML", []string{`ON `}},
+		{`:RESET`, "RESET", []string{""}},
+		{`RESET`, "RESET", []string{""}},
 	}
 
 	for _, test := range commands {