1 files changed, 209 insertions, 0 deletions
diff --git a/test/functional/legacy/061_undo_tree_spec.lua b/test/functional/legacy/061_undo_tree_spec.lua
new file mode 100644
index 0000000000..e3f3997440
--- /dev/null
+++ b/test/functional/legacy/061_undo_tree_spec.lua
@@ -0,0 +1,209 @@
+-- Tests for undo tree.
+-- Since this script is sourced we need to explicitly break changes up in
+-- undo-able pieces.  Do that by setting 'undolevels'.
+-- Also tests :earlier and :later.
+
+local helpers = require('test.functional.helpers')
+local feed, insert, source, eq, eval, clear, execute, expect, wait =
+  helpers.feed, helpers.insert, helpers.source, helpers.eq, helpers.eval,
+  helpers.clear, helpers.execute, helpers.expect, helpers.wait
+
+describe('the undo tree', function()
+  setup(clear)
+  teardown(function()
+    os.remove('Xtest')
+  end)
+
+  it('is working', function()
+    -- Assert that no undo history is present.
+    eq({}, eval('undotree().entries'))
+    insert([[
+      
+      123456789]])
+
+    -- Clear the undo history after the insertion (see :h clear-undo)
+    execute('let old_undolevels = &undolevels')
+    execute('set undolevels=-1')
+    feed('a <BS><Esc>') 
+    execute('let &undolevels = old_undolevels')
+    execute('unlet old_undolevels')
+    eq({}, eval('undotree().entries'))
+
+    -- Delete three characters and undo.
+    feed('Gx')
+    execute('set ul=100')
+    feed('x')
+    execute('set ul=100')
+    feed('x')
+    eq('456789', eval('getline(".")'))
+    feed('g-')
+    eq('3456789', eval('getline(".")'))
+    feed('g-')
+    eq('23456789', eval('getline(".")'))
+    feed('g-')
+    eq('123456789', eval('getline(".")'))
+    feed('g-')
+    eq('123456789', eval('getline(".")'))
+
+    -- Delete three other characters and go back in time step by step.
+    feed('$x')
+    execute('set ul=100')
+    feed('x')
+    execute('set ul=100')
+    feed('x')
+    eq('123456', eval('getline(".")'))
+    execute('sleep 1')
+    wait()
+    feed('g-')
+    eq('1234567', eval('getline(".")'))
+    feed('g-')
+    eq('12345678', eval('getline(".")'))
+    feed('g-')
+    eq('456789', eval('getline(".")'))
+    feed('g-')
+    eq('3456789', eval('getline(".")'))
+    feed('g-')
+    eq('23456789', eval('getline(".")'))
+    feed('g-')
+    eq('123456789', eval('getline(".")'))
+    feed('g-')
+    eq('123456789', eval('getline(".")'))
+    feed('g-')
+    eq('123456789', eval('getline(".")'))
+    feed('10g+')
+    eq('123456', eval('getline(".")'))
+
+    -- Delay for three seconds and go some seconds forward and backward.
+    execute('sleep 2')
+    wait()
+    feed('Aa<esc>')
+    execute('set ul=100')
+    feed('Ab<esc>')
+    execute('set ul=100')
+    feed('Ac<esc>')
+    execute('set ul=100')
+    eq('123456abc', eval('getline(".")'))
+    execute('ear 1s')
+    eq('123456', eval('getline(".")'))
+    execute('ear 3s')
+    eq('123456789', eval('getline(".")'))
+    execute('later 1s')
+    eq('123456', eval('getline(".")'))
+    execute('later 1h')
+    eq('123456abc', eval('getline(".")'))
+
+    -- Test undojoin.
+    feed('Goaaaa<esc>')
+    execute('set ul=100')
+    feed('obbbb<esc>u')
+    eq('aaaa', eval('getline(".")'))
+    feed('obbbb<esc>')
+    execute('set ul=100')
+    execute('undojoin')
+    feed('occcc<esc>u')
+    -- TODO At this point the original test will write "aaaa" to test.out.
+    -- Why is the line "bbbb" here?
+    eq('bbbb', eval('getline(".")'))
+
+    execute('e! Xtest')
+    feed('ione one one<esc>')
+    execute('set ul=100')
+    execute('w!')
+    feed('otwo<esc>')
+    execute('set ul=100')
+    feed('otwo<esc>')
+    execute('set ul=100')
+    execute('w')
+    feed('othree<esc>')
+    execute('earlier 1f')
+    expect([[
+      one one one
+      two
+      two]])
+    execute('earlier 1f')
+    expect('one one one')
+    execute('earlier 1f')
+    -- Expect an empty line (the space is needed for helpers.dedent but
+    -- removed).
+    expect(' ')
+    execute('later 1f')
+    expect('one one one')
+    execute('later 1f')
+    expect([[
+      one one one
+      two
+      two]])
+    execute('later 1f')
+    expect([[
+      one one one
+      two
+      two
+      three]])
+
+    execute('enew!')
+    feed('oa<esc>')
+    execute('set ul=100')
+    feed('ob<esc>')
+    execute('set ul=100')
+    feed([[o1<esc>a2<C-R>=setline('.','1234')<cr><esc>]])
+    expect([[
+      
+      a
+      b
+      12034]])
+
+    feed('uu')
+    expect([[
+      
+      a
+      b
+      1]])
+    feed('oc<esc>')
+    execute('set ul=100')
+    feed([[o1<esc>a2<C-R>=setline('.','1234')<cr><esc>]])
+    expect([[
+      
+      a
+      b
+      1
+      c
+      12034]])
+    feed('u')
+    expect([[
+      
+      a
+      b
+      1
+      c
+      12]])
+    feed('od<esc>')
+    execute('set ul=100')
+    feed('o1<esc>a2<C-R>=string(123)<cr><esc>')
+    expect([[
+      
+      a
+      b
+      1
+      c
+      12
+      d
+      12123]])
+
+    -- TODO there is a difference between the original test and this test at
+    -- this point.  The original tests expects the last line to go away after
+    -- the undo.  I do not know why this should be the case as the "o" and "a"
+    -- above are seperate changes.  I was able to confirm this manually with
+    -- vim and nvim.  Both end up in this state (treat "o" and "a" as two
+    -- edits).
+    feed('u')
+    expect([[
+      
+      a
+      b
+      1
+      c
+      12
+      d
+      1]])
+  end)
+end)