server/tests/utils/videos.ts

   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
  10 type VideoAttributes = {
  11   name?: string
  12   category?: number
  13   licence?: number
  14   language?: number
  15   nsfw?: boolean
  16   description?: string
  17   tags?: string[]
  18   fixture?: string
  19 }
  20
  21 function getVideoCategories (url: string) {
  22   const path = '/api/v1/videos/categories'
  23
  24   return makeGetRequest(url, path)
  25 }
  26
  27 function getVideoLicences (url: string) {
  28   const path = '/api/v1/videos/licences'
  29
  30   return makeGetRequest(url, path)
  31 }
  32
  33 function getVideoLanguages (url: string) {
  34   const path = '/api/v1/videos/languages'
  35
  36   return makeGetRequest(url, path)
  37 }
  38
  39 function getAllVideosListBy (url: string) {
  40   const path = '/api/v1/videos'
  41
  42   return request(url)
  43           .get(path)
  44           .query({ sort: 'createdAt' })
  45           .query({ start: 0 })
  46           .query({ count: 10000 })
  47           .set('Accept', 'application/json')
  48           .expect(200)
  49           .expect('Content-Type', /json/)
  50 }
  51
  52 function getVideo (url: string, id: number | string) {
  53   const path = '/api/v1/videos/' + id
  54
  55   return request(url)
  56           .get(path)
  57           .set('Accept', 'application/json')
  58           .expect(200)
  59           .expect('Content-Type', /json/)
  60 }
  61
  62 function getVideosList (url: string) {
  63   const path = '/api/v1/videos'
  64
  65   return request(url)
  66           .get(path)
  67           .query({ sort: 'name' })
  68           .set('Accept', 'application/json')
  69           .expect(200)
  70           .expect('Content-Type', /json/)
  71 }
  72
  73 function getVideosListPagination (url: string, start: number, count: number, sort?: string) {
  74   const path = '/api/v1/videos'
  75
  76   const req = request(url)
  77               .get(path)
  78               .query({ start: start })
  79               .query({ count: count })
  80
  81   if (sort) req.query({ sort })
  82
  83   return req.set('Accept', 'application/json')
  84            .expect(200)
  85            .expect('Content-Type', /json/)
  86 }
  87
  88 function getVideosListSort (url: string, sort: string) {
  89   const path = '/api/v1/videos'
  90
  91   return request(url)
  92           .get(path)
  93           .query({ sort: sort })
  94           .set('Accept', 'application/json')
  95           .expect(200)
  96           .expect('Content-Type', /json/)
  97 }
  98
  99 function removeVideo (url: string, token: string, id: number, expectedStatus = 204) {
 100   const path = '/api/v1/videos'
 101
 102   return request(url)
 103           .delete(path + '/' + id)
 104           .set('Accept', 'application/json')
 105           .set('Authorization', 'Bearer ' + token)
 106           .expect(expectedStatus)
 107 }
 108
 109 function searchVideo (url: string, search: string, field?: string) {
 110   const path = '/api/v1/videos'
 111   const req = request(url)
 112                 .get(path + '/search/' + search)
 113                 .set('Accept', 'application/json')
 114
 115   if (field) req.query({ field })
 116
 117   return req.expect(200)
 118             .expect('Content-Type', /json/)
 119 }
 120
 121 function searchVideoWithPagination (url: string, search: string, field: string, start: number, count: number, sort?: string) {
 122   const path = '/api/v1/videos'
 123
 124   const req = request(url)
 125                 .get(path + '/search/' + search)
 126                 .query({ start })
 127                 .query({ count })
 128                 .query({ field })
 129
 130   if (sort) req.query({ sort })
 131
 132   return req.set('Accept', 'application/json')
 133             .expect(200)
 134             .expect('Content-Type', /json/)
 135 }
 136
 137 function searchVideoWithSort (url: string, search: string, sort: string) {
 138   const path = '/api/v1/videos'
 139
 140   return request(url)
 141           .get(path + '/search/' + search)
 142           .query({ sort })
 143           .set('Accept', 'application/json')
 144           .expect(200)
 145           .expect('Content-Type', /json/)
 146 }
 147
 148 async function testVideoImage (url: string, imageName: string, imagePath: string) {
 149   // Don't test images if the node env is not set
 150   // Because we need a special ffmpeg version for this test
 151   if (process.env['NODE_TEST_IMAGE']) {
 152     const res = await request(url)
 153                         .get(imagePath)
 154                         .expect(200)
 155
 156     const data = await readFilePromise(join(__dirname, '..', 'api', 'fixtures', imageName + '.jpg'))
 157
 158     return data.equals(res.body)
 159   } else {
 160     console.log('Do not test images. Enable it by setting NODE_TEST_IMAGE env variable.')
 161     return true
 162   }
 163 }
 164
 165 function uploadVideo (url: string, accessToken: string, videoAttributesArg: VideoAttributes, specialStatus = 204) {
 166   const path = '/api/v1/videos'
 167
 168   // Default attributes
 169   let attributes = {
 170     name: 'my super video',
 171     category: 5,
 172     licence: 4,
 173     language: 3,
 174     nsfw: true,
 175     description: 'my super description',
 176     tags: [ 'tag' ],
 177     fixture: 'video_short.webm'
 178   }
 179   attributes = Object.assign(attributes, videoAttributesArg)
 180
 181   const req = request(url)
 182               .post(path)
 183               .set('Accept', 'application/json')
 184               .set('Authorization', 'Bearer ' + accessToken)
 185               .field('name', attributes.name)
 186               .field('category', attributes.category.toString())
 187               .field('licence', attributes.licence.toString())
 188               .field('language', attributes.language.toString())
 189               .field('nsfw', JSON.stringify(attributes.nsfw))
 190               .field('description', attributes.description)
 191
 192   for (let i = 0; i < attributes.tags.length; i++) {
 193     req.field('tags[' + i + ']', attributes.tags[i])
 194   }
 195
 196   let filepath = ''
 197   if (isAbsolute(attributes.fixture)) {
 198     filepath = attributes.fixture
 199   } else {
 200     filepath = join(__dirname, '..', 'api', 'fixtures', attributes.fixture)
 201   }
 202
 203   return req.attach('videofile', filepath)
 204             .expect(specialStatus)
 205 }
 206
 207 function updateVideo (url: string, accessToken: string, id: number, attributes: VideoAttributes, specialStatus = 204) {
 208   const path = '/api/v1/videos/' + id
 209   const body = {}
 210
 211   if (attributes.name) body['name'] = attributes.name
 212   if (attributes.category) body['category'] = attributes.category
 213   if (attributes.licence) body['licence'] = attributes.licence
 214   if (attributes.language) body['language'] = attributes.language
 215   if (attributes.nsfw) body['nsfw'] = attributes.nsfw
 216   if (attributes.description) body['description'] = attributes.description
 217   if (attributes.tags) body['tags'] = attributes.tags
 218
 219   return request(url)
 220           .put(path)
 221           .send(body)
 222           .set('Accept', 'application/json')
 223           .set('Authorization', 'Bearer ' + accessToken)
 224           .expect(specialStatus)
 225 }
 226
 227 function rateVideo (url: string, accessToken: string, id: number, rating: string, specialStatus = 204) {
 228   const path = '/api/v1/videos/' + id + '/rate'
 229
 230   return request(url)
 231           .put(path)
 232           .set('Accept', 'application/json')
 233           .set('Authorization', 'Bearer ' + accessToken)
 234           .send({ rating })
 235           .expect(specialStatus)
 236 }
 237
 238 function parseTorrentVideo (server: ServerInfo, videoUUID: string) {
 239   return new Promise<any>((res, rej) => {
 240     const torrentPath = join(__dirname, '..', '..', '..', 'test' + server.serverNumber, 'torrents', videoUUID + '.torrent')
 241     readFile(torrentPath, (err, data) => {
 242       if (err) return rej(err)
 243
 244       return res(parseTorrent(data))
 245     })
 246   })
 247 }
 248
 249 // ---------------------------------------------------------------------------
 250
 251 export {
 252   getVideoCategories,
 253   getVideoLicences,
 254   getVideoLanguages,
 255   getAllVideosListBy,
 256   getVideo,
 257   getVideosList,
 258   getVideosListPagination,
 259   getVideosListSort,
 260   removeVideo,
 261   searchVideo,
 262   searchVideoWithPagination,
 263   searchVideoWithSort,
 264   testVideoImage,
 265   uploadVideo,
 266   updateVideo,
 267   rateVideo,
 268   parseTorrentVideo
 269 }