From 6ee49862f6dcd9ea2527c043c4ef9677e7554678 Mon Sep 17 00:00:00 2001
From: freearhey <7253922+freearhey@users.noreply.github.com>
Date: Tue, 4 Feb 2025 04:48:04 +0300
Subject: [PATCH] Update pickx.be.test.js

---
 sites/pickx.be/pickx.be.test.js | 54 +++++++++++++++++++++------------
 1 file changed, 34 insertions(+), 20 deletions(-)

diff --git a/sites/pickx.be/pickx.be.test.js b/sites/pickx.be/pickx.be.test.js
index 05a37ae0..ee11e679 100644
--- a/sites/pickx.be/pickx.be.test.js
+++ b/sites/pickx.be/pickx.be.test.js
@@ -1,35 +1,49 @@
-jest.mock('./pickx.be.config.js', () => {
-  const originalModule = jest.requireActual('./pickx.be.config.js')
-  return {
-    ...originalModule,
-    fetchApiVersion: jest.fn(() => Promise.resolve())
-  }
-})
-
-const { parser, url, request, setApiVersion } = require('./pickx.be.config.js')
-
+const { parser, url, request } = require('./pickx.be.config.js')
+const axios = require('axios')
 const fs = require('fs')
 const path = require('path')
 const dayjs = require('dayjs')
 const utc = require('dayjs/plugin/utc')
-
 dayjs.extend(utc)
 
+jest.mock('axios')
+
+axios.get.mockImplementation((url, data) => {
+  if (url === 'https://www.pickx.be/nl/televisie/tv-gids') {
+    return Promise.resolve({
+      data: fs.readFileSync(path.resolve(__dirname, '__data__/hash.html'), 'utf8')
+    })
+  } else if (
+    url ===
+      'https://www.pickx.be/api/s-375ce5e452cf964b4158545d9ddf26cc97d6411f0998a2fa7ed5922c88d5bdc4' &&
+    JSON.stringify(data) ===
+      JSON.stringify({
+        headers: {
+          Origin: 'https://www.pickx.be',
+          Referer: 'https://www.pickx.be/'
+        }
+      })
+  ) {
+    return Promise.resolve({
+      status: 200,
+      data: JSON.parse(fs.readFileSync(path.resolve(__dirname, '__data__/version.json')))
+    })
+  } else {
+    return Promise.resolve({
+      data: ''
+    })
+  }
+})
+
 const date = dayjs.utc('2023-12-13').startOf('d')
 const channel = {
   lang: 'fr',
-  site_id: 'UID0118',
-  xmltv_id: 'Vedia.be'
+  site_id: 'UID0118'
 }
 
-beforeEach(() => {
-  setApiVersion('mockedApiVersion')
-})
-
 it('can generate valid url', async () => {
-  const generatedUrl = await url({ channel, date })
-  expect(generatedUrl).toBe(
-    'https://px-epg.azureedge.net/airings/mockedApiVersion/2023-12-13/channel/UID0118?timezone=Europe%2FBrussels'
+  expect(await url({ channel, date })).toBe(
+    'https://px-epg.azureedge.net/airings/21738594888692v.4.2/2023-12-13/channel/UID0118?timezone=Europe%2FBrussels'
   )
 })