acoolname/neotest-python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: search runtime for python script

Browse Source 
See https://github.com/nvim-neotest/neotest/issues/118
This commit is contained in:
Rónán Carrigan
2022-10-01 14:48:18 +01:00
parent a0c5c567fb
commit 143aacd04e
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
lua/neotest-python/init.lua
View File

@@ -1,14 +1,17 @@
local async = require("neotest.async")
local Path = require("plenary.path")
local lib = require("neotest.lib")
local base = require("neotest-python.base")
local function script_path()
local str = debug.getinfo(2, "S").source:sub(2)
return str:match(("(.*%s)"):format(lib.files.sep))
end
local function get_script()
local paths = vim.api.nvim_get_runtime_file("neotest.py", true)
for _, path in ipairs(paths) do
if vim.endswith(path, ("neotest-python%sneotest.py"):format(lib.files.sep)) then
return path
end
end
local python_script = (Path.new(script_path()):parent():parent() / "neotest.py").filename
error("neotest.py not found")
end
local dap_args
local is_test_file = base.is_test_file
@@ -104,9 +107,7 @@ function PythonNeotestAdapter.build_spec(args)
local position = args.tree:data()
local results_path = async.fn.tempname()
local stream_path = async.fn.tempname()
local x = io.open(stream_path, "w")
x:write("")
x:close()
lib.files.write(stream_path, "")
local root = PythonNeotestAdapter.root(position.path)
local python = get_python(root)
@@ -127,7 +128,7 @@ function PythonNeotestAdapter.build_spec(args)
end
local command = vim.tbl_flatten({
python,
python_script,
get_script(),
script_args,
})
local strategy_config = get_strategy_config(args.strategy, python, python_script, script_args)