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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
|
" Tests for memory usage.
if !has('terminal') || has('gui_running') || $ASAN_OPTIONS !=# ''
" Skip tests on Travis CI ASAN build because it's difficult to estimate
" memory usage.
finish
endif
source shared.vim
func s:pick_nr(str) abort
return substitute(a:str, '[^0-9]', '', 'g') * 1
endfunc
if has('win32')
if !executable('wmic')
finish
endif
func s:memory_usage(pid) abort
let cmd = printf('wmic process where processid=%d get WorkingSetSize', a:pid)
return s:pick_nr(system(cmd)) / 1024
endfunc
elseif has('unix')
if !executable('ps')
finish
endif
func s:memory_usage(pid) abort
return s:pick_nr(system('ps -o rss= -p ' . a:pid))
endfunc
else
finish
endif
" Wait for memory usage to level off.
func s:monitor_memory_usage(pid) abort
let proc = {}
let proc.pid = a:pid
let proc.hist = []
let proc.max = 0
func proc.op() abort
" Check the last 200ms.
let val = s:memory_usage(self.pid)
if self.max < val
let self.max = val
endif
call add(self.hist, val)
if len(self.hist) < 20
return 0
endif
let sample = remove(self.hist, 0)
return len(uniq([sample] + self.hist)) == 1
endfunc
call WaitFor({-> proc.op()}, 10000)
return {'last': get(proc.hist, -1), 'max': proc.max}
endfunc
let s:term_vim = {}
func s:term_vim.start(...) abort
let self.buf = term_start([GetVimProg()] + a:000)
let self.job = term_getjob(self.buf)
call WaitFor({-> job_status(self.job) ==# 'run'})
let self.pid = job_info(self.job).process
endfunc
func s:term_vim.stop() abort
call term_sendkeys(self.buf, ":qall!\<CR>")
call WaitFor({-> job_status(self.job) ==# 'dead'})
exe self.buf . 'bwipe!'
endfunc
func s:vim_new() abort
return copy(s:term_vim)
endfunc
func Test_memory_func_capture_vargs()
" Case: if a local variable captures a:000, funccall object will be free
" just after it finishes.
let testfile = 'Xtest.vim'
call writefile([
\ 'func s:f(...)',
\ ' let x = a:000',
\ 'endfunc',
\ 'for _ in range(10000)',
\ ' call s:f(0)',
\ 'endfor',
\ ], testfile)
let vim = s:vim_new()
call vim.start('--clean', '-c', 'set noswapfile', testfile)
let before = s:monitor_memory_usage(vim.pid).last
call term_sendkeys(vim.buf, ":so %\<CR>")
call WaitFor({-> term_getcursor(vim.buf)[0] == 1})
let after = s:monitor_memory_usage(vim.pid)
" Estimate the limit of max usage as 2x initial usage.
" The lower limit can fluctuate a bit, use 97%.
call assert_inrange(before * 97 / 100, 2 * before, after.max)
" In this case, garbage collecting is not needed.
" The value might fluctuate a bit, allow for 3% tolerance below and 5% above.
" Based on various test runs.
let lower = after.last * 97 / 100
let upper = after.last * 105 / 100
call assert_inrange(lower, upper, after.max)
call vim.stop()
call delete(testfile)
endfunc
func Test_memory_func_capture_lvars()
" Case: if a local variable captures l: dict, funccall object will not be
" free until garbage collector runs, but after that memory usage doesn't
" increase so much even when rerun Xtest.vim since system memory caches.
let testfile = 'Xtest.vim'
call writefile([
\ 'func s:f()',
\ ' let x = l:',
\ 'endfunc',
\ 'for _ in range(10000)',
\ ' call s:f()',
\ 'endfor',
\ ], testfile)
let vim = s:vim_new()
call vim.start('--clean', '-c', 'set noswapfile', testfile)
let before = s:monitor_memory_usage(vim.pid).last
call term_sendkeys(vim.buf, ":so %\<CR>")
call WaitFor({-> term_getcursor(vim.buf)[0] == 1})
let after = s:monitor_memory_usage(vim.pid)
" Rerun Xtest.vim.
for _ in range(3)
call term_sendkeys(vim.buf, ":so %\<CR>")
call WaitFor({-> term_getcursor(vim.buf)[0] == 1})
let last = s:monitor_memory_usage(vim.pid).last
endfor
" The usage may be a bit less than the last value, use 80%.
" Allow for 20% tolerance at the upper limit. That's very permissive, but
" otherwise the test fails sometimes.
let lower = before * 8 / 10
let upper = (after.max + (after.last - before)) * 12 / 10
call assert_inrange(lower, upper, last)
call vim.stop()
call delete(testfile)
endfunc
|
