2 files changed, 115 insertions, 0 deletions
diff --git a/test/functional/ex_cmds/wviminfo_spec.lua b/test/functional/ex_cmds/wviminfo_spec.lua
new file mode 100644
index 0000000000..21176f1845
--- /dev/null
+++ b/test/functional/ex_cmds/wviminfo_spec.lua
@@ -0,0 +1,57 @@
+local helpers, lfs = require('test.functional.helpers'), require('lfs')
+local clear, execute, eq, neq, spawn, nvim_prog, set_session, wait =
+  helpers.clear, helpers.execute, helpers.eq, helpers.neq, helpers.spawn,
+  helpers.nvim_prog, helpers.set_session, helpers.wait
+
+describe(':wviminfo', function()
+  local viminfo_file = 'wviminfo_test'
+  local session
+
+  before_each(function()
+    if session then
+      session:exit(0)
+    end
+
+    -- Override the default session because we need 'swapfile' for these tests.
+    local session = spawn({nvim_prog, '-u', 'NONE', '--embed',
+                           '--cmd', 'set swapfile'})
+    set_session(session)
+
+    os.remove(viminfo_file)
+  end)
+
+  it('creates a viminfo file', function()
+    -- file should _not_ exist
+    eq(nil, lfs.attributes(viminfo_file))
+    execute('wv! '..viminfo_file)
+    wait()
+    -- file _should_ exist
+    neq(nil, lfs.attributes(viminfo_file))
+  end)
+
+  it('overwrites existing files', function()
+    local text = 'wviminfo test'
+
+    -- Create a dummy file
+    local fp = io.open(viminfo_file, 'w')
+    fp:write(text)
+    fp:flush()
+    fp:close()
+
+    -- sanity check
+    eq(text, io.open(viminfo_file):read())
+    neq(nil, lfs.attributes(viminfo_file))
+
+    execute('wv! '..viminfo_file)
+    wait()
+
+    -- File should have been overwritten with a viminfo file.
+    local line1 = io.lines(viminfo_file)()
+    assert(nil ~= string.find(line1, 'This viminfo file was generated by Nvim'),
+      viminfo_file..' should be a viminfo-formatted file')
+  end)
+
+  teardown(function()
+    os.remove(viminfo_file)
+  end)
+end)
diff --git a/test/functional/legacy/074_global_var_in_viminfo_spec.lua b/test/functional/legacy/074_global_var_in_viminfo_spec.lua
new file mode 100644
index 0000000000..180071118a
--- /dev/null
+++ b/test/functional/legacy/074_global_var_in_viminfo_spec.lua
@@ -0,0 +1,58 @@
+-- Tests for storing global variables in the .viminfo file
+
+local helpers, lfs = require('test.functional.helpers'), require('lfs')
+local clear, execute, eq, neq, eval, wait, spawn =
+  helpers.clear, helpers.execute, helpers.eq, helpers.neq, helpers.eval,
+  helpers.wait, helpers.spawn
+
+describe('storing global variables in viminfo files', function()
+  setup(function()
+    clear()
+    os.remove("Xviminfo")
+  end)
+
+  it('is working', function()
+    local nvim2 = helpers.spawn({helpers.nvim_prog, '-u', 'NONE', '--embed'})
+    helpers.set_session(nvim2)
+
+    local test_dict = {foo = 1, bar = 0, longvarible = 1000}
+    local test_list = {1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
+      17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34,
+      35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52,
+      53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70,
+      71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88,
+      89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100}
+
+    execute(
+      -- This will cause a few errors, do it silently.
+      'set visualbell',
+      'set viminfo+=!',
+      "let MY_GLOBAL_DICT={'foo': 1, 'bar': 0, 'longvarible': 1000}",
+      -- Store a really long list, so line wrapping will occur in viminfo
+      -- file.
+      'let MY_GLOBAL_LIST=range(1,100)'
+    )
+    eq(test_dict, eval('MY_GLOBAL_DICT'))
+    eq(test_list, eval('MY_GLOBAL_LIST'))
+
+    execute('wv! Xviminfo')
+    wait()
+
+    -- Assert that the viminfo file exists.
+    neq(nil, lfs.attributes('Xviminfo'))
+    execute('unlet MY_GLOBAL_DICT',
+            'unlet MY_GLOBAL_LIST')
+    -- Assert that the variables where deleted.
+    eq(0, eval('exists("MY_GLOBAL_DICT")'))
+    eq(0, eval('exists("MY_GLOBAL_LIST")'))
+
+    execute('rv! Xviminfo')
+
+    eq(test_list, eval('MY_GLOBAL_LIST'))
+    eq(test_dict, eval('MY_GLOBAL_DICT'))
+  end)
+
+  teardown(function()
+    os.remove('Xviminfo')
+  end)
+end)