diff --git a/autoload/gotests.vim b/autoload/gotests.vim index 84983c3..0a27d6e 100644 --- a/autoload/gotests.vim +++ b/autoload/gotests.vim @@ -24,8 +24,8 @@ function! gotests#tests() range let tmplDir = '' if (!empty(g:gotests_template_dir)) let tmplDir = '-template_dir ' . shellescape(g:gotests_template_dir) - elseif (!empty(g:gotests_template)) - let tmplDir = '-template ' . shellescape(g:gotests_template) + elseif (!empty(g:gotests_template)) + let tmplDir = '-template ' . shellescape(g:gotests_template) endif let file = expand('%') @@ -44,8 +44,8 @@ function! gotests#alltests() abort let tmplDir = '' if (!empty(g:gotests_template_dir)) let tmplDir = '-template_dir ' . shellescape(g:gotests_template_dir) - elseif (!empty(g:gotests_template)) - let tmplDir = '-template ' . shellescape(g:gotests_template) + elseif (!empty(g:gotests_template)) + let tmplDir = '-template ' . shellescape(g:gotests_template) endif let file = expand('%') diff --git a/plugin/vim-gotests.vim b/plugin/vim-gotests.vim index c2d3045..a626366 100644 --- a/plugin/vim-gotests.vim +++ b/plugin/vim-gotests.vim @@ -14,5 +14,9 @@ if !exists('g:gotests_template_dir') let g:gotests_template_dir = '' endif +if !exists('g:gotests_template') + let g:gotests_template = '' +endif + command! -range GoTests ,call gotests#tests() command! GoTestsAll call gotests#alltests()