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
|
-- luacheck: globals describe it assert
local _99 = require("99")
local Tracking = require("99.state.tracking")
local test_utils = require("99.test.test_utils")
local eq = assert.are.same
local function run(provider, operation, status, prompt)
_99[operation]({ additional_prompt = prompt })
provider:resolve(status, "result")
end
describe("tracking", function()
it("serializes requests based on configured counts", function()
local previous_counts = vim.deepcopy(Tracking.__config.serialize_count)
Tracking.setup({
serialize_counts = {
vibe = 1,
search = 1,
tutorial = 3,
visual = 0,
},
})
local provider = test_utils.TestProvider.new()
_99.setup(test_utils.get_test_setup_options({
in_flight_options = { enable = false },
}, provider))
test_utils.create_file({ "local value = 1" }, "lua", 1, 0)
run(provider, "search", "success", "search one")
run(provider, "search", "success", "search two")
run(provider, "vibe", "success", "vibe one")
run(provider, "vibe", "success", "vibe two")
run(provider, "tutorial", "success", "tutorial one")
run(provider, "tutorial", "success", "tutorial two")
run(provider, "tutorial", "success", "tutorial three")
run(provider, "tutorial", "success", "tutorial four")
run(provider, "search", "failed", "search failed")
local serialized = _99.__get_state().tracking:serialize()
local actual_counts = {
search = 0,
vibe = 0,
tutorial = 0,
visual = 0,
}
for _, request in ipairs(serialized.requests) do
actual_counts[request.data.type] = actual_counts[request.data.type] + 1
end
eq(1, actual_counts.search)
eq(1, actual_counts.vibe)
eq(3, actual_counts.tutorial)
eq(0, actual_counts.visual)
eq(5, #serialized.requests)
Tracking.__config.serialize_count = previous_counts
end)
end)
|
