tools/knightcrawler
1
0
Fork 0
mirror of https://github.com/knightcrawler-stremio/knightcrawler.git synced 2024-12-20 03:29:51 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

add assign subtitles tests, and fix subtitles service

Browse Source
This commit is contained in:
iPromKnight
2024-02-09 15:32:02 +00:00
committed by iPromKnight
parent 36864e4db1
commit 740b75f106
2 changed files with 124 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

102
src/node/consumer/test/torrent_subtitle_service.test.ts Normal file
View File

@@ -0,0 +1,102 @@
import "reflect-metadata"; // required
import {ITorrentFileCollection} from "@interfaces/torrent_file_collection";
import {TorrentSubtitleService} from "@services/torrent_subtitle_service";
describe('TrrentSubtitleService tests', () => {
let torrentSubtitleService: TorrentSubtitleService;
beforeEach(() => {
jest.clearAllMocks();
torrentSubtitleService = new TorrentSubtitleService();
});
it('should assign subtitles to a single video', () => {
const fileCollection: ITorrentFileCollection = {
videos: [{ title: 'Test video', size: 123456, imdbId: 'tt1234567', infoHash: 'Test infoHash' }],
contents: [],
subtitles: [{ title: 'Test subtitle', fileIndex: 0, path: 'Test path', infoHash: 'Test infoHash' }],
};
const result = torrentSubtitleService.assignSubtitles(fileCollection);
expect(result.videos[0].subtitles).toEqual(fileCollection.subtitles);
expect(result.subtitles).toEqual([]);
});
it('should not assign subtitles if there are no videos', () => {
const fileCollection: ITorrentFileCollection = {
videos: [],
contents: [],
subtitles: [{ title: 'Test subtitle', fileIndex: 0, path: 'Test path', infoHash: 'Test infoHash' }],
};
const result = torrentSubtitleService.assignSubtitles(fileCollection);
expect(result).toEqual(fileCollection);
});
it('should not assign subtitles if there are no subtitles', () => {
const fileCollection: ITorrentFileCollection = {
videos: [{ title: 'Test video', size: 123456, imdbId: 'tt1234567', infoHash: 'Test infoHash' }],
contents: [],
subtitles: [],
};
const result = torrentSubtitleService.assignSubtitles(fileCollection);
expect(result).toEqual(fileCollection);
});
it('should assign subtitles to multiple videos', () => {
const fileCollection: ITorrentFileCollection = {
videos: [
{ title: 'Test video S01E01', size: 123456, imdbId: 'tt1234567', infoHash: 'Test infoHash' },
{ title: 'Test video S01E02', size: 123456, imdbId: 'tt1234567', infoHash: 'Test infoHash' }
],
contents: [],
subtitles: [
{ title: 'Test subtitle S01E01', fileIndex: 0, path: 'Test path', infoHash: 'Test infoHash' },
{ title: 'Test subtitle S01E02', fileIndex: 1, path: 'Test path', infoHash: 'Test infoHash' }
],
};
const result = torrentSubtitleService.assignSubtitles(fileCollection);
expect(result.videos[0].subtitles).toEqual([fileCollection.subtitles[0]]);
expect(result.videos[1].subtitles).toEqual([fileCollection.subtitles[1]]);
expect(result.subtitles).toEqual([]);
});
it('should not assign subtitles if there are no matching videos', () => {
const fileCollection: ITorrentFileCollection = {
videos: [{ title: 'Test video', size: 123456, imdbId: 'tt1234567', infoHash: 'Test infoHash' }],
contents: [],
subtitles: [{ title: 'Non-matching subtitle', fileIndex: 0, path: 'Test path', infoHash: 'Non-matching infoHash' }],
};
const result = torrentSubtitleService.assignSubtitles(fileCollection);
expect(result.videos[0].subtitles).toEqual([]);
expect(result.subtitles).toEqual([fileCollection.subtitles[0]]);
});
it('should assign subtitles to the most probable videos based on filename, title, season, and episode', () => {
const fileCollection: ITorrentFileCollection = {
videos: [
{ title: 'Test video S01E01', size: 123456, imdbId: 'tt1234567', infoHash: 'Test infoHash' },
{ title: 'Test video S01E02', size: 123456, imdbId: 'tt1234567', infoHash: 'Test infoHash' }
],
contents: [],
subtitles: [
{ title: 'Test subtitle S01E01', fileIndex: 0, path: 'Test path', infoHash: 'Test infoHash' },
{ title: 'Test subtitle S01E02', fileIndex: 1, path: 'Test path', infoHash: 'Test infoHash' }
],
};
const result = torrentSubtitleService.assignSubtitles(fileCollection);
expect(result.videos[0].subtitles).toEqual([fileCollection.subtitles[0]]);
expect(result.videos[1].subtitles).toEqual([fileCollection.subtitles[1]]);
expect(result.subtitles).toEqual([]);
});
});