1 files changed, 84 insertions, 0 deletions
diff --git a/test/benchmark/keycodes_spec.lua b/test/benchmark/keycodes_spec.lua
new file mode 100644
index 0000000000..443df2a6fb
--- /dev/null
+++ b/test/benchmark/keycodes_spec.lua
@@ -0,0 +1,84 @@
+local n = require('test.functional.testnvim')()
+local clear = n.clear
+local api = n.api
+local fn = n.fn
+
+local keycodes = require('src.nvim.keycodes')
+
+describe('nvim_replace_termcodes performance', function()
+  it('200 calls with a key repeated 5000 times', function()
+    clear()
+    local stats = {}
+    local sum = 0
+    local ms = 1 / 1000000
+
+    for _, keycode in ipairs(keycodes.names) do
+      local notation = ('<%s>'):format(keycode[2])
+      local str = notation:rep(5000)
+
+      local start = vim.uv.hrtime()
+      for _ = 1, 200 do
+        api.nvim_replace_termcodes(str, false, true, true)
+      end
+      local elapsed = vim.uv.hrtime() - start
+
+      table.insert(stats, elapsed)
+      sum = sum + elapsed
+      io.stdout:write(('\n%-20s%14.6f ms'):format(notation, elapsed * ms))
+      io.stdout:flush()
+    end
+    io.stdout:write('\n')
+
+    table.sort(stats)
+    print(('%18s'):rep(6):format('avg', 'min', '25%', 'median', '75%', 'max'))
+    print(
+      (' %14.6f ms'):rep(6):format(
+        sum / #stats * ms,
+        stats[1] * ms,
+        stats[1 + math.floor(#stats * 0.25)] * ms,
+        stats[1 + math.floor(#stats * 0.5)] * ms,
+        stats[1 + math.floor(#stats * 0.75)] * ms,
+        stats[#stats] * ms
+      )
+    )
+  end)
+end)
+
+describe('keytrans() performance', function()
+  it('200 calls with a key repeated 5000 times', function()
+    clear()
+    local stats = {}
+    local sum = 0
+    local ms = 1 / 1000000
+
+    for _, keycode in ipairs(keycodes.names) do
+      local notation = ('<%s>'):format(keycode[2])
+      local str = api.nvim_replace_termcodes(notation, false, true, true):rep(5000)
+
+      local start = vim.uv.hrtime()
+      for _ = 1, 200 do
+        fn.keytrans(str)
+      end
+      local elapsed = vim.uv.hrtime() - start
+
+      table.insert(stats, elapsed)
+      sum = sum + elapsed
+      io.stdout:write(('\n%-20s%14.6f ms'):format(notation, elapsed * ms))
+      io.stdout:flush()
+    end
+    io.stdout:write('\n')
+
+    table.sort(stats)
+    print((' %17s'):rep(6):format('avg', 'min', '25%', 'median', '75%', 'max'))
+    print(
+      (' %14.6f ms'):rep(6):format(
+        sum / #stats * ms,
+        stats[1] * ms,
+        stats[1 + math.floor(#stats * 0.25)] * ms,
+        stats[1 + math.floor(#stats * 0.5)] * ms,
+        stats[1 + math.floor(#stats * 0.75)] * ms,
+        stats[#stats] * ms
+      )
+    )
+  end)
+end)