@revoloo/cypress6/packages/server/test/e2e/6_video_compression_spec.js

Cypress.io end to end testing tool

// ffprobe is necessary to extract chapters data from mp4 files.
// ffprobe is usually installed with ffmpeg.
// But in our CI, it doesn't. That's why we're installing ffprobe here.
const ffprobePath = require('@ffprobe-installer/ffprobe').path
const ffmpeg = require('fluent-ffmpeg')

ffmpeg.setFfprobePath(ffprobePath)

const path = require('path')
const fs = require('fs-extra')
const humanInterval = require('human-interval')
const e2e = require('../support/helpers/e2e').default
const glob = require('../../lib/util/glob')
const videoCapture = require('../../lib/video_capture')
const Fixtures = require('../support/helpers/fixtures')

const NUM_TESTS = 40
const MS_PER_TEST = 500
const EXPECTED_DURATION_MS = NUM_TESTS * MS_PER_TEST

// ffmpeg command that extracts the final frame as a jpg
function outputFinalFrameAsJpg (inputFile, outputFile) {
  return new Promise((resolve, reject) => {
    return ffmpeg(inputFile)
    .inputOption('-sseof -3')
    .outputOptions(['-vsync 2', '-update 1'])
    .on('end', resolve)
    .on('error', reject)
    .save(outputFile)
  })
}

describe('e2e video compression', () => {
  e2e.setup()

  return [
    true,
    false,
  ].forEach((headed) => {
    e2e.it(`passes (head${headed ? 'ed' : 'less'})`, {
      spec: 'video_compression_spec.js',
      snapshot: false,
      headed,
      config: {
        env: {
          NUM_TESTS,
          MS_PER_TEST,
        },
      },
      onRun (exec) {
        process.env.VIDEO_COMPRESSION_THROTTLE = 10

        return exec()
        .tap(() => {
          const videosPath = Fixtures.projectPath('e2e/cypress/videos/*')

          return glob(videosPath)
          .tap(async (files) => {
            expect(files).to.have.length(1, `globbed for videos and found: ${files.length}. Expected to find 1 video. Search in videosPath: ${videosPath}.`)

            const lastFrameFile = path.join(path.dirname(files[0]), 'lastFrame.jpg')

            await outputFinalFrameAsJpg(files[0], lastFrameFile)
            // https://github.com/cypress-io/cypress/issues/9265
            // if video is seekable and not just one frozen frame, this file should exist
            await fs.stat(lastFrameFile).catch((err) => {
              throw new Error(`Expected video to have seekable ending frame, but it did not. The video may be corrupted.`)
            })

            return videoCapture.getCodecData(files[0])
            .then(({ duration }) => {
              const durationMs = videoCapture.getMsFromDuration(duration)

              expect(durationMs).to.be.ok

              expect(durationMs).to.be.closeTo(EXPECTED_DURATION_MS, humanInterval('15 seconds'))
            })
          })
          .then((files) => {
            return videoCapture.getChapters(files[0])
            .then(({ chapters }) => {
              // There are 40 chapters but we test only the first one
              // because what we want to check is if chapters are added properly.
              // In a chapter object, there are properties like 'end' and 'end_time'.
              // We don't check them here because they return the test time in milliseconds.
              // They cannot be guessed correctly and they can cause flakiness.
              expect(chapters[0].id).to.eq(0)
              expect(chapters[0].start).to.eq(0)
              expect(chapters[0].start_time).to.eq(0)
              expect(chapters[0]['TAG:title']).to.eq('num: 1 makes some long tests')
              expect(chapters[0].time_base).to.eq('1/1000')
              expect(chapters[0].end).to.be.a('number')
              expect(Number.isNaN(chapters[0].end)).to.be.false
              expect(chapters[0].end_time).to.be.a('number')
              expect(Number.isNaN(chapters[0].end_time)).to.be.false
            })
          })
        }).get('stdout')
        .then((stdout) => {
          expect(stdout).to.match(/Compression progress:\s+\d{1,3}%/)
        })
      },
    })
  })
})