mirror of
https://github.com/neovim/neovim.git
synced 2024-12-29 14:41:06 -07:00
unittest: convert helpers.moon to lua
This commit is contained in:
Thiago de Arruda
Nicolas Hillegeer
parent
097ff4b8b2
commit
dcda179e6a
155
test/unit/helpers.lua
Normal file
155
test/unit/helpers.lua
Normal file
@ -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
|
||||
}
|
||||
@ -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
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user