Refactoring index test

test/index.test.js

@@ -8,6 +8,7 @@ const BaseWorker = require('../lib/worker_base');
 const { Exporter } = require('../lib/kafka_storage');
 const { worker } = require('../blockchains/eth/eth_worker');
 const zkClientAsync = require('../lib/zookeeper_client_async');
+const TaskManager = require('../lib/task_manager');
 
 describe('Main', () => {
   const constants = {
@@ -159,15 +160,20 @@ describe('Main', () => {
     mainInstance.exporter = new Exporter('test-exporter');
     sinon.stub(worker.prototype, 'init').resolves();
     sinon.stub(mainInstance, 'handleInitPosition').resolves();
+    sinon.stub(TaskManager.prototype, 'initQueue').resolves();
+    mainInstance.lastProcessedPosition = { blockNumber: 10, primaryKey: 1 };
 
     await mainInstance.initWorker();
     assert(mainInstance.handleInitPosition.calledOnce);
+    assert.strictEqual(mainInstance.taskManager.lastPushedToQueue, 0);
   });
 
   it('workLoop throws error when worker can\'t be initialised', async () => {
     sinon.stub(BaseWorker.prototype, 'work').rejects(new Error('Error in worker "work" method'));
     const mainInstance = new Main();
     mainInstance.worker = new BaseWorker(constants);
+    mainInstance.taskManager = new TaskManager(0, 50);
+    await mainInstance.taskManager.initQueue(1);
     try {
       await mainInstance.workLoop();
       expect.fail('workLoop should have thrown an error');