1 files changed, 27 insertions, 0 deletions

diff --git a/server/tests/utils/miscs/miscs.ts b/server/tests/utils/miscs/miscs.ts
index e6666619b..f93e2372e 100644
--- a/server/tests/utils/miscs/miscs.ts
+++ b/server/tests/utils/miscs/miscs.ts
@@ -1,5 +1,10 @@
+/* tslint:disable:no-unused-expression */
+
+import { expect } from 'chai'
 import { join } from 'path'
+import * as request from 'supertest'
 import * as WebTorrent from 'webtorrent'
+import { readFileBufferPromise } from '../../../helpers/core-utils'
 
 let webtorrent = new WebTorrent()
 
@@ -30,6 +35,27 @@ function root () {
   return join(__dirname, '..', '..', '..', '..')
 }
 
+async function testImage (url: string, imageName: string, imagePath: string, extension = '.jpg') {
+  // Don't test images if the node env is not set
+  // Because we need a special ffmpeg version for this test
+  if (process.env[ 'NODE_TEST_IMAGE' ]) {
+    const res = await request(url)
+      .get(imagePath)
+      .expect(200)
+
+    const body = res.body
+
+    const data = await readFileBufferPromise(join(__dirname, '..', '..', 'api', 'fixtures', imageName + extension))
+    const minLength = body.length - ((50 * body.length) / 100)
+    const maxLength = body.length + ((50 * body.length) / 100)
+
+    return data.length > minLength && data.length < maxLength
+  } else {
+    console.log('Do not test images. Enable it by setting NODE_TEST_IMAGE env variable.')
+    return true
+  }
+}
+
 // ---------------------------------------------------------------------------
 
 export {
@@ -37,5 +63,6 @@ export {
   wait,
   webtorrentAdd,
   immutableAssign,
+  testImage,
   root
 }