fillstruct: add tests for gopls mode

fatih · Sep 20, 2020 · b00e272 · b00e272
1 parent cc6c738
commit b00e272
Showing 1 changed file with 129 additions and 0 deletions.
diff --git a/autoload/go/fillstruct_test.vim b/autoload/go/fillstruct_test.vim
@@ -4,6 +4,7 @@ set cpo&vim
 
 func! Test_fillstruct() abort
   try
+    let g:go_fillstruct_mode = 'fillstruct'
     let l:tmp = gotest#write_file('a/a.go', [
           \ 'package a',
           \ 'import "net/mail"',
@@ -16,12 +17,14 @@ func! Test_fillstruct() abort
           \ '\tAddress: "",',
           \ '}'])
   finally
+    unlet g:go_fillstruct_mode
     call delete(l:tmp, 'rf')
   endtry
 endfunc
 
 func! Test_fillstruct_line() abort
   try
+    let g:go_fillstruct_mode = 'fillstruct'
     let l:tmp = gotest#write_file('a/a.go', [
           \ 'package a',
           \ 'import "net/mail"',
@@ -34,12 +37,14 @@ func! Test_fillstruct_line() abort
           \ '\tAddress: "",',
           \ '}'])
   finally
+    unlet g:go_fillstruct_mode
     call delete(l:tmp, 'rf')
   endtry
 endfunc
 
 func! Test_fillstruct_two_line() abort
   try
+    let g:go_fillstruct_mode = 'fillstruct'
     let l:tmp = gotest#write_file('a/a.go', [
           \ 'package a',
           \ 'import (',
@@ -62,12 +67,14 @@ func! Test_fillstruct_two_line() abort
           \ '\tAddress: "",',
           \ '}) }'])
   finally
+    unlet g:go_fillstruct_mode
     call delete(l:tmp, 'rf')
   endtry
 endfunc
 
 func! Test_fillstruct_two_cursor() abort
   try
+    let g:go_fillstruct_mode = 'fillstruct'
     let l:tmp = gotest#write_file('a/a.go', [
           \ 'package a',
           \ 'import (',
@@ -87,10 +94,132 @@ func! Test_fillstruct_two_cursor() abort
           \ '\tAddress: "",',
           \ '}) }'])
   finally
+    unlet g:go_fillstruct_mode
     call delete(l:tmp, 'rf')
   endtry
 endfunc
 
+func! Test_gopls_fillstruct() abort
+  try
+    let g:go_fillstruct_mode = 'gopls'
+    let l:tmp = gotest#write_file('a/a.go', [
+          \ 'package a',
+          \ 'import "net/mail"',
+          \ "var addr = mail.\x1fAddress{}"])
+
+    call go#fillstruct#FillStruct()
+
+    let start = reltime()
+    while &modified == 0 && reltimefloat(reltime(start)) < 10
+      sleep 100m
+    endwhile
+
+    call gotest#assert_buffer(1, [
+          \ 'var addr = mail.Address{',
+          \ '\tName:    "",',
+          \ '\tAddress: "",',
+          \ '}'])
+  finally
+    unlet g:go_fillstruct_mode
+    call delete(l:tmp, 'rf')
+  endtry
+endfunc
+
+func! Test_gopls_fillstruct_line() abort
+  try
+    let g:go_fillstruct_mode = 'gopls'
+    let l:tmp = gotest#write_file('a/a.go', [
+          \ 'package a',
+          \ 'import "net/mail"',
+          \ "\x1f" . 'var addr = mail.Address{}'])
+
+    call go#fillstruct#FillStruct()
+
+    let start = reltime()
+    while &modified == 0 && reltimefloat(reltime(start)) < 10
+      sleep 100m
+    endwhile
+
+    call gotest#assert_buffer(1, [
+          \ 'var addr = mail.Address{',
+          \ '\tName:    "",',
+          \ '\tAddress: "",',
+          \ '}'])
+  finally
+    unlet g:go_fillstruct_mode
+    call delete(l:tmp, 'rf')
+  endtry
+endfunc
+
+func! Test_gopls_fillstruct_two_line() abort
+  try
+    let g:go_fillstruct_mode = 'gopls'
+    let l:tmp = gotest#write_file('a/a.go', [
+          \ 'package a',
+          \ 'import (',
+          \ '"fmt"',
+          \ '"net/mail"',
+          \ ')',
+          \ "\x1f" . 'func x() { fmt.Println(mail.Address{}, mail.Address{}) }'])
+          \ "\x1f" . 'func x() { fmt.Println(mail.Address{}, mail.Address{}) }'])
+
+    call go#fillstruct#FillStruct()
+
+    let start = reltime()
+    while &modified == 0 && reltimefloat(reltime(start)) < 10
+      sleep 100m
+    endwhile
+
+    call gotest#assert_buffer(1, [
+          \ 'import (',
+          \ '"fmt"',
+          \ '"net/mail"',
+          \ ')',
+          \ 'func x() { fmt.Println(mail.Address{',
+          \ '\tName:    "",',
+          \ '\tAddress: "",',
+          \ '}, mail.Address{',
+          \ '\tName:    "",',
+          \ '\tAddress: "",',
+          \ '}) }'])
+  finally
+    unlet g:go_fillstruct_mode
+    call delete(l:tmp, 'rf')
+  endtry
+endfunc
+
+func! Test_gopls_fillstruct_two_cursor() abort
+  try
+    let g:go_fillstruct_mode = 'gopls'
+    let l:tmp = gotest#write_file('a/a.go', [
+          \ 'package a',
+          \ 'import (',
+          \ '"fmt"',
+          \ '"net/mail"',
+          \ ')',
+          \ "func x() { fmt.Println(mail.Address{}, mail.Ad\x1fdress{}) }"])
+
+    call go#fillstruct#FillStruct()
+
+    let start = reltime()
+    while &modified == 0 && reltimefloat(reltime(start)) < 10
+      sleep 100m
+    endwhile
+
+    call gotest#assert_buffer(1, [
+          \ 'import (',
+          \ '"fmt"',
+          \ '"net/mail"',
+          \ ')',
+          \ 'func x() { fmt.Println(mail.Address{}, mail.Address{',
+          \ '\tName:    "",',
+          \ '\tAddress: "",',
+          \ '}) }'])
+  finally
+    unlet g:go_fillstruct_mode
+    call delete(l:tmp, 'rf')
+  endtry
+endfunc
 " restore Vi compatibility settings
 let &cpo = s:cpo_save
 unlet s:cpo_save