test: keep media runtime tests on same-directory provider mocks

This commit is contained in:
Peter Steinberger
2026-04-08 17:15:56 +01:00
parent 8f67f156ee
commit 253ecd2a5d
3 changed files with 30 additions and 3 deletions

View File

@@ -1,4 +1,4 @@
import { beforeEach, describe, expect, it } from "vitest";
import { beforeEach, describe, expect, it, vi } from "vitest";
import {
getMediaGenerationRuntimeMocks,
resetImageGenerationRuntimeMocks,
@@ -9,6 +9,15 @@ import type { ImageGenerationProvider } from "./types.js";
const mocks = getMediaGenerationRuntimeMocks();
vi.mock("./model-ref.js", () => ({
parseImageGenerationModelRef: mocks.parseImageGenerationModelRef,
}));
vi.mock("./provider-registry.js", () => ({
getImageGenerationProvider: mocks.getImageGenerationProvider,
listImageGenerationProviders: mocks.listImageGenerationProviders,
}));
describe("image-generation runtime", () => {
beforeEach(() => {
resetImageGenerationRuntimeMocks();

View File

@@ -1,4 +1,4 @@
import { beforeEach, describe, expect, it } from "vitest";
import { beforeEach, describe, expect, it, vi } from "vitest";
import {
getMediaGenerationRuntimeMocks,
resetMusicGenerationRuntimeMocks,
@@ -9,6 +9,15 @@ import type { MusicGenerationProvider } from "./types.js";
const mocks = getMediaGenerationRuntimeMocks();
vi.mock("./model-ref.js", () => ({
parseMusicGenerationModelRef: mocks.parseMusicGenerationModelRef,
}));
vi.mock("./provider-registry.js", () => ({
getMusicGenerationProvider: mocks.getMusicGenerationProvider,
listMusicGenerationProviders: mocks.listMusicGenerationProviders,
}));
describe("music-generation runtime", () => {
beforeEach(() => {
resetMusicGenerationRuntimeMocks();

View File

@@ -1,4 +1,4 @@
import { beforeEach, describe, expect, it } from "vitest";
import { beforeEach, describe, expect, it, vi } from "vitest";
import {
getMediaGenerationRuntimeMocks,
resetVideoGenerationRuntimeMocks,
@@ -9,6 +9,15 @@ import type { VideoGenerationProvider } from "./types.js";
const mocks = getMediaGenerationRuntimeMocks();
vi.mock("./model-ref.js", () => ({
parseVideoGenerationModelRef: mocks.parseVideoGenerationModelRef,
}));
vi.mock("./provider-registry.js", () => ({
getVideoGenerationProvider: mocks.getVideoGenerationProvider,
listVideoGenerationProviders: mocks.listVideoGenerationProviders,
}));
describe("video-generation runtime", () => {
beforeEach(() => {
resetVideoGenerationRuntimeMocks();