tests: fix tests relying on Copilot (#2431)

Co-authored-by: Oli Morris <[email protected]>

Author: olimorris

tests/adapters/test_adapters.lua

@@ -8,6 +8,7 @@ T = new_set({
     pre_case = function()
       h.child_start(child)
       child.lua([[
+        h = require('tests.helpers')
         utils = require("codecompanion.utils.adapters")
 
         _G.test_adapter = {
@@ -290,7 +291,7 @@ end
 
 T["Adapter"]["can pass in the name of the model"] = function()
   local result = child.lua([[
-    require("codecompanion").setup({
+    h.setup_plugin({
       strategies = {
         chat = {
           adapter = {

tests/helpers.lua

@@ -16,10 +16,29 @@ local function mock_config()
 end
 
 ---Set up the CodeCompanion plugin with test configuration
+---@param config? table
 ---@return nil
-Helpers.setup_plugin = function()
-  local test_config = require("tests.config")
-  test_config.strategies.chat.adapter = "test_adapter"
+Helpers.setup_plugin = function(config)
+  local test_config = config or require("tests.config")
+
+  -- To be safe, ensure that we mock some of the Copilot adapter's features
+  -- that make HTTP requests. This slows tests down and makes them fail
+  -- in GitHub Actions.
+  local function mock_external_calls()
+    local ok, copilot = pcall(require, "codecompanion.adapters.http.copilot")
+    if ok then
+      copilot.schema.max_tokens.default = 16000
+
+      local get_models_ok, get_models = pcall(require, "codecompanion.adapters.http.copilot.get_models")
+      if get_models_ok then
+        get_models.choices = function(adapter, opts, provided_token)
+          return { "gpt-4.1" }
+        end
+      end
+    end
+  end
+
+  mock_external_calls()
 
   local codecompanion = require("codecompanion")
   codecompanion.setup(test_config)

tests/strategies/chat/memory/test_memory.lua

@@ -182,6 +182,7 @@ T["Memory.make() integration: memory is added to a real chat messages stack"] =
   child.lua(string.format(
     [[
     local h = require("tests.helpers")
+
     -- Initialize plugin with test config (h.setup_plugin returns the codecompanion module)
     local cc = h.setup_plugin()
 

tests/strategies/chat/test_chat.lua

@@ -13,7 +13,7 @@ T = new_set({
         codecompanion = require("codecompanion")
         h = require('tests.helpers')
         _G.chat, _G.tools = h.setup_chat_buffer()
-        ]])
+      ]])
     end,
     post_case = function()
       child.lua([[h.teardown_chat_buffer()]])
@@ -233,11 +233,11 @@ T["Chat"]["ftplugin window options override plugin defaults"] = function()
     _G.test_temp_dir = temp_dir
     _G.test_ftplugin_path = ftplugin_path
 
+    h = require('tests.helpers')
+
     -- Setup codecompanion
-    codecompanion = require("codecompanion")
+    local codecompanion = h.setup_plugin()
     codecompanion.setup()
-
-    -- Open a new chat
     codecompanion.chat()
 
     -- Manually source the ftplugin file to simulate user's after/ftplugin

tests/test_cmds.lua

@@ -12,7 +12,7 @@ T = new_set({
         h = require('tests.helpers')
         config = require('codecompanion.config')
         config.memory.opts.chat.enabled = false
-        h.setup_plugin()
+        h.setup_plugin(config)
       ]])
     end,
     post_once = child.stop,
@@ -22,6 +22,7 @@ T = new_set({
 T["cmds"] = new_set()
 T["cmds"][":CodeCompanionChat"] = function()
   child.lua([[
+
     -- Mock the submit function
     local original = h.mock_submit("This is a mocked response: 1 + 1 = 2")
 
@@ -161,7 +162,7 @@ T["cmds"]["chat variable syntax highlighting"] = function()
     vim.bo[buf].filetype = 'codecompanion'
     vim.cmd('doautocmd FileType codecompanion')
     vim.cmd('doautocmd BufEnter ' .. tostring(buf))
-    
+
     -- wait for some time so that the `vim.schedule`ed :syntax commands are executed
     vim.wait(10)
 

tests/test_prompt_library.lua

@@ -8,7 +8,8 @@ T = new_set({
     pre_case = function()
       h.child_start(child)
       child.lua([[
-        codecompanion = require("codecompanion")
+        h = require('tests.helpers')
+        codecompanion = h.setup_plugin()
       ]])
     end,
     post_once = child.stop,