unittest: convert helpers.moon to lua

2 files changed, 155 insertions, 131 deletions

diff --git a/test/unit/helpers.lua b/test/unit/helpers.lua
new file mode 100644
index 0000000000..5ad0766d61
--- /dev/null
+++ b/test/unit/helpers.lua
@@ -0,0 +1,155 @@
+local ffi = require('ffi')
+local lpeg = require('lpeg')
+local formatc = require('test.unit.formatc')
+local Set = require('test.unit.set')
+local Preprocess = require('test.unit.preprocess')
+local Paths = require('test.config.paths')
+
+-- add some standard header locations
+for i, p in ipairs(Paths.include_paths) do
+  Preprocess.add_to_include_path(p)
+end
+
+-- load neovim shared library
+local libnvim = ffi.load(Paths.test_libnvim_path)
+
+function trim(s)
+  return s:match('^%s*(.*%S)') or ''
+end
+
+-- a Set that keeps around the lines we've already seen
+if cdefs == nil then
+  cdefs = Set:new()
+end
+
+if imported == nil then
+  imported = Set:new()
+end
+
+-- some things are just too complex for the LuaJIT C parser to digest. We
+-- usually don't need them anyway.
+function filter_complex_blocks(body)
+  local result = {}
+
+  for line in body:gmatch("[^\r\n]+") do
+    if not (string.find(line, "(^)", 1, true) ~= nil or
+      string.find(line, "_ISwupper", 1, true)) then
+      result[#result + 1] = line
+    end
+  end
+
+  return table.concat(result, "\n")
+end
+
+-- use this helper to import C files, you can pass multiple paths at once,
+-- this helper will return the C namespace of the nvim library.
+-- cimport = (path) ->
+function cimport(...)
+  local paths = {}
+  local args = {...}
+
+  -- filter out paths we've already imported
+  for i = 1, #args do
+    local path = args[i]
+    if not imported:contains(path) then
+      paths[#paths + 1] = path
+    end
+  end
+
+  for i = 1, #paths do
+    imported:add(paths[i])
+  end
+
+  if #paths == 0 then
+    return libnvim
+  end
+
+  -- preprocess the header
+  local stream = Preprocess.preprocess_stream(unpack(paths))
+  local body = stream:read("*a")
+  stream:close()
+
+  -- format it (so that the lines are "unique" statements), also filter out
+  -- Objective-C blocks
+  body = formatc(body)
+  body = filter_complex_blocks(body)
+
+  -- add the formatted lines to a set
+  local new_cdefs = Set:new()
+  for line in body:gmatch("[^\r\n]+") do
+    new_cdefs:add(trim(line))
+  end
+
+  -- subtract the lines we've already imported from the new lines, then add
+  -- the new unique lines to the old lines (so they won't be imported again)
+  new_cdefs:diff(cdefs)
+  cdefs:union(new_cdefs)
+
+  if new_cdefs:size() == 0 then
+    -- if there's no new lines, just return
+    return libnvim
+  end
+
+  -- request a sorted version of the new lines (same relative order as the
+  -- original preprocessed file) and feed that to the LuaJIT ffi
+  local new_lines = new_cdefs:to_table()
+  ffi.cdef(table.concat(new_lines, "\n"))
+
+  return libnvim
+end
+
+function cppimport(path)
+  return cimport(Paths.test_include_path .. '/' .. path)
+end
+
+cimport('./src/nvim/types.h')
+
+-- take a pointer to a C-allocated string and return an interned
+-- version while also freeing the memory
+function internalize(cdata, len)
+  ffi.gc(cdata, ffi.C.free)
+  return ffi.string(cdata, len)
+end
+
+local cstr = ffi.typeof('char[?]')
+function to_cstr(string)
+  return cstr((string.len(string)) + 1, string)
+end
+
+-- initialize some global variables, this is still necessary to unit test
+-- functions that rely on global state.
+function vim_init()
+  if vim_init_called ~= nil then
+    return 
+  end
+  -- import os_unix.h for mch_early_init(), which initializes some globals
+  local os = cimport('./src/nvim/os_unix.h')
+  os.mch_early_init()
+  vim_init_called = true
+end
+
+-- C constants.
+local NULL = ffi.cast('void*', 0)
+
+local OK = 1
+local FAIL = 0
+
+return {
+  cimport = cimport,
+  cppimport = cppimport,
+  internalize = internalize,
+  eq = function(expected, actual)
+    return assert.are.same(expected, actual)
+  end,
+  neq = function(expected, actual)
+    return assert.are_not.same(expected, actual)
+  end,
+  ffi = ffi,
+  lib = libnvim,
+  cstr = cstr,
+  to_cstr = to_cstr,
+  vim_init = vim_init,
+  NULL = NULL,
+  OK = OK,
+  FAIL = FAIL
+}
diff --git a/test/unit/helpers.moon b/test/unit/helpers.moon
deleted file mode 100644
index f533b9a1c6..0000000000
--- a/test/unit/helpers.moon
+++ /dev/null
@@ -1,131 +0,0 @@
-ffi = require 'ffi'
-lpeg = require 'lpeg'
-formatc = require 'test.unit.formatc'
-Set = require 'test.unit.set'
-Preprocess = require 'test.unit.preprocess'
-Paths = require 'test.config.paths'
-
--- add some standard header locations
-for i,p in ipairs(Paths.include_paths)
-  Preprocess.add_to_include_path(p)
-
--- load neovim shared library
-libnvim = ffi.load Paths.test_libnvim_path
-
-trim = (s) ->
-  s\match'^%s*(.*%S)' or ''
-
--- a Set that keeps around the lines we've already seen
-export cdefs
-if cdefs == nil
-  cdefs = Set!
-
-export imported
-if imported == nil
-  imported = Set!
-
--- some things are just too complex for the LuaJIT C parser to digest. We
--- usually don't need them anyway.
-filter_complex_blocks = (body) ->
-  result = {}
-  for line in body\gmatch("[^\r\n]+")
-    -- remove all lines that contain Objective-C block syntax, the LuaJIT ffi
-    -- doesn't understand it.
-    if string.find(line, "(^)", 1, true) ~= nil
-      continue
-    if string.find(line, "_ISwupper", 1, true) ~= nil
-      continue
-    result[#result + 1] = line
-  table.concat(result, "\n")
-
--- use this helper to import C files, you can pass multiple paths at once,
--- this helper will return the C namespace of the nvim library.
--- cimport = (path) ->
-cimport = (...) ->
-  -- filter out paths we've already imported
-  paths = [path for path in *{...} when not imported\contains(path)]
-  for path in *paths
-    imported\add(path)
-
-  if #paths == 0
-    return libnvim
-
-  -- preprocess the header
-  stream = Preprocess.preprocess_stream(unpack(paths))
-  body = stream\read("*a")
-  stream\close!
-
-  -- format it (so that the lines are "unique" statements), also filter out
-  -- Objective-C blocks
-  body = formatc(body)
-  body = filter_complex_blocks(body)
-
-  -- add the formatted lines to a set
-  new_cdefs = Set!
-  for line in body\gmatch("[^\r\n]+")
-    new_cdefs\add(trim(line))
-
-  -- subtract the lines we've already imported from the new lines, then add
-  -- the new unique lines to the old lines (so they won't be imported again)
-  new_cdefs\diff(cdefs)
-  cdefs\union(new_cdefs)
-
-  if new_cdefs\size! == 0
-    -- if there's no new lines, just return
-    return libnvim
-
-  -- request a sorted version of the new lines (same relative order as the
-  -- original preprocessed file) and feed that to the LuaJIT ffi
-  new_lines = new_cdefs\to_table!
-  ffi.cdef(table.concat(new_lines, "\n"))
-
-  return libnvim
-
-cppimport = (path) ->
-  return cimport Paths.test_include_path .. '/' .. path
-
-cimport './src/nvim/types.h'
-
--- take a pointer to a C-allocated string and return an interned
--- version while also freeing the memory
-internalize = (cdata, len) ->
-  ffi.gc cdata, ffi.C.free
-  return ffi.string cdata, len
-
-cstr = ffi.typeof 'char[?]'
-
-to_cstr = (string) ->
-  cstr (string.len string) + 1, string
-
-export vim_init_called
--- initialize some global variables, this is still necessary to unit test
--- functions that rely on global state.
-vim_init = ->
-  if vim_init_called ~= nil
-    return
-  -- import os_unix.h for mch_early_init(), which initializes some globals
-  os = cimport './src/nvim/os_unix.h'
-  os.mch_early_init!
-  vim_init_called = true
-
--- C constants.
-NULL = ffi.cast 'void*', 0
-
-OK   = 1
-FAIL = 0
-
-return {
-  cimport: cimport
-  cppimport: cppimport
-  internalize: internalize
-  eq: (expected, actual) -> assert.are.same expected, actual
-  neq: (expected, actual) -> assert.are_not.same expected, actual
-  ffi: ffi
-  lib: libnvim
-  cstr: cstr
-  to_cstr: to_cstr
-  vim_init: vim_init
-  NULL: NULL
-  OK: OK
-  FAIL: FAIL
-}