multiline yank writes to 0 register; fixes VSCodeVim#1214

jkillian · Oct 2, 2018 · afbfccc · afbfccc
1 parent bd153ad
commit afbfccc
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 3 deletions.
diff --git a/src/register/register.ts b/src/register/register.ts
@@ -302,9 +302,7 @@ export class Register {
    */
   private static processNumberedRegister(content: RegisterContent, vimState: VimState): void {
     // Find the BaseOperator of the current actions
-    const baseOperator = vimState.recordedState.actionsRun.find(value => {
-      return value instanceof BaseOperator || value instanceof BaseCommand;
-    });
+    const baseOperator = vimState.recordedState.operator || vimState.recordedState.command;
 
     if (baseOperator instanceof YankOperator || baseOperator instanceof CommandYankFullLine) {
       // 'yank' to 0 only if no register was specified

diff --git a/test/register/register.test.ts b/test/register/register.test.ts
@@ -105,6 +105,28 @@ suite('register', () => {
     assertEqualLines(['test2', 'test2', 'test3']);
   });
 
+  test("Multiline yank stores text in Register '0'", async () => {
+    modeHandler.vimState.editor = vscode.window.activeTextEditor!;
+
+    await modeHandler.handleMultipleKeyEvents('itest1\ntest2\ntest3'.split(''));
+
+    await modeHandler.handleMultipleKeyEvents([
+      '<Esc>',
+      'g',
+      'g',
+      '2',
+      'y',
+      'y',
+      'd',
+      'd',
+      '"',
+      '0',
+      'P',
+    ]);
+
+    assertEqualLines(['test1', 'test2', 'test2', 'test3']);
+  });
+
   test("Register '1'-'9' stores delete content", async () => {
     modeHandler.vimState.editor = vscode.window.activeTextEditor!;