Setting existing tests

santiment · Jan 24, 2024 · 40cc0d1 · 40cc0d1
1 parent 7d2c3ed
commit 40cc0d1
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 5 deletions.
diff --git a/lib/worker_base.js b/lib/worker_base.js
@@ -20,7 +20,7 @@ class WorkerBase {
    * Upon returning from the method call the implementation should have updated all the member variables of the
    * base class.
    */
-  work() {
+  async work() {
     throw new Error('"work" method need to be overriden');
   }
   // To be implemented on inheritance.

diff --git a/test/eth/worker.spec.js b/test/eth/worker.spec.js
@@ -54,9 +54,9 @@ describe('Test worker', function () {
       return [feeResult, callResult];
     };
 
-    const result = await worker.work();
+    const result = await worker.work(0);
 
-    assert.deepStrictEqual(result, [feeResultWithPrimaryKey, callResultWithPrimaryKey]);
+    assert.deepStrictEqual(result, [0, [feeResultWithPrimaryKey, callResultWithPrimaryKey]]);
   });
 });
 

diff --git a/test/index.test.js b/test/index.test.js
@@ -165,7 +165,7 @@ describe('Main', () => {
 
     await mainInstance.initWorker();
     assert(mainInstance.handleInitPosition.calledOnce);
-    assert.strictEqual(mainInstance.taskManager.lastPushedToQueue, 0);
+    assert.strictEqual(mainInstance.taskManager.lastPushedToBuffer, 0);
   });
 
   it('workLoop throws error when worker can\'t be initialised', async () => {
@@ -188,7 +188,8 @@ describe('Main', () => {
     const mainInstance = new Main();
     mainInstance.worker = new BaseWorker(constants);
     mainInstance.exporter = new Exporter('test-exporter');
-    mainInstance.taskManager = new TaskManager(0, 50);
+    mainInstance.taskManager = new TaskManager();
+    mainInstance.taskManager.buffer = [ 1 ]; // So that we get to see that storeEvents fails
     await mainInstance.taskManager.initQueue(1);
     try {
       await mainInstance.workLoop();
@@ -208,6 +209,7 @@ describe('Main', () => {
     mainInstance.worker = new BaseWorker(constants);
     mainInstance.exporter = new Exporter('test-exporter');
     mainInstance.taskManager = new TaskManager(0, 50);
+    mainInstance.taskManager.buffer = [ 1 ]; // So that we get to see that savePosition fails
     await mainInstance.taskManager.initQueue(1);
     try {
       await mainInstance.workLoop();