Improve support for python package manage: pipenv, poetry and uv (#4825)

1 files changed, 31 insertions, 11 deletions

diff --git a/test/fixers/test_autoflake_fixer_callback.vader b/test/fixers/test_autoflake_fixer_callback.vader
index 91fc62b5..2abe5fc7 100644
--- a/test/fixers/test_autoflake_fixer_callback.vader
+++ b/test/fixers/test_autoflake_fixer_callback.vader
@@ -18,32 +18,52 @@ After:
 
   call ale#test#RestoreDirectory()
 
-Execute(The autoflake callback should return the correct default values):
-  AssertEqual
-  \ 0,
-  \ ale#fixers#autoflake#Fix(bufnr(''))
+Execute(The autoflake callback should include options):
+  let g:ale_python_autoflake_options = '--some-option'
 
   silent execute 'file ' . fnameescape(g:dir . '/../test-files/python/with_virtualenv/subdir/foo/bar.py')
   AssertEqual
   \ {
   \   'command': ale#Escape(ale#path#Simplify(g:dir . '/../test-files/python/with_virtualenv/env/' . b:bin_dir . '/autoflake'))
+  \     . ' --some-option'
   \     . ' --in-place '
   \     . ' %t',
   \   'read_temporary_file': 1,
   \ },
   \ ale#fixers#autoflake#Fix(bufnr(''))
 
+Execute(pipenv is detected when python_autoflake_auto_pipenv is set):
+  let g:ale_python_autoflake_auto_pipenv = 1
 
-Execute(The autoflake callback should include options):
-  let g:ale_python_autoflake_options = '--some-option'
+  call ale#test#SetFilename('../test-files/python/pipenv/whatever.py')
 
-  silent execute 'file ' . fnameescape(g:dir . '/../test-files/python/with_virtualenv/subdir/foo/bar.py')
   AssertEqual
   \ {
-  \   'command': ale#Escape(ale#path#Simplify(g:dir . '/../test-files/python/with_virtualenv/env/' . b:bin_dir . '/autoflake'))
-  \     . ' --some-option'
-  \     . ' --in-place '
-  \     . ' %t',
+  \   'command': ale#Escape('pipenv') . ' run autoflake --in-place  %t',
+  \   'read_temporary_file': 1,
+  \ },
+  \ ale#fixers#autoflake#Fix(bufnr(''))
+
+Execute(Poetry is detected when python_autoflake_auto_poetry is set):
+  let g:ale_python_autoflake_auto_poetry = 1
+
+  call ale#test#SetFilename('../test-files/python/poetry/whatever.py')
+
+  AssertEqual
+  \ {
+  \   'command': ale#Escape('poetry') . ' run autoflake --in-place  %t',
+  \   'read_temporary_file': 1,
+  \ },
+  \ ale#fixers#autoflake#Fix(bufnr(''))
+
+Execute(uv is detected when python_autoflake_auto_uv is set):
+  let g:ale_python_autoflake_auto_uv = 1
+
+  call ale#test#SetFilename('../test-files/python/uv/whatever.py')
+
+  AssertEqual
+  \ {
+  \   'command': ale#Escape('uv') . ' run autoflake --in-place  %t',
   \   'read_temporary_file': 1,
   \ },
   \ ale#fixers#autoflake#Fix(bufnr(''))