From 6ecb9211dc8ebc3f89aefe993f2613d000f08bbc Mon Sep 17 00:00:00 2001
From: Nazar Duchak <nazar@iovlabs.org>
Date: Tue, 8 Sep 2020 13:25:26 +0300
Subject: [PATCH] feat: fix linter

---
 src/services/storage/handlers/stake.ts  | 18 +++++++++++++-----
 test/services/storage/processor.spec.ts |  4 ++--
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/src/services/storage/handlers/stake.ts b/src/services/storage/handlers/stake.ts
index f8d47f01..1ac6345f 100644
--- 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
index a6057c38..e919e00e 100644
--- 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))
       })
     })