Files
dc-recorder/tests/routes/mediaRoutes.test.ts
MythEclipse d50ce8698f feat: implement media echo fix and YouTube screenshare design
- Introduced a new `ScreenShareController` to manage YouTube screenshare functionality.
- Updated `DiscordPlayer` to track ownership of audio streams, preventing conflicts between music playback and screenshare.
- Added error handling for various states including voice connection checks and media busy states.
- Created unit tests for `ScreenShareController` and `DiscordPlayer` ownership rules to ensure correct functionality.
- Added documentation for the new media echo fix and screenshare design.
2026-05-16 15:48:28 +07:00

158 lines
3.7 KiB
TypeScript

import type { Request, Response } from "express";
import { describe, expect, it, vi } from "vitest";
import { createMediaRoutes } from "../../src/routes/mediaRoutes";
function getHandler(
router: ReturnType<typeof createMediaRoutes>,
path: string,
method: string,
) {
const layer = router.stack.find((item) => item.route?.path === path);
return layer?.route?.stack.find((item) => item.method === method)?.handle;
}
describe("createMediaRoutes", () => {
it("returns media status", async () => {
const controller = {
getState: vi.fn(() => ({
playing: false,
activeMode: null,
current: null,
queue: [],
})),
queue: vi.fn(),
skip: vi.fn(),
stop: vi.fn(),
};
const handler = getHandler(
createMediaRoutes(controller),
"/media/status",
"get",
);
const json = vi.fn();
await handler?.({} as Request, { json } as unknown as Response, vi.fn());
expect(json).toHaveBeenCalledWith({
playing: false,
activeMode: null,
current: null,
queue: [],
});
});
it("queues a source", async () => {
const state = { playing: true, activeMode: null, current: null, queue: [] };
const controller = {
getState: vi.fn(),
queue: vi.fn(async () => state),
skip: vi.fn(),
stop: vi.fn(),
};
const handler = getHandler(
createMediaRoutes(controller),
"/media/queue",
"post",
);
const json = vi.fn();
await handler?.(
{ body: { source: "https://example.com/song.mp3" } } as Request,
{ json } as unknown as Response,
vi.fn(),
);
expect(controller.queue).toHaveBeenCalledWith(
"https://example.com/song.mp3",
{ mode: "music" },
);
expect(json).toHaveBeenCalledWith(state);
});
it("queues a screen source", async () => {
const state = {
playing: true,
activeMode: "screen" as const,
current: null,
queue: [],
};
const controller = {
getState: vi.fn(),
queue: vi.fn(async () => state),
skip: vi.fn(),
stop: vi.fn(),
};
const handler = getHandler(
createMediaRoutes(controller),
"/media/queue",
"post",
);
const json = vi.fn();
await handler?.(
{ body: { source: "https://youtu.be/video", mode: "screen" } } as Request,
{ json } as unknown as Response,
vi.fn(),
);
expect(controller.queue).toHaveBeenCalledWith("https://youtu.be/video", {
mode: "screen",
});
expect(json).toHaveBeenCalledWith(state);
});
it("passes invalid mode errors to Express", async () => {
const controller = {
getState: vi.fn(),
queue: vi.fn(),
skip: vi.fn(),
stop: vi.fn(),
};
const handler = getHandler(
createMediaRoutes(controller),
"/media/queue",
"post",
);
const next = vi.fn();
await handler?.(
{
body: { source: "https://example.com/song.mp3", mode: "video" },
} as Request,
{ json: vi.fn() } as unknown as Response,
next,
);
expect(next.mock.calls[0][0]).toMatchObject({
code: "INVALID_MEDIA_MODE",
statusCode: 400,
});
});
it("passes missing source errors to Express", async () => {
const controller = {
getState: vi.fn(),
queue: vi.fn(),
skip: vi.fn(),
stop: vi.fn(),
};
const handler = getHandler(
createMediaRoutes(controller),
"/media/queue",
"post",
);
const next = vi.fn();
await handler?.(
{ body: {} } as Request,
{ json: vi.fn() } as unknown as Response,
next,
);
expect(next.mock.calls[0][0]).toMatchObject({
code: "MISSING_MEDIA_SOURCE",
statusCode: 400,
});
});
});