mirror of
https://github.com/kristoferssolo/telescope-frecency.nvim.git
synced 2025-10-21 20:10:38 +00:00
8f593064f7
* feat: add function to query the DB * docs: add documentation for frecency.query() * test: fix tests to run with `timestamps` property * test: add tests for frecency.query()
75 lines
2.1 KiB
Lua
75 lines
2.1 KiB
Lua
local FS = require "frecency.fs"
|
|
local Database = require "frecency.database"
|
|
local config = require "frecency.config"
|
|
local async = require "plenary.async" --[[@as FrecencyPlenaryAsync]]
|
|
local util = require "frecency.tests.util"
|
|
async.tests.add_to_env()
|
|
|
|
---@param datetime string?
|
|
---@return integer
|
|
local function make_epoch(datetime)
|
|
if not datetime then
|
|
return os.time()
|
|
end
|
|
local tz_fix = datetime:gsub("+(%d%d):(%d%d)$", "+%1%2")
|
|
return util.time_piece(tz_fix)
|
|
end
|
|
|
|
local function with_database(f)
|
|
local fs = FS.new { ignore_patterns = {} }
|
|
local dir, close = util.tmpdir()
|
|
dir:joinpath("file_frecency.bin"):touch()
|
|
return function()
|
|
config.setup { db_root = dir.filename }
|
|
local database = Database.new(fs)
|
|
f(database)
|
|
close()
|
|
end
|
|
end
|
|
|
|
---@async
|
|
---@param database FrecencyDatabase
|
|
---@param tbl table<string, FrecencyDatabaseRecordValue>
|
|
---@param epoch integer
|
|
---@return FrecencyEntry[]
|
|
local function save_and_load(database, tbl, epoch)
|
|
database:raw_save(util.v1_table(tbl))
|
|
async.util.sleep(100)
|
|
local entries = database:get_entries(nil, epoch)
|
|
table.sort(entries, function(a, b)
|
|
return a.path < b.path
|
|
end)
|
|
return entries
|
|
end
|
|
|
|
a.describe("frecency.database", function()
|
|
a.describe("updated by another process", function()
|
|
a.it(
|
|
"returns valid entries",
|
|
---@param database FrecencyDatabase
|
|
with_database(function(database)
|
|
assert.are.same(
|
|
{
|
|
{
|
|
path = "hoge1.txt",
|
|
count = 1,
|
|
ages = { 60 },
|
|
timestamps = { make_epoch "2023-08-21T00:00:00+09:00" },
|
|
},
|
|
{
|
|
path = "hoge2.txt",
|
|
count = 1,
|
|
ages = { 60 },
|
|
timestamps = { make_epoch "2023-08-21T00:00:00+09:00" },
|
|
},
|
|
},
|
|
save_and_load(database, {
|
|
["hoge1.txt"] = { count = 1, timestamps = { make_epoch "2023-08-21T00:00:00+09:00" } },
|
|
["hoge2.txt"] = { count = 1, timestamps = { make_epoch "2023-08-21T00:00:00+09:00" } },
|
|
}, make_epoch "2023-08-21T01:00:00+09:00")
|
|
)
|
|
end)
|
|
)
|
|
end)
|
|
end)
|