functests: Start adding some tests

1 files changed, 101 insertions, 0 deletions

diff --git a/test/functional/ui/cmdline_highlight_spec.lua b/test/functional/ui/cmdline_highlight_spec.lua
new file mode 100644
index 0000000000..812841ab7e
--- /dev/null
+++ b/test/functional/ui/cmdline_highlight_spec.lua
@@ -0,0 +1,101 @@
+local helpers = require('test.functional.helpers')(after_each)
+local Screen = require('test.functional.ui.screen')
+
+local feed = helpers.feed
+local clear = helpers.clear
+local meths = helpers.meths
+local source = helpers.source
+
+local screen
+
+before_each(function()
+  clear()
+  screen = Screen.new(40, 2)
+  screen:attach()
+  source([[
+    highlight RBP1 guifg=Red
+    highlight RBP2 guifg=Yellow
+    highlight RBP3 guifg=Green
+    highlight RBP4 guifg=Blue
+    let g:NUM_LVLS = 4
+    function Redraw()
+      redraw!
+      return ''
+    endfunction
+    cnoremap <expr> {REDRAW} Redraw()
+    function RainBowParens(cmdline)
+      let ret = []
+      let i = 0
+      let lvl = 0
+      while i < len(a:cmdline)
+        if a:cmdline[i] is# '('
+          call add(ret, [i, i + 1, 'RBP' . ((lvl % g:NUM_LVLS) + 1)])
+          let lvl += 1
+        elseif a:cmdline[i] is# ')'
+          let lvl -= 1
+          call add(ret, [i, i + 1, 'RBP' . ((lvl % g:NUM_LVLS) + 1)])
+        endif
+        let i += 1
+      endwhile
+      return ret
+    endfunction
+  ]])
+  screen:set_default_attr_ids({
+    RBP1={foreground = Screen.colors.Red},
+    RBP2={foreground = Screen.colors.Yellow},
+    RBP3={foreground = Screen.colors.Green},
+    RBP4={foreground = Screen.colors.Blue},
+  })
+end)
+
+describe('Command-line coloring', function()
+  it('works', function()
+    meths.set_var('Nvim_color_cmdline', 'RainBowParens')
+    meths.set_option('more', false)
+    feed(':')
+    screen:expect([[
+                                              |
+      :^                                       |
+    ]])
+    feed('e')
+    screen:expect([[
+                                              |
+      :e^                                      |
+    ]])
+    feed('cho ')
+    screen:expect([[
+                                              |
+      :echo ^                                  |
+    ]])
+    feed('(')
+    screen:expect([[
+                                              |
+      :echo {RBP1:(}^                                 |
+    ]])
+    feed('(')
+    screen:expect([[
+                                              |
+      :echo {RBP1:(}{RBP2:(}^                                |
+    ]])
+    feed('42')
+    screen:expect([[
+                                              |
+      :echo {RBP1:(}{RBP2:(}42^                              |
+    ]])
+    feed('))')
+    screen:expect([[
+                                              |
+      :echo {RBP1:(}{RBP2:(}42{RBP2:)}{RBP1:)}^                            |
+    ]])
+    feed('<BS>')
+    screen:expect([[
+                                              |
+      :echo {RBP1:(}{RBP2:(}42{RBP2:)}^                             |
+    ]])
+    feed('{REDRAW}')
+    screen:expect([[
+                                              |
+      :echo {RBP1:(}{RBP2:(}42{RBP2:)}^                             |
+    ]])
+  end)
+end)