From c034f001e8ffb7d348b273fd0dede753c14b2f8b Mon Sep 17 00:00:00 2001
From: Aleksandr Statciuk <github@oiya.xyz>
Date: Sun, 16 Jan 2022 15:12:38 +0300
Subject: [PATCH] Update load-cluster.js

---
 scripts/commands/load-cluster.js    | 31 ++++++++++++++---------------
 tests/commands/load-cluster.test.js | 14 ++-----------
 2 files changed, 17 insertions(+), 28 deletions(-)

diff --git a/scripts/commands/load-cluster.js b/scripts/commands/load-cluster.js
index 4592c441..c55e8441 100644
--- a/scripts/commands/load-cluster.js
+++ b/scripts/commands/load-cluster.js
@@ -46,30 +46,29 @@ async function main() {
       }
     })
 
-    const programs = await grabber.grab(channel, config, async (data, err) => {
-      await db.channels.update({ _id: channel._id }, { $set: { logo: data.channel.logo } })
+    await grabber.grab(channel, config, async (data, err) => {
+      await file.append(
+        clusterLog,
+        JSON.stringify({
+          _id: channel._id,
+          site: channel.site,
+          country: channel.country,
+          logo: data.channel.logo,
+          gid: channel.gid,
+          programs: data.programs
+        }) + '\n'
+      )
 
       logger.info(
-        `[${i}/${total}] ${channel.site} - ${data.id} - ${data.date.format('MMM D, YYYY')} (${
-          data.programs.length
-        } programs)`
+        `[${i}/${total}] ${channel.site} - ${channel.xmltv_id} - ${data.date.format(
+          'MMM D, YYYY'
+        )} (${data.programs.length} programs)`
       )
 
       if (err) logger.error(err.message)
 
       if (i < total) i++
     })
-
-    await file.append(
-      clusterLog,
-      JSON.stringify({
-        _id: channel._id,
-        site: channel.site,
-        country: channel.country,
-        gid: channel.gid,
-        programs
-      }) + '\n'
-    )
   }
 
   db.channels.compact()
diff --git a/tests/commands/load-cluster.test.js b/tests/commands/load-cluster.test.js
index 6f4db1d7..e603ed0c 100644
--- a/tests/commands/load-cluster.test.js
+++ b/tests/commands/load-cluster.test.js
@@ -24,6 +24,7 @@ it('can load cluster', () => {
   expect(output[0]).toMatchObject({
     _id: '0Wefq0oMR3feCcuY',
     site: 'chaines-tv.orange.fr',
+    logo: 'https://example.com/logo.png',
     country: 'US',
     gid: 'fr'
   })
@@ -31,21 +32,10 @@ it('can load cluster', () => {
   expect(output[1]).toMatchObject({
     _id: '1XzrxNkSF2AQNBrT',
     site: 'magticom.ge',
+    logo: 'https://www.magticom.ge/images/channels/MjAxOC8wOS8xMC9lZmJhNWU5Yy0yMmNiLTRkMTAtOWY5Ny01ODM0MzY0ZTg0MmEuanBn.jpg',
     country: 'US',
     gid: 'ge'
   })
-
-  const database = content('tests/__data__/temp/database/channels.db')
-
-  expect(database[1]).toMatchObject({
-    _id: '0Wefq0oMR3feCcuY',
-    logo: 'https://example.com/logo.png'
-  })
-
-  expect(database[2]).toMatchObject({
-    _id: '1XzrxNkSF2AQNBrT',
-    logo: 'https://www.magticom.ge/images/channels/MjAxOC8wOS8xMC9lZmJhNWU5Yy0yMmNiLTRkMTAtOWY5Ny01ODM0MzY0ZTg0MmEuanBn.jpg'
-  })
 })
 
 function content(filepath) {