✨ Support toggle test

Nguyen-Hoang-Nam · Mar 5, 2022 · f2a3c9e · f2a3c9e
1 parent fc6b276
commit f2a3c9e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -405,7 +405,7 @@ Check [wiki](https://github.com/Nguyen-Hoang-Nam/nvim-dotfiles-kitty/wiki/Test)
 | &lt;Leader&gt;q | Turn off search's highlight |
 | &lt;Leader&gt;r | Rename variable             |
 | &lt;Leader&gt;s | Save file                   |
-| &lt;Leader&gt;t | Run test                    |
+| &lt;Leader&gt;t | Toggle test                 |
 | &lt;Leader&gt;u | Toggle debug sidebar        |
 | &lt;Leader&gt;v | Run Restful API             |
 | &lt;Leader&gt;w | Delete current buffer       |

diff --git a/lua/mappings.lua b/lua/mappings.lua
@@ -72,7 +72,7 @@ map(
     end)
 )
 
-map('n', '<Leader>t', [[<Cmd>Ultest<CR>]], cmd_options)
+map('n', '<Leader>t', [[<Cmd>Ultest<CR>]], cmd_option(utils_core.toggle_test))
 map('n', ']t', '<Plug>(ultest-next-fail)', { noremap = false, silent = true })
 map('n', '[t', '<Plug>(ultest-prev-fail)', { noremap = false, silent = true })
 

diff --git a/lua/utils/core.lua b/lua/utils/core.lua
@@ -219,7 +219,16 @@ function M.rename_popup()
 
     -- Normal to insert
     api.nvim_feedkeys('A', 'n', true)
-    -- api.nvim_command('startinsert')
 end
 
+function M.toggle_test()
+    local current_test = api.nvim_eval('ultest#status()')
+    if current_test['tests'] > 0 then
+        if current_test['passed'] > 0 or current_test['failed'] > 0 or current_test['running'] > 0 then
+            vim.api.nvim_command('UltestClear')
+        else
+            vim.api.nvim_command('UltestNearest')
+        end
+    end
+end
 return M