mirror of
https://github.com/iptv-org/epg.git
synced 2025-05-10 17:10:07 -04:00
98 lines
3.9 KiB
TypeScript
98 lines
3.9 KiB
TypeScript
import { execSync } from 'child_process'
|
|
import fs from 'fs-extra'
|
|
import path from 'path'
|
|
import os from 'os'
|
|
import { Zip } from '@freearhey/core'
|
|
|
|
let ENV_VAR =
|
|
'SITES_DIR=tests/__data__/input/epg-grab/sites CURR_DATE=2022-10-20 DATA_DIR=tests/__data__/input/temp/data'
|
|
if (os.platform() === 'win32') {
|
|
ENV_VAR =
|
|
'SET "SITES_DIR=tests/__data__/input/epg-grab/sites" && SET "CURR_DATE=2022-10-20" && SET "DATA_DIR=tests/__data__/input/temp/data" &&'
|
|
}
|
|
|
|
beforeEach(() => {
|
|
fs.emptyDirSync('tests/__data__/output')
|
|
})
|
|
|
|
describe('epg:grab', () => {
|
|
it('can grab epg by site name', () => {
|
|
const cmd = `${ENV_VAR} npm run grab --- --site=example.com --output=${path.resolve(
|
|
'tests/__data__/output/guide.xml'
|
|
)}`
|
|
execSync(cmd, { encoding: 'utf8' })
|
|
|
|
expect(content('tests/__data__/output/guide.xml')).toEqual(
|
|
content('tests/__data__/expected/guide2.xml')
|
|
)
|
|
})
|
|
|
|
it('can grab epg with multiple channels.xml files', () => {
|
|
const cmd = `${ENV_VAR} npm run grab --- --channels=tests/__data__/input/epg-grab/sites/**/*.channels.xml --output=tests/__data__/output/guide.xml`
|
|
execSync(cmd, { encoding: 'utf8' })
|
|
|
|
expect(content('tests/__data__/output/guide.xml')).toEqual(
|
|
content('tests/__data__/expected/guide.xml')
|
|
)
|
|
})
|
|
|
|
it('can grab epg with gzip option enabled', async () => {
|
|
const cmd = `${ENV_VAR} npm run grab --- --channels=tests/__data__/input/epg-grab/sites/**/*.channels.xml --output=${path.resolve(
|
|
'tests/__data__/output/guide.xml'
|
|
)} --gzip`
|
|
execSync(cmd, { encoding: 'utf8' })
|
|
|
|
expect(content('tests/__data__/output/guide.xml')).toEqual(
|
|
content('tests/__data__/expected/guide.xml')
|
|
)
|
|
|
|
const zip = new Zip()
|
|
const expected = await zip.decompress(fs.readFileSync('tests/__data__/output/guide.xml.gz'))
|
|
const result = await zip.decompress(fs.readFileSync('tests/__data__/expected/guide.xml.gz'))
|
|
expect(expected).toEqual(result)
|
|
})
|
|
|
|
it('can grab epg with wildcard as output', () => {
|
|
const cmd = `${ENV_VAR} npm run grab --- --channels=tests/__data__/input/epg-grab/sites/example.com/example.com.channels.xml --output=tests/__data__/output/guides/{lang}/{site}.xml`
|
|
execSync(cmd, { encoding: 'utf8' })
|
|
|
|
expect(content('tests/__data__/output/guides/en/example.com.xml')).toEqual(
|
|
content('tests/__data__/expected/guides/en/example.com.xml')
|
|
)
|
|
|
|
expect(content('tests/__data__/output/guides/fr/example.com.xml')).toEqual(
|
|
content('tests/__data__/expected/guides/fr/example.com.xml')
|
|
)
|
|
})
|
|
|
|
it('can grab epg then language filter enabled', () => {
|
|
const cmd = `${ENV_VAR} npm run grab --- --channels=tests/__data__/input/epg-grab/sites/example.com/example.com.channels.xml --output=tests/__data__/output/guides/{lang}/{site}.xml --lang=fr`
|
|
execSync(cmd, { encoding: 'utf8' })
|
|
|
|
expect(content('tests/__data__/output/guides/fr/example.com.xml')).toEqual(
|
|
content('tests/__data__/expected/guides/fr/example.com.xml')
|
|
)
|
|
})
|
|
|
|
it('can grab epg using custom channels list', () => {
|
|
const cmd = `${ENV_VAR} npm run grab --- --channels=tests/__data__/input/epg-grab/custom.channels.xml --output=tests/__data__/output/guide.xml`
|
|
execSync(cmd, { encoding: 'utf8' })
|
|
|
|
expect(content('tests/__data__/output/guide.xml')).toEqual(
|
|
content('tests/__data__/expected/guide.xml')
|
|
)
|
|
})
|
|
|
|
it('it will raise an error if the timeout is exceeded', () => {
|
|
const cmd = `${ENV_VAR} npm run grab --- --channels=tests/__data__/input/epg-grab/custom.channels.xml --output=tests/__data__/output/guide.xml --timeout=0`
|
|
const stdout = execSync(cmd, { encoding: 'utf8' })
|
|
|
|
expect(stdout).toContain('ERR: Connection timeout')
|
|
})
|
|
})
|
|
|
|
function content(filepath: string) {
|
|
return fs.readFileSync(path.resolve(filepath), {
|
|
encoding: 'utf8'
|
|
})
|
|
}
|