davidl/MyTube
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

test: Mock database and dependencies in taskCleanup.test file

Browse Source
This commit is contained in:
Peifan Li
2025-12-28 15:46:40 -05:00
parent 23849de390
commit 6d07967a04
1 changed files with 18 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
backend/src/__tests__/services/continuousDownload/taskCleanup.test.ts
View File

@@ -1,9 +1,17 @@
import { beforeEach, describe, expect, it, vi } from 'vitest';
import { TaskCleanup } from '../../../services/continuousDownload/taskCleanup';
import { ContinuousDownloadTask } from '../../../services/continuousDownload/types';
import { VideoUrlFetcher } from '../../../services/continuousDownload/videoUrlFetcher';
import * as storageService from '../../../services/storageService';
import { logger } from '../../../utils/logger';
// Mock database first to prevent initialization errors
vi.mock('../../../db', () => ({
db: {
select: vi.fn(),
insert: vi.fn(),
delete: vi.fn(),
update: vi.fn(),
},
sqlite: {
prepare: vi.fn(),
},
}));
// Mock dependencies
vi.mock('../../../services/continuousDownload/videoUrlFetcher');
@@ -48,6 +56,11 @@ vi.mock('fs-extra', () => ({
}
}));
import { TaskCleanup } from '../../../services/continuousDownload/taskCleanup';
import { ContinuousDownloadTask } from '../../../services/continuousDownload/types';
import { VideoUrlFetcher } from '../../../services/continuousDownload/videoUrlFetcher';
import * as storageService from '../../../services/storageService';
import { logger } from '../../../utils/logger';
import { getVideoInfo } from '../../../services/downloadService';
import { cleanupVideoArtifacts } from '../../../utils/downloadUtils';