nvim-java-core/lua/java-core/api/test.lua at 3467886b6fd64dc2b67b07d9d2a14f847008ced9 · nvim-java/nvim-java-core · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
local log = require('java-core.utils.log')
local data_adapters = require('java-core.adapters')
local notify = require('java-core.utils.notify')

local DebugClient = require('java-core.ls.clients.java-debug-client')
local TestClient = require('java-core.ls.clients.java-test-client')

---@class java_core.TestApi
---@field private client java-core.JdtlsClient
---@field private debug_client java-core.DebugClient
---@field private test_client java-core.TestClient
---@field private runner java-dap.DapRunner
local M = {}

---Returns a new test helper client
---@param args { client: LspClient, runner: java-dap.DapRunner }
---@return java_core.TestApi
function M:new(args)
	local o = {
		client = args.client,
	}

	o.debug_client = DebugClient(args.client)
	o.test_client = TestClient(args.client)
	o.runner = args.runner

	setmetatable(o, self)
	self.__index = self

	return o
end

---Returns a list of test methods
---@param file_uri string uri of the class
---@return java-core.TestDetailsWithRange[] # list of test methods
function M:get_test_methods(file_uri)
	local classes = self.test_client:find_test_types_and_methods(file_uri)
	local methods = {}

	for _, class in ipairs(classes) do
		for _, method in ipairs(class.children) do
			---@diagnostic disable-next-line: inject-field
			method.class = class
			table.insert(methods, method)
		end
	end

	return methods
end

---comment
---@param buffer number
---@param report java_test.JUnitTestReport
---@param config? java-dap.DapLauncherConfigOverridable config to override the default values in test launcher config
function M:run_class_by_buffer(buffer, report, config)
	local tests = self:get_test_class_by_buffer(buffer)

	if #tests < 1 then
		notify.warn('No tests found in the current buffer')
		return
	end

	self:run_test(tests, report, config)
end

---Returns test classes in the given buffer
---@priate
---@param buffer integer
---@return java-core.TestDetailsWithChildrenAndRange # get test class details
function M:get_test_class_by_buffer(buffer)
	log.debug('finding test class by buffer')

	local uri = vim.uri_from_bufnr(buffer)
	return self.test_client:find_test_types_and_methods(uri)
end

---Run the given test
---@param tests java-core.TestDetails[]
---@param report java_test.JUnitTestReport
---@param config? java-dap.DapLauncherConfigOverridable config to override the default values in test launcher config
function M:run_test(tests, report, config)
	local launch_args = self.test_client:resolve_junit_launch_arguments(
		data_adapters.get_junit_launch_argument_params(tests)
	)

	local java_exec = self.debug_client:resolve_java_executable(
		launch_args.mainClass,
		launch_args.projectName
	)

	local dap_launcher_config =
		data_adapters.get_dap_launcher_config(launch_args, java_exec, {
			debug = true,
			label = 'Launch All Java Tests',
		})

	dap_launcher_config =
		vim.tbl_deep_extend('force', dap_launcher_config, config or {})

	self.runner:run_by_config(dap_launcher_config, report)
end

return M