From e96b454b230ae33a76939491c55d2a629e8040f4 Mon Sep 17 00:00:00 2001
From: Marty Oehme <marty.oehme@gmail.com>
Date: Fri, 30 Oct 2020 13:48:17 +0100
Subject: [PATCH] Add simple zettel retrieval function

---
 lua/zettelkasten/list.lua      |  8 +++++
 lua/zettelkasten/list_spec.lua | 54 +++++++++++++++++++++-------------
 2 files changed, 41 insertions(+), 21 deletions(-)

diff --git a/lua/zettelkasten/list.lua b/lua/zettelkasten/list.lua
index 28cd7e4..056b063 100644
--- a/lua/zettelkasten/list.lua
+++ b/lua/zettelkasten/list.lua
@@ -31,4 +31,12 @@ function ls.get_anchors_and_paths(path, recursive, options)
     return zettel
 end
 
+-- Returns the path to the zettel defined by the anchor argument.
+-- Take a list of zettel as an optional variable, without which
+-- it will use the (recursive) results of the zettel_root directory.
+function ls.get_zettel(anchor, all)
+    return all[anchor]
+end
+
+
 return ls
diff --git a/lua/zettelkasten/list_spec.lua b/lua/zettelkasten/list_spec.lua
index c1ae1aa..11ffd6d 100644
--- a/lua/zettelkasten/list_spec.lua
+++ b/lua/zettelkasten/list_spec.lua
@@ -1,29 +1,30 @@
 ls = require'zettelkasten.list'
 -- these tests, I suppose, only work on unix due to the file structure
 
-describe("zettel listing", function()
+local function simple_api_mock(files)
+    return {
+        g = {},
+        b = {},
+        loop = {
+            fs_scandir = function()
+                if #files == 0 then
+                    return false
+                else
+                    return true
+                end
+            end,
+            fs_scandir_next = function()
+                return table.remove(files)
+            end
+        }
+    }
+end
+
+describe("get_anchors_and_paths", function()
     before_each(function()
-        _G.get_api_mock = function(files)
-            return {
-                g = {},
-                b = {},
-                loop = {
-                    fs_scandir = function()
-                        if #files == 0 then
-                            return false
-                        else
-                            return true
-                        end
-                    end,
-                    fs_scandir_next = function()
-                        return table.remove(files)
-                    end
-                }
-            }
-        end
+        get_api_mock = simple_api_mock
     end)
     after_each(function()
-        _G.get_api_mock = nil
         _G.vim = nil
     end)
 
@@ -138,5 +139,16 @@ describe("zettel listing", function()
         assert.same(expected, ls.get_anchors_and_paths('mydirectory', false, vim.g))
 
     end)
-
+end)
+
+describe("get_zettel", function()
+    it("should return the correct zettel by id", function()
+        local file_list = {
+            ["1910291645"] = "1910291645 myfile.md",
+            ["2345678901"] = "2345678901 another.md",
+        }
+        _G.vim = simple_api_mock(file_list)
+
+        assert.same("1910291645 myfile.md", ls.get_zettel("1910291645", file_list))
+    end)
 end)