epg-grabber 0.33.0 → 0.34.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "epg-grabber",
-  "version": "0.33.0",
+  "version": "0.34.0",
   "description": "Node.js CLI tool for grabbing EPG from different sites",
   "main": "src/index.js",
   "preferGlobal": true,

package/src/index.d.ts

@@ -101,10 +101,19 @@ export type GrabCallbackData = {
 }
 
 export declare class EPGGrabber {
-  constructor(config: object)
+  constructor(config: SiteConfig)
   grab(
     channel: Channel,
     date: string | dayjs.Dayjs,
-    cb: (data: GrabCallbackData, err: Error) => void
+    cb: (data: GrabCallbackData, err: Error | null) => void
+  ): Promise<Program[]>
+}
+
+export declare class EPGGrabberMock {
+  constructor(config: SiteConfig)
+  grab(
+    channel: Channel,
+    date: string | dayjs.Dayjs,
+    cb: (data: GrabCallbackData, err: Error | null) => void
   ): Promise<Program[]>
 }

package/src/index.js

@@ -53,4 +53,26 @@ class EPGGrabber {
   }
 }
 
+class EPGGrabberMock {
+  constructor(config) {
+    this.config = config
+  }
+
+  async grab(channel, date, cb) {
+    let _date = getUTCDate(date)
+    let _programs = await this.config.parser({ channel, date: _date })
+    let programs = _programs.map(data => new Program(data, channel))
+
+    if (this.config.request?.timeout !== undefined && this.config.request.timeout < 1) {
+      cb({ programs: [], date: _date, channel }, new Error('Connection timeout'))
+      return []
+    }
+
+    cb({ programs, date: _date, channel }, null)
+
+    return programs
+  }
+}
+
 module.exports.EPGGrabber = EPGGrabber
+module.exports.EPGGrabberMock = EPGGrabberMock

package/tests/__data__/output/guide.xml

@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="UTF-8" ?><tv date="20230927">
+<?xml version="1.0" encoding="UTF-8" ?><tv date="20230930">
 <channel id="1TV.com"><display-name>1 TV</display-name><icon src="https://example.com/logos/1TV.png"/><url>https://example.com</url></channel>
 <channel id="2TV.com"><display-name>2 TV</display-name><icon src="http://example.com/logos/1TV.png?x=шеллы&amp;sid=777"/><url>https://example.com</url></channel>
 <channel id="3TV.com"><display-name>3 TV</display-name><icon src="http://example.com/logos/1TV.png?x=шеллы&amp;sid=777"/><url>https://example2.com</url></channel>

package/tests/index.test.js

@@ -2,7 +2,7 @@
  * @jest-environment node
  */
 
-import { EPGGrabber, Channel } from '../src/index'
+import { EPGGrabber, EPGGrabberMock, Channel } from '../src/index'
 import axios from 'axios'
 
 jest.mock('axios')
@@ -66,3 +66,58 @@ it('can grab single channel programs', done => {
       done()
     })
 })
+
+it('can mock epg grabber', done => {
+  const config = {
+    site: 'example.com',
+    url: 'http://example.com/20210319/1tv.json',
+    parser: ({ channel, date }) => [
+      { title: `Test (${channel.name})`, start: '2021-03-19T04:30:00.000Z' }
+    ]
+  }
+  const channel = new Channel({
+    site: 'example.com',
+    site_id: '1',
+    xmltv_id: '1TV.fr',
+    lang: 'fr',
+    name: '1TV'
+  })
+  const grabber = new EPGGrabberMock(config)
+  grabber
+    .grab(channel, '2022-01-01', (data, err) => {
+      if (err) {
+        done()
+      }
+    })
+    .then(programs => {
+      expect(programs.length).toBe(1)
+      expect(programs).toEqual([
+        {
+          site: 'example.com',
+          channel: '1TV.fr',
+          titles: [{ value: 'Test (1TV)', lang: 'fr' }],
+          sub_titles: [],
+          descriptions: [],
+          icon: { src: '' },
+          episodeNumbers: [],
+          date: null,
+          start: 1616128200000,
+          stop: null,
+          urls: [],
+          ratings: [],
+          categories: [],
+          directors: [],
+          actors: [],
+          writers: [],
+          adapters: [],
+          producers: [],
+          composers: [],
+          editors: [],
+          presenters: [],
+          commentators: [],
+          guests: []
+        }
+      ])
+      done()
+    })
+})