fix: tse shouldn't toggle \begin{document}

refer: #2210
lervag · Oct 18, 2021 · 05dbb2c · 05dbb2c
1 parent 0824ade
commit 05dbb2c
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 1 deletion.
diff --git a/autoload/vimtex/env.vim b/autoload/vimtex/env.vim
@@ -96,7 +96,8 @@ endfunction
 
 function! vimtex#env#toggle_star() abort " {{{1
   let [l:open, l:close] = vimtex#delim#get_surrounding('env_tex')
-  if empty(l:open) | return | endif
+  if empty(l:open)
+        \ || l:open.name ==# 'document' | return | endif
 
   call vimtex#env#change(l:open, l:close,
         \ l:open.starred ? l:open.name : l:open.name . '*')

diff --git a/test/test-env/Makefile b/test/test-env/Makefile
@@ -0,0 +1,14 @@
+MYVIM ?= nvim --headless
+
+INMAKE := 1
+export INMAKE
+
+TESTS := $(wildcard *.vim)
+TESTS := $(TESTS:.vim=)
+
+.PHONY: test $(TESTS)
+
+test: $(TESTS)
+
+$(TESTS):
+	@$(MYVIM) -u $@.vim
diff --git a/test/test-env/test-toggle-star.tex b/test/test-env/test-toggle-star.tex
@@ -0,0 +1,6 @@
+\documentclass{minimal}
+\begin{document}
+
+Hello World!
+
+\end{document}
diff --git a/test/test-env/test-toggle-star.vim b/test/test-env/test-toggle-star.vim
@@ -0,0 +1,14 @@
+set nocompatible
+set runtimepath^=../..
+filetype plugin on
+syntax on
+
+nnoremap q :qall!<cr>
+
+silent edit test-toggle-star.tex
+
+normal! 4G
+call vimtex#env#toggle_star()
+call assert_equal('\end{document}', getline('$'))
+
+call vimtex#test#finished()