Adds tests to yanking values

cuducos · Nov 3, 2023 · be021cc · be021cc
1 parent e1b4060
commit be021cc
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/yaml_nvim/commands_spec.lua b/tests/yaml_nvim/commands_spec.lua
@@ -16,4 +16,24 @@ describe("commands", function()
 
 		assert(vim.fn.getreg("7"), "worldcup")
 	end)
+
+	it("yanks value to the default register", function()
+		require("yaml_nvim")
+
+		vim.cmd(":e sample.yaml")
+		vim.cmd("norm 10j")
+		vim.cmd(":YAMLYankValue")
+
+		assert(vim.fn.getreg('"'), "7")
+	end)
+
+	it("yanks value to a custom register", function()
+		require("yaml_nvim")
+
+		vim.cmd(":e sample.yaml")
+		vim.cmd("norm 10j")
+		vim.cmd(":YAMLYankValue 7")
+
+		assert(vim.fn.getreg("7"), "7")
+	end)
 end)