@@ -53,6 +53,9 @@ const mediaRuntimeMocks = vi.hoisted(() => {
5353vi.fn<(raw?: string) => ModelRef | undefined>(parseGenerationModelRef),
5454parseVideoGenerationModelRef:
5555vi.fn<(raw?: string) => ModelRef | undefined>(parseGenerationModelRef),
56+ensureAuthProfileStore: vi.fn(() => ({ version: 1, profiles: {} })),
57+listProfilesForProvider: vi.fn(() => []),
58+resolveEnvApiKey: vi.fn(() => undefined),
5659resolveAgentModelFallbackValues: vi.fn<(value: unknown) => string[]>(() => []),
5760resolveAgentModelPrimaryValue: vi.fn<(value: unknown) => string | undefined>(() => undefined),
5861resolveProviderAuthEnvVarCandidates: vi.fn(() => ({})),
@@ -61,10 +64,20 @@ const mediaRuntimeMocks = vi.hoisted(() => {
6164};
6265});
636667+vi.mock("../../../src/agents/auth-profiles.js", () => ({
68+ensureAuthProfileStore: mediaRuntimeMocks.ensureAuthProfileStore,
69+listProfilesForProvider: mediaRuntimeMocks.listProfilesForProvider,
70+}));
71+vi.mock("../../../src/agents/defaults.js", () => ({
72+DEFAULT_PROVIDER: "openai",
73+}));
6474vi.mock("../../../src/agents/failover-error.js", () => ({
6575describeFailoverError: mediaRuntimeMocks.describeFailoverError,
6676isFailoverError: mediaRuntimeMocks.isFailoverError,
6777}));
78+vi.mock("../../../src/agents/model-auth-env.js", () => ({
79+resolveEnvApiKey: mediaRuntimeMocks.resolveEnvApiKey,
80+}));
6881vi.mock("../../../src/config/model-input.js", () => ({
6982resolveAgentModelFallbackValues: mediaRuntimeMocks.resolveAgentModelFallbackValues,
7083resolveAgentModelPrimaryValue: mediaRuntimeMocks.resolveAgentModelPrimaryValue,
@@ -104,6 +117,7 @@ export function getMediaGenerationRuntimeMocks() {
104117}
105118106119export function resetImageGenerationRuntimeMocks(): void {
120+resetSharedRuntimeImportMocks();
107121resetGenerationRuntimeMocks({
108122 ...mediaRuntimeMocks,
109123getProvider: mediaRuntimeMocks.getImageGenerationProvider,
@@ -113,6 +127,7 @@ export function resetImageGenerationRuntimeMocks(): void {
113127}
114128115129export function resetMusicGenerationRuntimeMocks(): void {
130+resetSharedRuntimeImportMocks();
116131resetGenerationRuntimeMocks({
117132 ...mediaRuntimeMocks,
118133getProvider: mediaRuntimeMocks.getMusicGenerationProvider,
@@ -122,10 +137,20 @@ export function resetMusicGenerationRuntimeMocks(): void {
122137}
123138124139export function resetVideoGenerationRuntimeMocks(): void {
140+resetSharedRuntimeImportMocks();
125141resetGenerationRuntimeMocks({
126142 ...mediaRuntimeMocks,
127143getProvider: mediaRuntimeMocks.getVideoGenerationProvider,
128144listProviders: mediaRuntimeMocks.listVideoGenerationProviders,
129145parseModelRef: mediaRuntimeMocks.parseVideoGenerationModelRef,
130146});
131147}
148+149+function resetSharedRuntimeImportMocks(): void {
150+mediaRuntimeMocks.ensureAuthProfileStore.mockReset();
151+mediaRuntimeMocks.ensureAuthProfileStore.mockReturnValue({ version: 1, profiles: {} });
152+mediaRuntimeMocks.listProfilesForProvider.mockReset();
153+mediaRuntimeMocks.listProfilesForProvider.mockReturnValue([]);
154+mediaRuntimeMocks.resolveEnvApiKey.mockReset();
155+mediaRuntimeMocks.resolveEnvApiKey.mockReturnValue(undefined);
156+}
此内容由惯性聚合(RSS阅读器)自动聚合整理，仅供阅读参考。 原文来自 — 版权归原作者所有。