copilot.lua/tests/test_logger.lua at master · 03kev/copilot.lua · 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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
local eq = MiniTest.expect.equality
local child_helper = require("tests.child_helper")
local child = child_helper.new_child_neovim("test_logger")

local T = MiniTest.new_set({
  hooks = {
    pre_once = function() end,
    pre_case = function()
      child.run_pre_case(false)
    end,
    post_once = child.stop,
  },
})

T["logger()"] = MiniTest.new_set()

T["logger()"]["setup configures log file and levels"] = function()
  local result = child.lua([[
    local logger = require("copilot.logger")
    logger.setup({
      file = "/tmp/test-copilot.log",
      file_log_level = vim.log.levels.DEBUG,
      print_log_level = vim.log.levels.ERROR,
    })
    return {
      file = logger.log_file,
      file_level = logger.file_log_level,
      print_level = logger.print_log_level,
    }
  ]])
  eq(result.file, "/tmp/test-copilot.log")
  eq(result.file_level, 1) -- DEBUG
  eq(result.print_level, 4) -- ERROR
end

T["logger()"]["log level OFF suppresses all output"] = function()
  local result = child.lua([[
    local logger = require("copilot.logger")
    logger.file_log_level = vim.log.levels.OFF
    logger.print_log_level = vim.log.levels.OFF
    local notify_called = false
    local orig_notify = vim.notify
    vim.notify = function(msg, level)
      notify_called = true
      orig_notify(msg, level)
    end
    logger.error("test error")
    vim.notify = orig_notify
    return notify_called
  ]])
  eq(result, false)
end

T["logger()"]["handle_lsp_trace does not error on nil result"] = function()
  child.lua([[
    local logger = require("copilot.logger")
    logger.handle_lsp_trace(nil, nil, nil)
  ]])
end

T["logger()"]["handle_lsp_progress does not error on nil result"] = function()
  child.lua([[
    local logger = require("copilot.logger")
    logger.handle_lsp_progress(nil, nil, nil)
  ]])
end

T["logger()"]["handle_log_lsp_messages does not error on nil result"] = function()
  child.lua([[
    local logger = require("copilot.logger")
    logger.handle_log_lsp_messages(nil, nil, nil)
  ]])
end

T["logger()"]["handle_log_lsp_messages maps type 2 to warn level"] = function()
  local result = child.lua([[
    local logger = require("copilot.logger")
    local captured_level = nil
    local orig_log = logger.log
    logger.log = function(level, msg, ...)
      captured_level = level
    end
    logger.handle_log_lsp_messages(nil, { type = 2, message = "test warning" }, nil)
    logger.log = orig_log
    return captured_level
  ]])
  eq(result, 3) -- WARN
end

T["logger()"]["handle_log_lsp_messages maps type 1 to error level"] = function()
  local result = child.lua([[
    local logger = require("copilot.logger")
    local captured_level = nil
    local orig_log = logger.log
    logger.log = function(level, msg, ...)
      captured_level = level
    end
    logger.handle_log_lsp_messages(nil, { type = 1, message = "test error" }, nil)
    logger.log = orig_log
    return captured_level
  ]])
  eq(result, 4) -- ERROR
end

T["logger()"]["handle_log_lsp_messages forces abort errors to trace"] = function()
  local result = child.lua([[
    local logger = require("copilot.logger")
    local captured_level = nil
    local orig_log = logger.log
    logger.log = function(level, msg, ...)
      captured_level = level
    end
    logger.handle_log_lsp_messages(nil, {
      type = 1,
      message = "Request textDocument/copilotInlineEdit: AbortError: The operation was aborted"
    }, nil)
    logger.log = orig_log
    return captured_level
  ]])
  eq(result, 0) -- TRACE (forced down from ERROR)
end

return T