fix: add selection2link button for CreateScratchNoteCommand (#2496)

* fix: add selection2link button for CreateScratchNoteCommand * chore: add test
dendronhq · Feb 28, 2022 · a881757 · a881757
1 parent 2524589
commit a881757
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/packages/plugin-core/src/commands/CreateScratchNoteCommand.ts b/packages/plugin-core/src/commands/CreateScratchNoteCommand.ts
@@ -7,6 +7,7 @@ import {
   CopyNoteLinkBtn,
   HorizontalSplitBtn,
   ScratchBtn,
+  Selection2LinkBtn,
 } from "../components/lookup/buttons";
 import { CommandRunOpts as NoteLookupRunOpts } from "./NoteLookupCommand";
 import { AutoCompletableRegistrar } from "../utils/registers/AutoCompletableRegistrar";
@@ -44,6 +45,7 @@ export class CreateScratchNoteCommand extends BasicCommand<
       vaultButtonPressed,
       extraButtons: [
         ScratchBtn.create({ pressed: true, canToggle: false }),
+        Selection2LinkBtn.create(true),
         CopyNoteLinkBtn.create(false),
         HorizontalSplitBtn.create(false),
       ],

diff --git a/packages/plugin-core/src/test/suite-integ/CreateScratchNoteCommand.test.ts b/packages/plugin-core/src/test/suite-integ/CreateScratchNoteCommand.test.ts
@@ -32,6 +32,24 @@ suite("CreateScratchNoteCommand", function () {
 
         expect(activeNote.fname.startsWith("scratch.")).toBeTruthy();
       });
+      test("THEN selection2link is applied.", async () => {
+        const ext = ExtensionProvider.getExtension();
+        const wsUtils = ext.wsUtils;
+        const cmd = new CreateScratchNoteCommand(ext);
+        const { vaults, engine } = ext.getDWorkspace();
+        const note = NoteUtils.getNoteByFnameFromEngine({
+          fname: "foo",
+          vault: vaults[0],
+          engine,
+        }) as NoteProps;
+        const fooNoteEditor = await wsUtils.openNote(note);
+        fooNoteEditor.selection = new vscode.Selection(7, 0, 7, 12);
+        await cmd.run({ noConfirm: true });
+        const activeNote = getNoteFromTextEditor();
+        expect(activeNote.fname.endsWith(".foo-body")).toBeTruthy();
+        const changedFooNoteText = fooNoteEditor.document.getText();
+        expect(changedFooNoteText.endsWith(".foo-body]]\n"));
+      });
     }
   );
 });