feat: fix linter

rsksmart · Sep 8, 2020 · 85c3124 · 85c3124
1 parent 7954f35
commit 85c3124
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 7 deletions.
diff --git a/src/services/storage/handlers/stake.ts b/src/services/storage/handlers/stake.ts
@@ -13,12 +13,16 @@ const handlers = {
 
     const stake = await StakeModel.findOne({ where: { token, account } })
 
-    stake!.total = total
-    await stake!.save()
+    if (!stake) {
+      throw new Error(`Stake for acoount ${account}, token ${token} not created`)
+    }
+
+    stake.total = total
+    await stake.save()
     logger.info(`Account ${account} stake amount ${amount}, final balance ${total}`)
 
     if (stakeService.emit) {
-      stakeService.emit('updated', stake!.toJSON())
+      stakeService.emit('updated', stake.toJSON())
     }
   },
 
@@ -27,8 +31,12 @@ const handlers = {
 
     const stake = await StakeModel.findOne({ where: { token, account } })
 
-    stake!.total = total
-    await stake!.save()
+    if (!stake) {
+      throw new Error(`Stake for acoount ${account}, token ${token} not created`)
+    }
+
+    stake.total = total
+    await stake.save()
     logger.info(`Account ${account} stake amount ${amount}, final balance ${total}`)
 
     if (stakeService.emit) {

diff --git a/test/services/storage/processor.spec.ts b/test/services/storage/processor.spec.ts
@@ -458,7 +458,7 @@ describe('Storage services: Events Processor', () => {
         const updatedStake = await StakeModel.findOne({ where: { token, account } })
 
         expect(updatedStake?.total).to.be.eql(new BigNumber(event.returnValues.total))
-        expect(stakeServiceEmitSpy).to.have.been.calledWith('updated', updatedStake!.toJSON())
+        expect(stakeServiceEmitSpy).to.have.been.calledWith('updated', updatedStake?.toJSON())
       })
     })
     describe('Unstaked', () => {
@@ -476,7 +476,7 @@ describe('Storage services: Events Processor', () => {
         await processor(event)
         const updatedStake = await StakeModel.findOne({ where: { token, account } })
 
-        expect(stakeServiceEmitSpy).to.have.been.calledWith('updated', updatedStake!.toJSON())
+        expect(stakeServiceEmitSpy).to.have.been.calledWith('updated', updatedStake?.toJSON())
         expect(updatedStake?.total).to.be.eql(new BigNumber(event.returnValues.total))
       })
     })