unittests: Add tests for vim_to_object function

1 files changed, 146 insertions, 11 deletions

diff --git a/test/unit/eval/helpers.lua b/test/unit/eval/helpers.lua
index 2367f03e0d..528657320b 100644
--- a/test/unit/eval/helpers.lua
+++ b/test/unit/eval/helpers.lua
@@ -11,6 +11,10 @@ local null_string = {[true]='NULL string'}
 local null_list = {[true]='NULL list'}
 local type_key = {[true]='type key'}
 local list_type = {[true]='list type'}
+local dict_type = {[true]='dict type'}
+local func_type = {[true]='func type'}
+local int_type = {[true]='int type'}
+local flt_type = {[true]='flt type'}
 
 local function list(...)
   local ret = ffi.gc(eval.list_alloc(), eval.list_unref)
@@ -37,7 +41,51 @@ local function list(...)
   return ret
 end
 
-local lst2tbl = function(l)
+local special_tab = {
+  [eval.kSpecialVarFalse] = false,
+  [eval.kSpecialVarNull] = NIL,
+  [eval.kSpecialVarTrue] = true,
+}
+
+local lst2tbl
+local dct2tbl
+
+local typvalt2lua_tab = {
+  [tonumber(eval.VAR_SPECIAL)] = function(t)
+    return special_tab[t.vval.v_special]
+  end,
+  [tonumber(eval.VAR_NUMBER)] = function(t)
+    return {[type_key]=int_type, value=tonumber(t.vval.v_number)}
+  end,
+  [tonumber(eval.VAR_FLOAT)] = function(t)
+    return tonumber(t.vval.v_float)
+  end,
+  [tonumber(eval.VAR_STRING)] = function(t)
+    str = t.vval.v_string
+    if str == nil then
+      return null_string
+    else
+      return ffi.string(str)
+    end
+  end,
+  [tonumber(eval.VAR_LIST)] = function(t)
+    return lst2tbl(t.vval.v_list)
+  end,
+  [tonumber(eval.VAR_DICT)] = function(t)
+    return dct2tbl(t.vval.v_dict)
+  end,
+  [tonumber(eval.VAR_FUNC)] = function(t)
+    return {[type_key]=func_type, value=typvalt2lua_tab[eval.VAR_STRING](t)}
+  end,
+}
+
+local typvalt2lua = function(t)
+  return ((typvalt2lua_tab[tonumber(t.v_type)] or function(t)
+    assert(false, 'Converting ' .. tonumber(t.v_type) .. ' was not implemented yet')
+  end)(t))
+end
+
+lst2tbl = function(l)
   local ret = {[type_key]=list_type}
   if l == nil then
     return ret
@@ -45,28 +93,115 @@ local lst2tbl = function(l)
   local li = l.lv_first
   -- (listitem_T *) NULL is equal to nil, but yet it is not false.
   while li ~= nil do
-    local typ = li.li_tv.v_type
-    if typ == eval.VAR_STRING then
-      local str = li.li_tv.vval.v_string
-      if str == nil then
-        ret[#ret + 1] = null_string
-      else
-        ret[#ret + 1] = ffi.string(str)
+    ret[#ret + 1] = typvalt2lua(li.li_tv)
+    li = li.li_next
+  end
+  if ret[1] then
+    ret[type_key] = nil
+  end
+  return ret
+end
+
+dct2tbl = function(d)
+  local ret = {}
+  assert(false, 'Converting dictionaries is not implemented yet')
+end
+
+local lua2typvalt
+
+local typvalt = function(typ, vval)
+  if type(typ) == 'string' then
+    typ = eval[typ]
+  end
+  return ffi.gc(ffi.new('typval_T', {v_type=typ, vval=vval}), eval.clear_tv)
+end
+
+local lua2typvalt_type_tab = {
+  [int_type] = function(l, processed)
+    return typvalt(eval.VAR_NUMBER, {v_number=l.value})
+  end,
+  [flt_type] = function(l, processed)
+    return lua2typvalt(l.value, processed)
+  end,
+  [list_type] = function(l, processed)
+    if processed[l] then
+      processed[l].lv_refcount = processed[l].lv_refcount + 1
+      return typvalt(eval.VAR_LIST, {v_list=processed[l]})
+    end
+    local lst = eval.list_alloc()
+    lst.lv_refcount = 1
+    processed[l] = lst
+    local ret = typvalt(eval.VAR_LIST, {v_list=lst})
+    for i = 1, #l do
+      local item_tv = ffi.gc(lua2typvalt(l[i], processed), nil)
+      eval.list_append_tv(lst, item_tv)
+      eval.clear_tv(item_tv)
+    end
+    return ret
+  end,
+  [dict_type] = function(l, processed)
+    if processed[l] then
+      processed[l].dv_refcount = processed[l].dv_refcount + 1
+      return typvalt(eval.VAR_DICT, {v_dict=processed[l]})
+    end
+    local dct = eval.dict_alloc()
+    dct.dv_refcount = 1
+    processed[l] = dct
+    local ret = typvalt(eval.VAR_DICT, {v_dict=dct})
+    for k, v in pairs(l) do
+      if type(k) == 'string' then
+        local di = eval.dictitem_alloc(to_cstr(k))
+        local val_tv = ffi.gc(lua2typvalt(v, processed), nil)
+        eval.copy_tv(val_tv, di.di_tv)
+        eval.clear_tv(val_tv)
+        eval.dict_add(dct, di)
       end
+    end
+    return ret
+  end,
+}
+
+lua2typvalt = function(l, processed)
+  processed = processed or {}
+  if type(l) == 'table' then
+    if l[type_key] then
+      return lua2typvalt_type_tab[l[type_key]](l, processed)
     else
-      assert(false, 'Not implemented yet')
+      if l[1] then
+        return lua2typvalt_type_tab[list_type](l, processed)
+      else
+        return lua2typvalt_type_tab[dict_type](l, processed)
+      end
     end
-    li = li.li_next
+  elseif type(l) == 'number' then
+    return typvalt(eval.VAR_FLOAT, {v_float=l})
+  elseif type(l) == 'boolean' then
+    return typvalt(eval.VAR_SPECIAL, {
+      v_special=(l and eval.kSpecialVarTrue or eval.kSpecialVarFalse)
+    })
+  elseif type(l) == 'string' then
+    return typvalt(eval.VAR_STRING, {v_string=eval.xmemdupz(to_cstr(l), #l)})
+  elseif l == nil or l == NIL then
+    return typvalt(eval.VAR_SPECIAL, {v_special=eval.kSpecialVarNull})
   end
-  return ret
 end
 
 return {
   null_string=null_string,
   null_list=null_list,
   list_type=list_type,
+  dict_type=dict_type,
+  func_type=func_type,
+  int_type=int_type,
+  flt_type=flt_type,
   type_key=type_key,
 
   list=list,
   lst2tbl=lst2tbl,
+  dct2tbl=dct2tbl,
+
+  lua2typvalt=lua2typvalt,
+  typvalt2lua=typvalt2lua,
+
+  typvalt=typvalt,
 }