From 642b09051f3c261f8b105ac3a4b04d0c8b60d6df Mon Sep 17 00:00:00 2001
From: Aleksandr Statciuk <github@oiya.xyz>
Date: Fri, 21 Jan 2022 19:25:50 +0300
Subject: [PATCH] Update load-cluster.js

---
 scripts/commands/load-cluster.js    |  4 ++--
 tests/commands/load-cluster.test.js | 21 +++++++++++++++------
 2 files changed, 17 insertions(+), 8 deletions(-)

diff --git a/scripts/commands/load-cluster.js b/scripts/commands/load-cluster.js
index 4b077d68..b859afa5 100644
--- a/scripts/commands/load-cluster.js
+++ b/scripts/commands/load-cluster.js
@@ -56,9 +56,9 @@ async function main() {
       if (err) logger.error(err.message)
 
       const result = {
-        _id: channel._id,
-        logo: data.channel.logo,
+        channel: data.channel,
         programs: data.programs,
+        date: data.date.format(),
         error: err ? err.message : null
       }
 
diff --git a/tests/commands/load-cluster.test.js b/tests/commands/load-cluster.test.js
index c2c768b1..fd0576b7 100644
--- a/tests/commands/load-cluster.test.js
+++ b/tests/commands/load-cluster.test.js
@@ -1,7 +1,11 @@
 const fs = require('fs')
 const path = require('path')
+const dayjs = require('dayjs')
+const utc = require('dayjs/plugin/utc')
 const { execSync } = require('child_process')
 
+dayjs.extend(utc)
+
 beforeEach(() => {
   fs.rmdirSync('tests/__data__/temp', { recursive: true })
   fs.rmdirSync('tests/__data__/output', { recursive: true })
@@ -21,17 +25,22 @@ beforeEach(() => {
 it('can load cluster', () => {
   const output = content('tests/__data__/output/logs/load-cluster/cluster_1.log')
 
+  expect(Object.keys(output[0]).sort()).toEqual(['channel', 'date', 'error', 'programs'])
+
   expect(output[0]).toMatchObject({
-    _id: '0Wefq0oMR3feCcuY',
-    logo: 'https://example.com/logo.png',
+    channel: {
+      _id: '0Wefq0oMR3feCcuY',
+      logo: 'https://example.com/logo.png'
+    },
+    date: dayjs.utc().startOf('d').format(),
     error: null
   })
 
-  expect(Object.keys(output[0]).sort()).toEqual(['_id', 'error', 'logo', 'programs'])
-
   expect(output[1]).toMatchObject({
-    _id: '1XzrxNkSF2AQNBrT',
-    logo: 'https://www.magticom.ge/images/channels/MjAxOC8wOS8xMC9lZmJhNWU5Yy0yMmNiLTRkMTAtOWY5Ny01ODM0MzY0ZTg0MmEuanBn.jpg'
+    channel: {
+      _id: '1XzrxNkSF2AQNBrT',
+      logo: 'https://www.magticom.ge/images/channels/MjAxOC8wOS8xMC9lZmJhNWU5Yy0yMmNiLTRkMTAtOWY5Ny01ODM0MzY0ZTg0MmEuanBn.jpg'
+    }
   })
 })