vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 12:00:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

test: narrow hotspot mocks

Browse Source
This commit is contained in:
Peter Steinberger
2026-04-17 01:53:16 +01:00
parent cd330f5f98
commit dbc8179f31
4 changed files with 73 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
test/helpers/media-generation/runtime-module-mocks.ts
View File

@@ -53,6 +53,9 @@ const mediaRuntimeMocks = vi.hoisted(() => {
vi.fn<(raw?: string) => ModelRef | undefined>(parseGenerationModelRef),
parseVideoGenerationModelRef:
vi.fn<(raw?: string) => ModelRef | undefined>(parseGenerationModelRef),
ensureAuthProfileStore: vi.fn(() => ({ version: 1, profiles: {} })),
listProfilesForProvider: vi.fn(() => []),
resolveEnvApiKey: vi.fn(() => undefined),
resolveAgentModelFallbackValues: vi.fn<(value: unknown) => string[]>(() => []),
resolveAgentModelPrimaryValue: vi.fn<(value: unknown) => string | undefined>(() => undefined),
resolveProviderAuthEnvVarCandidates: vi.fn(() => ({})),
@@ -61,10 +64,20 @@ const mediaRuntimeMocks = vi.hoisted(() => {
};
});
vi.mock("../../../src/agents/auth-profiles.js", () => ({
ensureAuthProfileStore: mediaRuntimeMocks.ensureAuthProfileStore,
listProfilesForProvider: mediaRuntimeMocks.listProfilesForProvider,
}));
vi.mock("../../../src/agents/defaults.js", () => ({
DEFAULT_PROVIDER: "openai",
}));
vi.mock("../../../src/agents/failover-error.js", () => ({
describeFailoverError: mediaRuntimeMocks.describeFailoverError,
isFailoverError: mediaRuntimeMocks.isFailoverError,
}));
vi.mock("../../../src/agents/model-auth-env.js", () => ({
resolveEnvApiKey: mediaRuntimeMocks.resolveEnvApiKey,
}));
vi.mock("../../../src/config/model-input.js", () => ({
resolveAgentModelFallbackValues: mediaRuntimeMocks.resolveAgentModelFallbackValues,
resolveAgentModelPrimaryValue: mediaRuntimeMocks.resolveAgentModelPrimaryValue,
@@ -104,6 +117,7 @@ export function getMediaGenerationRuntimeMocks() {
}
export function resetImageGenerationRuntimeMocks(): void {
resetSharedRuntimeImportMocks();
resetGenerationRuntimeMocks({
...mediaRuntimeMocks,
getProvider: mediaRuntimeMocks.getImageGenerationProvider,
@@ -113,6 +127,7 @@ export function resetImageGenerationRuntimeMocks(): void {
}
export function resetMusicGenerationRuntimeMocks(): void {
resetSharedRuntimeImportMocks();
resetGenerationRuntimeMocks({
...mediaRuntimeMocks,
getProvider: mediaRuntimeMocks.getMusicGenerationProvider,
@@ -122,6 +137,7 @@ export function resetMusicGenerationRuntimeMocks(): void {
}
export function resetVideoGenerationRuntimeMocks(): void {
resetSharedRuntimeImportMocks();
resetGenerationRuntimeMocks({
...mediaRuntimeMocks,
getProvider: mediaRuntimeMocks.getVideoGenerationProvider,
@@ -129,3 +145,12 @@ export function resetVideoGenerationRuntimeMocks(): void {
parseModelRef: mediaRuntimeMocks.parseVideoGenerationModelRef,
});
}
function resetSharedRuntimeImportMocks(): void {
mediaRuntimeMocks.ensureAuthProfileStore.mockReset();
mediaRuntimeMocks.ensureAuthProfileStore.mockReturnValue({ version: 1, profiles: {} });
mediaRuntimeMocks.listProfilesForProvider.mockReset();
mediaRuntimeMocks.listProfilesForProvider.mockReturnValue([]);
mediaRuntimeMocks.resolveEnvApiKey.mockReset();
mediaRuntimeMocks.resolveEnvApiKey.mockReturnValue(undefined);
}