aboutsummaryrefslogtreecommitdiff
path: root/test/functional/runtime/autoload/provider/python_spec.lua
blob: 1a726652d6e46de1033d84d34c19b1c814cc0141 (plain) (blame)
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
do
  local proc =
    io.popen([[python -c 'import neovim, sys; sys.stdout.write("ok")' 2> /dev/null]])
  if proc:read() ~= 'ok' then
    -- Don't run these tests if python is not available
    return
  end
end


local helpers = require('test.functional.helpers')
local eval, command, feed = helpers.eval, helpers.command, helpers.feed
local eq, clear, insert = helpers.eq, helpers.clear, helpers.insert
local expect = helpers.expect


describe('python commands and functions', function()
  before_each(function()
    clear()
    command('python import vim')
  end)

  describe('feature test', function()
    it('ok', function()
      eq(1, eval('has("python")'))
    end)
  end)

  describe('python_execute', function()
    it('ok', function()
      command('python vim.vars["set_by_python"] = [100, 0]')
      eq({100, 0}, eval('g:set_by_python'))
    end)
  end)

  describe('python_execute with nested commands', function()
    it('ok', function()
      command([[python vim.command('python vim.command("python vim.command(\'let set_by_nested_python = 555\')")')]])
      eq(555, eval('g:set_by_nested_python'))
    end)
  end)

  describe('python_execute with range', function()
    it('ok', function()
      insert([[
        line1
        line2
        line3
        line4]])
      feed('ggjvj:python vim.vars["range"] = vim.current.range[:]<CR>')
      eq({'line2', 'line3'}, eval('g:range'))
    end)
  end)

  describe('pyfile', function()
    it('ok', function()
      local fname = 'pyfile.py'
      local F = io.open(fname, 'w')
      F:write('vim.command("let set_by_pyfile = 123")')
      F:close()
      command('pyfile pyfile.py')
      eq(123, eval('g:set_by_pyfile'))
      os.remove(fname)
    end)
  end)

  describe('pydo', function()
    it('ok', function()
      -- :pydo 42 returns None for all lines,
      -- the buffer should not be changed
      command('normal :pydo 42')
      eq(0, eval('&mod'))
      -- insert some text
      insert('abc\ndef\nghi')
      expect([[
        abc
        def
        ghi]])
      -- go to top and select and replace the first two lines
      feed('ggvj:pydo return str(linenr)<CR>')
      expect([[
        1
        2
        ghi]])
    end)
  end)

  describe('pyeval', function()
    it('ok', function()
      eq({1, 2, {['key'] = 'val'}}, eval([[pyeval('[1, 2, {"key": "val"}]')]]))
    end)
  end)
end)