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
|
# -*- coding: utf-8 -*-
import time
from typing import List
from pynvim.api import Nvim
def test_call_and_reply(vim: Nvim) -> None:
cid = vim.channel_id
def setup_cb() -> None:
cmd = 'let g:result = rpcrequest(%d, "client-call", 1, 2, 3)' % cid
vim.command(cmd)
assert vim.vars['result'] == [4, 5, 6]
vim.stop_loop()
def request_cb(name: str, args: List[int]) -> List[int]:
assert name == 'client-call'
assert args == [1, 2, 3]
return [4, 5, 6]
vim.run_loop(request_cb, None, setup_cb)
def test_call_api_before_reply(vim: Nvim) -> None:
cid = vim.channel_id
def setup_cb() -> None:
cmd = 'let g:result = rpcrequest(%d, "client-call2", 1, 2, 3)' % cid
vim.command(cmd)
assert vim.vars['result'] == [7, 8, 9]
vim.stop_loop()
def request_cb(name: str, args: List[int]) -> List[int]:
vim.command('let g:result2 = [7, 8, 9]')
return vim.vars['result2']
vim.run_loop(request_cb, None, setup_cb)
def test_async_call(vim: Nvim) -> None:
def request_cb(name: str, args: List[int]) -> None:
if name == "test-event":
vim.vars['result'] = 17
vim.stop_loop()
# this would have dead-locked if not async
vim.funcs.rpcrequest(vim.channel_id, "test-event", async_=True)
vim.run_loop(request_cb, None, None)
# TODO(blueyed): This sleep is required on Travis, where it hangs with
# "Entering event loop" otherwise (asyncio's EpollSelector._epoll.poll).
time.sleep(0.1)
assert vim.vars['result'] == 17
def test_recursion(vim: Nvim) -> None:
cid = vim.channel_id
def setup_cb() -> None:
vim.vars['result1'] = 0
vim.vars['result2'] = 0
vim.vars['result3'] = 0
vim.vars['result4'] = 0
cmd = 'let g:result1 = rpcrequest(%d, "call", %d)' % (cid, 2,)
vim.command(cmd)
assert vim.vars['result1'] == 4
assert vim.vars['result2'] == 8
assert vim.vars['result3'] == 16
assert vim.vars['result4'] == 32
vim.stop_loop()
def request_cb(name: str, args: List[int]) -> int:
n = args[0]
n *= 2
if n <= 16:
if n == 4:
cmd = 'let g:result2 = rpcrequest(%d, "call", %d)' % (cid, n,)
elif n == 8:
cmd = 'let g:result3 = rpcrequest(%d, "call", %d)' % (cid, n,)
elif n == 16:
cmd = 'let g:result4 = rpcrequest(%d, "call", %d)' % (cid, n,)
vim.command(cmd)
return n
vim.run_loop(request_cb, None, setup_cb)
|
