iptv/tests/commands/playlist/test.test.ts

import child_process, { execSync } from 'node:child_process'
import path from 'node:path'
import { pathToFileURL } from 'node:url'
import { promisify } from 'node:util'
import * as fs from 'fs-extra'
import { glob } from 'glob'

const exec = promisify(child_process.exec)

type ExecError = {
  status: number
  stdout: string
  stderr: string
}

const ENV_VAR = 'cross-env DATA_DIR=tests/__data__/input/data ROOT_DIR=tests/__data__/output'

beforeEach(() => {
  fs.emptyDirSync('tests/__data__/output')
  fs.copySync('tests/__data__/input/playlist_test/streams', 'tests/__data__/output/streams')
})

describe('playlist:test', () => {
  it('shows an error if the playlist contains a broken link', async () => {
    const cmd = `${ENV_VAR} npm run playlist:test streams/ag.m3u`

    try {
      await exec(cmd, { encoding: 'utf8' })
      if (process.env.DEBUG === 'true') console.log(cmd)
      process.exit(0)
    } catch (error) {
      if (process.env.DEBUG === 'true') console.log(cmd, error)
      expect((error as ExecError).stdout).toContain('streams/ag.m3u')
      expect((error as ExecError).stdout).toContain('2 problems (1 errors, 1 warnings)')
    }
  })

  it('it can remove all broken links from the playlist', async () => {
    const cmd = `${ENV_VAR} npm run playlist:test streams/ag.m3u --- --fix`
    try {
      await exec(cmd, { encoding: 'utf8' })
      if (process.env.DEBUG === 'true') console.log(cmd)
      process.exit(0)
    } catch (error) {
      if (process.env.DEBUG === 'true') console.log(cmd, error)
      const files = glob.sync('tests/__data__/expected/playlist_test/*.m3u').map(filepath => {
        const fileUrl = pathToFileURL(filepath).toString()
        const pathToRemove = pathToFileURL('tests/__data__/expected/playlist_test/').toString()

        return fileUrl.replace(pathToRemove, '')
      })

      files.forEach(filepath => {
        expect(content(`tests/__data__/output/streams/${filepath}`)).toBe(
          content(`tests/__data__/expected/playlist_test/${filepath}`)
        )
      })
    }
  })

  it('fails when the network is unavailable', () => {
    const preload = path.resolve('tests/__data__/input/offline_dns.cjs').replaceAll('\\', '/')
    const cmd =
      `cross-env NODE_OPTIONS="--require=${preload}" npm run playlist:test -- streams/af.m3u`

    try {
      execSync(cmd, { encoding: 'utf8' })
      throw new Error('Expected playlist:test to fail offline')
    } catch (error) {
      const output = `${(error as ExecError).stdout || ''}${(error as ExecError).stderr || ''}`
      expect(output).toContain('Internet connection is required for the script to work')
    }
  })
})

function content(filepath: string) {
  return fs.readFileSync(pathToFileURL(filepath), { encoding: 'utf8' })
}