xref: /vim-8.2.3635/src/testdir/test_let.vim (revision 90df4b9d) 
1" Tests for the :let command.
2
3func Test_let()
4  " Test to not autoload when assigning.  It causes internal error.
5  set runtimepath+=./sautest
6  let Test104#numvar = function('tr')
7  call assert_equal("function('tr')", string(Test104#numvar))
8
9  let a = 1
10  let b = 2
11
12  let out = execute('let a b')
13  let s = "\na                     #1\nb                     #2"
14  call assert_equal(s, out)
15
16  let out = execute('let {0 == 1 ? "a" : "b"}')
17  let s = "\nb                     #2"
18  call assert_equal(s, out)
19
20  let out = execute('let {0 == 1 ? "a" : "b"} a')
21  let s = "\nb                     #2\na                     #1"
22  call assert_equal(s, out)
23
24  let out = execute('let a {0 == 1 ? "a" : "b"}')
25  let s = "\na                     #1\nb                     #2"
26  call assert_equal(s, out)
27
28  " Test for displaying a string variable
29  let s = 'vim'
30  let out = execute('let s')
31  let s = "\ns                      vim"
32  call assert_equal(s, out)
33
34  " Test for displaying a list variable
35  let l = [1, 2]
36  let out = execute('let l')
37  let s = "\nl                     [1, 2]"
38  call assert_equal(s, out)
39
40  " Test for displaying a dict variable
41  let d = {'k' : 'v'}
42  let out = execute('let d')
43  let s = "\nd                     {'k': 'v'}"
44  call assert_equal(s, out)
45
46  " Test for displaying a function reference variable
47  let F = function('min')
48  let out = execute('let F')
49  let s = "\nF                     *min()"
50  call assert_equal(s, out)
51
52  let x = 0
53  if 0 | let x = 1 | endif
54  call assert_equal(0, x)
55
56  " Display a list item using an out of range index
57  let l = [10]
58  call assert_fails('let l[1]', 'E684:')
59
60  " List special variable dictionaries
61  let g:Test_Global_Var = 5
62  call assert_match("\nTest_Global_Var       #5", execute('let g:'))
63  unlet g:Test_Global_Var
64
65  let b:Test_Buf_Var = 8
66  call assert_match("\nb:Test_Buf_Var        #8", execute('let b:'))
67  unlet b:Test_Buf_Var
68
69  let w:Test_Win_Var = 'foo'
70  call assert_equal("\nw:Test_Win_Var         foo", execute('let w:'))
71  unlet w:Test_Win_Var
72
73  let t:Test_Tab_Var = 'bar'
74  call assert_equal("\nt:Test_Tab_Var         bar", execute('let t:'))
75  unlet t:Test_Tab_Var
76
77  let s:Test_Script_Var = [7]
78  call assert_match("\ns:Test_Script_Var     \\[7]", execute('let s:'))
79  unlet s:Test_Script_Var
80
81  let l:Test_Local_Var = {'k' : 5}
82  call assert_match("\nl:Test_Local_Var      {'k': 5}", execute('let l:'))
83  call assert_match("v:errors              []", execute('let v:'))
84
85  " Test for assigning multiple list items
86  let l = [1, 2, 3]
87  let [l[0], l[1]] = [10, 20]
88  call assert_equal([10, 20, 3], l)
89
90  " Test for errors in conditional expression
91  call assert_fails('let val = [] ? 1 : 2', 'E745:')
92  call assert_fails('let val = 1 ? 5+ : 6', 'E121:')
93  call assert_fails('let val = 1 ? 0 : 5+', 'E15:')
94  call assert_false(exists('val'))
95
96  " Test for errors in logical operators
97  let @a = 'if [] || 0 | let val = 2 | endif'
98  call assert_fails('exe @a', 'E745:')
99  call assert_fails('call feedkeys(":let val = 0 || []\<cr>", "xt")', 'E745:')
100  call assert_fails('exe "let val = [] && 5"', 'E745:')
101  call assert_fails('exe "let val = 6 && []"', 'E745:')
102endfunc
103
104func s:set_arg1(a) abort
105  let a:a = 1
106endfunction
107
108func s:set_arg2(a) abort
109  let a:b = 1
110endfunction
111
112func s:set_arg3(a) abort
113  let b = a:
114  let b['a'] = 1
115endfunction
116
117func s:set_arg4(a) abort
118  let b = a:
119  let b['a'] = 1
120endfunction
121
122func s:set_arg5(a) abort
123  let b = a:
124  let b['a'][0] = 1
125endfunction
126
127func s:set_arg6(a) abort
128  let a:a[0] = 1
129endfunction
130
131func s:set_arg7(a) abort
132  call extend(a:, {'a': 1})
133endfunction
134
135func s:set_arg8(a) abort
136  call extend(a:, {'b': 1})
137endfunction
138
139func s:set_arg9(a) abort
140  let a:['b'] = 1
141endfunction
142
143func s:set_arg10(a) abort
144  let b = a:
145  call extend(b, {'a': 1})
146endfunction
147
148func s:set_arg11(a) abort
149  let b = a:
150  call extend(b, {'b': 1})
151endfunction
152
153func s:set_arg12(a) abort
154  let b = a:
155  let b['b'] = 1
156endfunction
157
158func Test_let_arg_fail()
159  call assert_fails('call s:set_arg1(1)', 'E46:')
160  call assert_fails('call s:set_arg2(1)', 'E461:')
161  call assert_fails('call s:set_arg3(1)', 'E46:')
162  call assert_fails('call s:set_arg4(1)', 'E46:')
163  call assert_fails('call s:set_arg5(1)', 'E46:')
164  call s:set_arg6([0])
165  call assert_fails('call s:set_arg7(1)', 'E742:')
166  call assert_fails('call s:set_arg8(1)', 'E742:')
167  call assert_fails('call s:set_arg9(1)', 'E461:')
168  call assert_fails('call s:set_arg10(1)', 'E742:')
169  call assert_fails('call s:set_arg11(1)', 'E742:')
170  call assert_fails('call s:set_arg12(1)', 'E461:')
171endfunction
172
173func s:set_varg1(...) abort
174  let a:000 = []
175endfunction
176
177func s:set_varg2(...) abort
178  let a:000[0] = 1
179endfunction
180
181func s:set_varg3(...) abort
182  let a:000 += [1]
183endfunction
184
185func s:set_varg4(...) abort
186  call add(a:000, 1)
187endfunction
188
189func s:set_varg5(...) abort
190  let a:000[0][0] = 1
191endfunction
192
193func s:set_varg6(...) abort
194  let b = a:000
195  let b[0] = 1
196endfunction
197
198func s:set_varg7(...) abort
199  let b = a:000
200  let b += [1]
201endfunction
202
203func s:set_varg8(...) abort
204  let b = a:000
205  call add(b, 1)
206endfunction
207
208func s:set_varg9(...) abort
209  let b = a:000
210  let b[0][0] = 1
211endfunction
212
213func Test_let_varg_fail()
214  call assert_fails('call s:set_varg1(1)', 'E46:')
215  call assert_fails('call s:set_varg2(1)', 'E742:')
216  call assert_fails('call s:set_varg3(1)', 'E46:')
217  call assert_fails('call s:set_varg4(1)', 'E742:')
218  call s:set_varg5([0])
219  call assert_fails('call s:set_varg6(1)', 'E742:')
220  call assert_fails('call s:set_varg7(1)', 'E742:')
221  call assert_fails('call s:set_varg8(1)', 'E742:')
222  call s:set_varg9([0])
223endfunction
224
225func Test_let_utf8_environment()
226  let $a = 'ĀĒĪŌŪあいうえお'
227  call assert_equal('ĀĒĪŌŪあいうえお', $a)
228endfunc
229
230func Test_let_no_type_checking()
231  let v = 1
232  let v = [1,2,3]
233  let v = {'a': 1, 'b': 2}
234  let v = 3.4
235  let v = 'hello'
236endfunc
237
238func Test_let_termcap()
239  " Terminal code
240  let old_t_te = &t_te
241  let &t_te = "\<Esc>[yes;"
242  call assert_match('t_te.*^[[yes;', execute("set termcap"))
243  let &t_te = old_t_te
244
245  if exists("+t_k1")
246    " Key code
247    let old_t_k1 = &t_k1
248    let &t_k1 = "that"
249    call assert_match('t_k1.*that', execute("set termcap"))
250    let &t_k1 = old_t_k1
251  endif
252
253  call assert_fails('let x = &t_xx', 'E113:')
254  let &t_xx = "yes"
255  call assert_equal("yes", &t_xx)
256  let &t_xx = ""
257  call assert_fails('let x = &t_xx', 'E113:')
258endfunc
259
260func Test_let_option_error()
261  let _w = &tw
262  let &tw = 80
263  call assert_fails('let &tw .= 1', 'E734:')
264  call assert_equal(80, &tw)
265  let &tw = _w
266
267  let _w = &fillchars
268  let &fillchars = "vert:|"
269  call assert_fails('let &fillchars += "diff:-"', 'E734:')
270  call assert_equal("vert:|", &fillchars)
271  let &fillchars = _w
272endfunc
273
274" Errors with the :let statement
275func Test_let_errors()
276  let s = 'abcd'
277  call assert_fails('let s[1] = 5', 'E689:')
278
279  let l = [1, 2, 3]
280  call assert_fails('let l[:] = 5', 'E709:')
281
282  call assert_fails('let x:lnum=5', ['E121:', 'E488:'])
283  call assert_fails('let v:=5', 'E461:')
284  call assert_fails('let [a]', 'E474:')
285  call assert_fails('let [a, b] = [', 'E697:')
286  call assert_fails('let [a, b] = [10, 20', 'E696:')
287  call assert_fails('let [a, b] = 10', 'E714:')
288  call assert_fails('let [a, , b] = [10, 20]', 'E475:')
289  call assert_fails('let [a, b&] = [10, 20]', 'E475:')
290  call assert_fails('let $ = 10', 'E475:')
291  call assert_fails('let $FOO[1] = "abc"', 'E18:')
292  call assert_fails('let &buftype[1] = "nofile"', 'E18:')
293  let s = "var"
294  let var = 1
295  call assert_fails('let var += [1,2]', 'E734:')
296  call assert_fails('let {s}.1 = 2', 'E1203:')
297  call assert_fails('let a[1] = 5', 'E121:')
298  let l = [[1,2]]
299  call assert_fails('let l[:][0] = [5]', 'E708:')
300  let d = {'k' : 4}
301  call assert_fails('let d.# = 5', 'E488:')
302  call assert_fails('let d.m += 5', 'E716:')
303  call assert_fails('let m = d[{]', 'E15:')
304  let l = [1, 2]
305  call assert_fails('let l[2] = 0', 'E684:')
306  call assert_fails('let l[0:1] = [1, 2, 3]', 'E710:')
307  call assert_fails('let l[-2:-3] = [3, 4]', 'E684:')
308  call assert_fails('let l[0:4] = [5, 6]', 'E711:')
309  call assert_fails('let l -= 2', 'E734:')
310  call assert_fails('let l += 2', 'E734:')
311  call assert_fails('let g:["a;b"] = 10', 'E461:')
312  call assert_fails('let g:.min = function("max")', 'E704:')
313  if has('channel')
314    let ch = test_null_channel()
315    call assert_fails('let ch += 1', 'E734:')
316  endif
317  call assert_fails('let name = "a" .. "b",', 'E488: Trailing characters: ,')
318
319  " This test works only when the language is English
320  if v:lang == "C" || v:lang =~ '^[Ee]n'
321    call assert_fails('let [a ; b;] = [10, 20]',
322          \ 'Double ; in list of variables')
323  endif
324endfunc
325
326func Test_let_heredoc_fails()
327  call assert_fails('let v =<< marker', 'E991:')
328  try
329    exe "let v =<< TEXT | abc | TEXT"
330    call assert_report('No exception thrown')
331  catch /E488:/
332  catch
333    call assert_report("Caught exception: " .. v:exception)
334  endtry
335
336  let text =<< trim END
337  func WrongSyntax()
338    let v =<< that there
339  endfunc
340  END
341  call writefile(text, 'XheredocFail')
342  call assert_fails('source XheredocFail', 'E1145:')
343  call delete('XheredocFail')
344
345  let text =<< trim CodeEnd
346  func MissingEnd()
347    let v =<< END
348  endfunc
349  CodeEnd
350  call writefile(text, 'XheredocWrong')
351  call assert_fails('source XheredocWrong', 'E1145:')
352  call delete('XheredocWrong')
353
354  let text =<< trim TEXTend
355    let v =<< " comment
356  TEXTend
357  call writefile(text, 'XheredocNoMarker')
358  call assert_fails('source XheredocNoMarker', 'E172:')
359  call delete('XheredocNoMarker')
360
361  let text =<< trim TEXTend
362    let v =<< text
363  TEXTend
364  call writefile(text, 'XheredocBadMarker')
365  call assert_fails('source XheredocBadMarker', 'E221:')
366  call delete('XheredocBadMarker')
367
368  call writefile(['let v =<< TEXT', 'abc'], 'XheredocMissingMarker')
369  call assert_fails('source XheredocMissingMarker', 'E990:')
370  call delete('XheredocMissingMarker')
371endfunc
372
373func Test_let_heredoc_trim_no_indent_marker()
374  let text =<< trim END
375  Text
376  with
377  indent
378END
379  call assert_equal(['Text', 'with', 'indent'], text)
380endfunc
381
382" Test for the setting a variable using the heredoc syntax
383func Test_let_heredoc()
384  let var1 =<< END
385Some sample text
386	Text with indent
387  !@#$%^&*()-+_={}|[]\~`:";'<>?,./
388END
389
390  call assert_equal(["Some sample text", "\tText with indent", "  !@#$%^&*()-+_={}|[]\\~`:\";'<>?,./"], var1)
391
392  let var2 =<< XXX
393Editor
394XXX
395  call assert_equal(['Editor'], var2)
396
397  let var3 =<<END
398END
399  call assert_equal([], var3)
400
401  let var3 =<<END
402vim
403
404end
405  END
406END
407END
408  call assert_equal(['vim', '', 'end', '  END', 'END '], var3)
409
410	let var1 =<< trim END
411	Line1
412	  Line2
413		Line3
414	 END
415	END
416  call assert_equal(['Line1', '  Line2', "\tLine3", ' END'], var1)
417
418  let var1 =<< trim !!!
419	Line1
420	 line2
421		Line3
422	!!!
423  !!!
424  call assert_equal(['Line1', ' line2', "\tLine3", '!!!',], var1)
425
426  let var1 =<< trim XX
427    Line1
428  XX
429  call assert_equal(['Line1'], var1)
430
431  " ignore "endfunc"
432  let var1 =<< END
433something
434endfunc
435END
436  call assert_equal(['something', 'endfunc'], var1)
437
438  " ignore "endfunc" with trim
439  let var1 =<< trim END
440  something
441  endfunc
442  END
443  call assert_equal(['something', 'endfunc'], var1)
444
445  " not concatenate lines
446  let var1 =<< END
447some
448  \thing
449  \ else
450END
451  call assert_equal(['some', '  \thing', '  \ else'], var1)
452
453  " ignore "python << xx"
454  let var1 =<<END
455something
456python << xx
457END
458  call assert_equal(['something', 'python << xx'], var1)
459
460  " ignore "python << xx" with trim
461  let var1 =<< trim END
462  something
463  python << xx
464  END
465  call assert_equal(['something', 'python << xx'], var1)
466
467  " ignore "append"
468  let var1 =<< E
469something
470app
471E
472  call assert_equal(['something', 'app'], var1)
473
474  " ignore "append" with trim
475  let var1 =<< trim END
476  something
477  app
478  END
479  call assert_equal(['something', 'app'], var1)
480
481  let check = []
482  if 0
483     let check =<< trim END
484       from heredoc
485     END
486  endif
487  call assert_equal([], check)
488
489  " unpack assignment
490  let [a, b, c] =<< END
491     x
492     \y
493     z
494END
495  call assert_equal(['     x', '     \y', '     z'], [a, b, c])
496endfunc
497
498" vim: shiftwidth=2 sts=2 expandtab
499