From 6af2b5011b4bde8089a03c24c393d3fdf04f0d7a Mon Sep 17 00:00:00 2001
From: Ade Attwood <adrian.attwood@marketdojo.com>
Date: Sun, 22 Oct 2023 15:18:41 +0100
Subject: [PATCH] chore: add benchmark for `set_items`

---
 scripts/benchmark.lua | 22 +++++++++++++++++++---
 1 file changed, 19 insertions(+), 3 deletions(-)

diff --git a/scripts/benchmark.lua b/scripts/benchmark.lua
index 457b671..56c3377 100644
--- a/scripts/benchmark.lua
+++ b/scripts/benchmark.lua
@@ -1,5 +1,7 @@
 package.path = "lua/?.lua;" .. package.path
 local libivy = require "ivy.libivy"
+local vim_mock = require "ivy.vim_mock"
+local window = require "ivy.window"
 
 local benchmark = function(name, n, callback)
   local status = {
@@ -25,7 +27,7 @@ local benchmark = function(name, n, callback)
 
   print(
     string.format(
-      "| %-30s | %09.6f (s) | %09.6f (s) | %09.6f (s) | %09.6f (s) |",
+      "| %-41s | %09.6f (s) | %09.6f (s) | %09.6f (s) | %09.6f (s) |",
       name,
       status.running_total,
       status.running_total / n,
@@ -35,8 +37,8 @@ local benchmark = function(name, n, callback)
   )
 end
 
-print "| Name                           | Total         | Average       | Min           | Max           |"
-print "|--------------------------------|---------------|---------------|---------------|---------------|"
+print "| Name                                      | Total         | Average       | Min           | Max           |"
+print "|-------------------------------------------|---------------|---------------|---------------|---------------|"
 
 benchmark("ivy_match(file.lua) 1000000x", 1000000, function()
   libivy.ivy_match("file.lua", "some/long/path/to/file/file.lua")
@@ -46,3 +48,17 @@ libivy.ivy_init "/tmp/ivy-trees/kubernetes"
 benchmark("ivy_files(kubernetes) 100x", 100, function()
   libivy.ivy_files("file.go", "/tmp/ivy-trees/kubernetes")
 end)
+
+-- Mock the vim API so we can run `vim.` functions. Override the
+-- `nvim_buf_set_lines` function, this is so very slow. It saves all of the
+-- lines so we can assert on them in the tests. For benchmarking we don't need
+-- any of this, we can't control the vim internals.
+vim_mock.reset()
+_G.vim.api.nvim_buf_set_lines = function() end
+
+window.initialize()
+
+benchmark("ivy_files_with_set_items(kubernetes) 100x", 100, function()
+  local items = libivy.ivy_files(".go", "/tmp/ivy-trees/kubernetes")
+  window.set_items(items)
+end)