-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
index.test.js
105 lines (89 loc) · 2.83 KB
/
index.test.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import { expect, test, vi } from 'vitest'
import episodic from './index.js'
vi.mock('node:fs/promises', () => ({
mkdir: vi.fn(),
readdir: vi.fn(() =>
[1, 2, 3].map(i => `The.Foo.Bar.S01E0${i}.720p.HDTV.x264-WTF.mkv`),
),
rename: vi.fn(),
}))
vi.mock('imdb-api', () => ({
get: vi.fn(({ name }) =>
Promise.resolve({
name: name.replace(/(^| )(\w)/g, s => s.toUpperCase()),
episodes: vi.fn(() =>
Promise.resolve(
[1, 2, 3].map(episode => ({
episode,
name: `Episode #${episode}`,
season: 1,
})),
),
),
imdburl: 'https://www.some-imdb-url.com/',
start_year: 2018,
}),
),
}))
vi.mock('inquirer', () => ({
default: {
prompt: vi.fn(() => ({ rename: true })),
},
}))
vi.mock('ora', () => ({
default: () => ({
start: vi.fn(() => ({
stop: vi.fn(),
})),
}),
}))
vi.spyOn(console, 'log').mockImplementation(() => {})
vi.spyOn(console, 'error').mockImplementation(() => {})
vi.spyOn(process, 'exit').mockImplementation(() => {})
test('should error out if no API key is provided', async () => {
await episodic('path/to/files')
expect(console.error).toHaveBeenCalled()
expect(process.exit).toHaveBeenCalledWith(1)
})
test('should rename files in-place', async () => {
await episodic('path/to/files', { apiKey: 'omdb-api-key' })
expect(console.log).toHaveBeenCalledTimes(1)
expect(console.log.mock.calls[0][0]).toMatchInlineSnapshot(`
"
The Foo Bar (2018)
https://www.some-imdb-url.com/
The.Foo.Bar.S01E01.720p.HDTV.x264-WTF.mkv -> 1. Episode #1.mkv
The.Foo.Bar.S01E02.720p.HDTV.x264-WTF.mkv -> 2. Episode #2.mkv
The.Foo.Bar.S01E03.720p.HDTV.x264-WTF.mkv -> 3. Episode #3.mkv
"
`)
})
test('should move files to an output directory', async () => {
await episodic('path/to/files', {
apiKey: 'omdb-api-key',
output: 'path/to/dest',
})
expect(console.log).toHaveBeenCalledTimes(1)
expect(console.log.mock.calls[0][0]).toMatchInlineSnapshot(`
"
The Foo Bar (2018)
https://www.some-imdb-url.com/
The.Foo.Bar.S01E01.720p.HDTV.x264-WTF.mkv -> ../dest/1. Episode #1.mkv
The.Foo.Bar.S01E02.720p.HDTV.x264-WTF.mkv -> ../dest/2. Episode #2.mkv
The.Foo.Bar.S01E03.720p.HDTV.x264-WTF.mkv -> ../dest/3. Episode #3.mkv
"
`)
})
test('should organize files into show name/season directories', async () => {
await episodic('path/to/files', { apiKey: 'omdb-api-key', tree: true })
expect(console.log).toHaveBeenCalledTimes(1)
expect(console.log.mock.calls[0][0]).toMatchInlineSnapshot(`
"
The Foo Bar (2018)
https://www.some-imdb-url.com/
The.Foo.Bar.S01E01.720p.HDTV.x264-WTF.mkv -> The Foo Bar (2018)/Season 1/1. Episode #1.mkv
The.Foo.Bar.S01E02.720p.HDTV.x264-WTF.mkv -> The Foo Bar (2018)/Season 1/2. Episode #2.mkv
The.Foo.Bar.S01E03.720p.HDTV.x264-WTF.mkv -> The Foo Bar (2018)/Season 1/3. Episode #3.mkv
"
`)
})