]>
Commit | Line | Data |
---|---|---|
1 | import { readFile } from 'fs' | |
2 | import * as request from 'supertest' | |
3 | import { join, isAbsolute } from 'path' | |
4 | import * as parseTorrent from 'parse-torrent' | |
5 | ||
6 | import { makeGetRequest } from './requests' | |
7 | import { readFilePromise } from './miscs' | |
8 | import { ServerInfo } from './servers' | |
9 | import { getMyUserInformation } from './users' | |
10 | import { VideoPrivacy } from '../../../shared' | |
11 | ||
12 | type VideoAttributes = { | |
13 | name?: string | |
14 | category?: number | |
15 | licence?: number | |
16 | language?: number | |
17 | nsfw?: boolean | |
18 | description?: string | |
19 | tags?: string[] | |
20 | channelId?: number | |
21 | privacy?: VideoPrivacy | |
22 | fixture?: string | |
23 | } | |
24 | ||
25 | function getVideoCategories (url: string) { | |
26 | const path = '/api/v1/videos/categories' | |
27 | ||
28 | return makeGetRequest(url, path) | |
29 | } | |
30 | ||
31 | function getVideoLicences (url: string) { | |
32 | const path = '/api/v1/videos/licences' | |
33 | ||
34 | return makeGetRequest(url, path) | |
35 | } | |
36 | ||
37 | function getVideoLanguages (url: string) { | |
38 | const path = '/api/v1/videos/languages' | |
39 | ||
40 | return makeGetRequest(url, path) | |
41 | } | |
42 | ||
43 | function getVideoPrivacies (url: string) { | |
44 | const path = '/api/v1/videos/privacies' | |
45 | ||
46 | return makeGetRequest(url, path) | |
47 | } | |
48 | ||
49 | function getVideo (url: string, id: number | string, expectedStatus = 200) { | |
50 | const path = '/api/v1/videos/' + id | |
51 | ||
52 | return request(url) | |
53 | .get(path) | |
54 | .set('Accept', 'application/json') | |
55 | .expect(expectedStatus) | |
56 | } | |
57 | ||
58 | function viewVideo (url: string, id: number | string, expectedStatus = 204) { | |
59 | const path = '/api/v1/videos/' + id + '/views' | |
60 | ||
61 | return request(url) | |
62 | .post(path) | |
63 | .set('Accept', 'application/json') | |
64 | .expect(expectedStatus) | |
65 | } | |
66 | ||
67 | function getVideoWithToken (url: string, token: string, id: number | string, expectedStatus = 200) { | |
68 | const path = '/api/v1/videos/' + id | |
69 | ||
70 | return request(url) | |
71 | .get(path) | |
72 | .set('Authorization', 'Bearer ' + token) | |
73 | .set('Accept', 'application/json') | |
74 | .expect(expectedStatus) | |
75 | } | |
76 | ||
77 | function getVideoDescription (url: string, descriptionPath: string) { | |
78 | return request(url) | |
79 | .get(descriptionPath) | |
80 | .set('Accept', 'application/json') | |
81 | .expect(200) | |
82 | .expect('Content-Type', /json/) | |
83 | } | |
84 | ||
85 | function getVideosList (url: string) { | |
86 | const path = '/api/v1/videos' | |
87 | ||
88 | return request(url) | |
89 | .get(path) | |
90 | .query({ sort: 'name' }) | |
91 | .set('Accept', 'application/json') | |
92 | .expect(200) | |
93 | .expect('Content-Type', /json/) | |
94 | } | |
95 | ||
96 | function getMyVideos (url: string, accessToken: string, start: number, count: number, sort?: string) { | |
97 | const path = '/api/v1/users/me/videos' | |
98 | ||
99 | const req = request(url) | |
100 | .get(path) | |
101 | .query({ start: start }) | |
102 | .query({ count: count }) | |
103 | ||
104 | if (sort) req.query({ sort }) | |
105 | ||
106 | return req.set('Accept', 'application/json') | |
107 | .set('Authorization', 'Bearer ' + accessToken) | |
108 | .expect(200) | |
109 | .expect('Content-Type', /json/) | |
110 | } | |
111 | ||
112 | function getVideosListPagination (url: string, start: number, count: number, sort?: string) { | |
113 | const path = '/api/v1/videos' | |
114 | ||
115 | const req = request(url) | |
116 | .get(path) | |
117 | .query({ start: start }) | |
118 | .query({ count: count }) | |
119 | ||
120 | if (sort) req.query({ sort }) | |
121 | ||
122 | return req.set('Accept', 'application/json') | |
123 | .expect(200) | |
124 | .expect('Content-Type', /json/) | |
125 | } | |
126 | ||
127 | function getVideosListSort (url: string, sort: string) { | |
128 | const path = '/api/v1/videos' | |
129 | ||
130 | return request(url) | |
131 | .get(path) | |
132 | .query({ sort: sort }) | |
133 | .set('Accept', 'application/json') | |
134 | .expect(200) | |
135 | .expect('Content-Type', /json/) | |
136 | } | |
137 | ||
138 | function removeVideo (url: string, token: string, id: number, expectedStatus = 204) { | |
139 | const path = '/api/v1/videos' | |
140 | ||
141 | return request(url) | |
142 | .delete(path + '/' + id) | |
143 | .set('Accept', 'application/json') | |
144 | .set('Authorization', 'Bearer ' + token) | |
145 | .expect(expectedStatus) | |
146 | } | |
147 | ||
148 | function searchVideo (url: string, search: string) { | |
149 | const path = '/api/v1/videos' | |
150 | const req = request(url) | |
151 | .get(path + '/search') | |
152 | .query({ search }) | |
153 | .set('Accept', 'application/json') | |
154 | ||
155 | return req.expect(200) | |
156 | .expect('Content-Type', /json/) | |
157 | } | |
158 | ||
159 | function searchVideoWithPagination (url: string, search: string, start: number, count: number, sort?: string) { | |
160 | const path = '/api/v1/videos' | |
161 | ||
162 | const req = request(url) | |
163 | .get(path + '/search') | |
164 | .query({ start }) | |
165 | .query({ search }) | |
166 | .query({ count }) | |
167 | ||
168 | if (sort) req.query({ sort }) | |
169 | ||
170 | return req.set('Accept', 'application/json') | |
171 | .expect(200) | |
172 | .expect('Content-Type', /json/) | |
173 | } | |
174 | ||
175 | function searchVideoWithSort (url: string, search: string, sort: string) { | |
176 | const path = '/api/v1/videos' | |
177 | ||
178 | return request(url) | |
179 | .get(path + '/search') | |
180 | .query({ search }) | |
181 | .query({ sort }) | |
182 | .set('Accept', 'application/json') | |
183 | .expect(200) | |
184 | .expect('Content-Type', /json/) | |
185 | } | |
186 | ||
187 | async function testVideoImage (url: string, imageName: string, imagePath: string) { | |
188 | // Don't test images if the node env is not set | |
189 | // Because we need a special ffmpeg version for this test | |
190 | if (process.env['NODE_TEST_IMAGE']) { | |
191 | const res = await request(url) | |
192 | .get(imagePath) | |
193 | .expect(200) | |
194 | ||
195 | const data = await readFilePromise(join(__dirname, '..', 'api', 'fixtures', imageName + '.jpg')) | |
196 | ||
197 | return data.equals(res.body) | |
198 | } else { | |
199 | console.log('Do not test images. Enable it by setting NODE_TEST_IMAGE env variable.') | |
200 | return true | |
201 | } | |
202 | } | |
203 | ||
204 | async function uploadVideo (url: string, accessToken: string, videoAttributesArg: VideoAttributes, specialStatus = 201) { | |
205 | const path = '/api/v1/videos/upload' | |
206 | let defaultChannelId = '1' | |
207 | ||
208 | try { | |
209 | const res = await getMyUserInformation(url, accessToken) | |
210 | defaultChannelId = res.body.videoChannels[0].id | |
211 | } catch (e) { /* empty */ } | |
212 | ||
213 | // Default attributes | |
214 | let attributes = { | |
215 | name: 'my super video', | |
216 | category: 5, | |
217 | licence: 4, | |
218 | language: 3, | |
219 | channelId: defaultChannelId, | |
220 | nsfw: true, | |
221 | description: 'my super description', | |
222 | tags: [ 'tag' ], | |
223 | privacy: VideoPrivacy.PUBLIC, | |
224 | fixture: 'video_short.webm' | |
225 | } | |
226 | attributes = Object.assign(attributes, videoAttributesArg) | |
227 | ||
228 | const req = request(url) | |
229 | .post(path) | |
230 | .set('Accept', 'application/json') | |
231 | .set('Authorization', 'Bearer ' + accessToken) | |
232 | .field('name', attributes.name) | |
233 | .field('category', attributes.category.toString()) | |
234 | .field('licence', attributes.licence.toString()) | |
235 | .field('nsfw', JSON.stringify(attributes.nsfw)) | |
236 | .field('description', attributes.description) | |
237 | .field('privacy', attributes.privacy.toString()) | |
238 | .field('channelId', attributes.channelId) | |
239 | ||
240 | if (attributes.language !== undefined) { | |
241 | req.field('language', attributes.language.toString()) | |
242 | } | |
243 | ||
244 | for (let i = 0; i < attributes.tags.length; i++) { | |
245 | req.field('tags[' + i + ']', attributes.tags[i]) | |
246 | } | |
247 | ||
248 | let filePath = '' | |
249 | if (isAbsolute(attributes.fixture)) { | |
250 | filePath = attributes.fixture | |
251 | } else { | |
252 | filePath = join(__dirname, '..', 'api', 'fixtures', attributes.fixture) | |
253 | } | |
254 | ||
255 | return req.attach('videofile', filePath) | |
256 | .expect(specialStatus) | |
257 | } | |
258 | ||
259 | function updateVideo (url: string, accessToken: string, id: number, attributes: VideoAttributes, specialStatus = 204) { | |
260 | const path = '/api/v1/videos/' + id | |
261 | const body = {} | |
262 | ||
263 | if (attributes.name) body['name'] = attributes.name | |
264 | if (attributes.category) body['category'] = attributes.category | |
265 | if (attributes.licence) body['licence'] = attributes.licence | |
266 | if (attributes.language) body['language'] = attributes.language | |
267 | if (attributes.nsfw) body['nsfw'] = attributes.nsfw | |
268 | if (attributes.description) body['description'] = attributes.description | |
269 | if (attributes.tags) body['tags'] = attributes.tags | |
270 | if (attributes.privacy) body['privacy'] = attributes.privacy | |
271 | ||
272 | return request(url) | |
273 | .put(path) | |
274 | .send(body) | |
275 | .set('Accept', 'application/json') | |
276 | .set('Authorization', 'Bearer ' + accessToken) | |
277 | .expect(specialStatus) | |
278 | } | |
279 | ||
280 | function rateVideo (url: string, accessToken: string, id: number, rating: string, specialStatus = 204) { | |
281 | const path = '/api/v1/videos/' + id + '/rate' | |
282 | ||
283 | return request(url) | |
284 | .put(path) | |
285 | .set('Accept', 'application/json') | |
286 | .set('Authorization', 'Bearer ' + accessToken) | |
287 | .send({ rating }) | |
288 | .expect(specialStatus) | |
289 | } | |
290 | ||
291 | function parseTorrentVideo (server: ServerInfo, videoUUID: string, resolution: number) { | |
292 | return new Promise<any>((res, rej) => { | |
293 | const torrentName = videoUUID + '-' + resolution + '.torrent' | |
294 | const torrentPath = join(__dirname, '..', '..', '..', 'test' + server.serverNumber, 'torrents', torrentName) | |
295 | readFile(torrentPath, (err, data) => { | |
296 | if (err) return rej(err) | |
297 | ||
298 | return res(parseTorrent(data)) | |
299 | }) | |
300 | }) | |
301 | } | |
302 | ||
303 | // --------------------------------------------------------------------------- | |
304 | ||
305 | export { | |
306 | getVideoDescription, | |
307 | getVideoCategories, | |
308 | getVideoLicences, | |
309 | getVideoPrivacies, | |
310 | getVideoLanguages, | |
311 | getMyVideos, | |
312 | getVideo, | |
313 | getVideoWithToken, | |
314 | getVideosList, | |
315 | getVideosListPagination, | |
316 | getVideosListSort, | |
317 | removeVideo, | |
318 | searchVideo, | |
319 | searchVideoWithPagination, | |
320 | searchVideoWithSort, | |
321 | testVideoImage, | |
322 | uploadVideo, | |
323 | updateVideo, | |
324 | rateVideo, | |
325 | viewVideo, | |
326 | parseTorrentVideo | |
327 | } |