refactor: use spyOn instead of vi.mock for agent loader mocks
Use spyOn which is more idiomatic for bun:test, as suggested in PR review.
This commit is contained in:
@@ -5,14 +5,7 @@ import type { DelegateTaskArgs } from "./types"
|
|||||||
import type { ExecutorContext } from "./executor-types"
|
import type { ExecutorContext } from "./executor-types"
|
||||||
import * as logger from "../../shared/logger"
|
import * as logger from "../../shared/logger"
|
||||||
import * as connectedProvidersCache from "../../shared/connected-providers-cache"
|
import * as connectedProvidersCache from "../../shared/connected-providers-cache"
|
||||||
|
import * as agentLoader from "../../features/claude-code-agent-loader"
|
||||||
const mockLoadUserAgents = vi.fn().mockReturnValue({})
|
|
||||||
const mockLoadProjectAgents = vi.fn().mockReturnValue({})
|
|
||||||
|
|
||||||
vi.mock("../../features/claude-code-agent-loader", () => ({
|
|
||||||
loadUserAgents: mockLoadUserAgents,
|
|
||||||
loadProjectAgents: mockLoadProjectAgents,
|
|
||||||
}))
|
|
||||||
|
|
||||||
function createBaseArgs(overrides?: Partial<DelegateTaskArgs>): DelegateTaskArgs {
|
function createBaseArgs(overrides?: Partial<DelegateTaskArgs>): DelegateTaskArgs {
|
||||||
return {
|
return {
|
||||||
@@ -45,14 +38,20 @@ function createExecutorContext(
|
|||||||
|
|
||||||
describe("resolveSubagentExecution", () => {
|
describe("resolveSubagentExecution", () => {
|
||||||
let logSpy: ReturnType<typeof spyOn> | undefined
|
let logSpy: ReturnType<typeof spyOn> | undefined
|
||||||
|
let mockLoadUserAgents: ReturnType<typeof spyOn>
|
||||||
|
let mockLoadProjectAgents: ReturnType<typeof spyOn>
|
||||||
|
|
||||||
beforeEach(() => {
|
beforeEach(() => {
|
||||||
mock.restore()
|
mock.restore()
|
||||||
logSpy = spyOn(logger, "log").mockImplementation(() => {})
|
logSpy = spyOn(logger, "log").mockImplementation(() => {})
|
||||||
|
mockLoadUserAgents = spyOn(agentLoader, "loadUserAgents").mockReturnValue({})
|
||||||
|
mockLoadProjectAgents = spyOn(agentLoader, "loadProjectAgents").mockReturnValue({})
|
||||||
})
|
})
|
||||||
|
|
||||||
afterEach(() => {
|
afterEach(() => {
|
||||||
logSpy?.mockRestore()
|
logSpy?.mockRestore()
|
||||||
|
mockLoadUserAgents?.mockRestore()
|
||||||
|
mockLoadProjectAgents?.mockRestore()
|
||||||
})
|
})
|
||||||
|
|
||||||
test("returns delegation error when agent discovery fails instead of silently proceeding", async () => {
|
test("returns delegation error when agent discovery fails instead of silently proceeding", async () => {
|
||||||
|
|||||||
Reference in New Issue
Block a user