Fixup ErrNoSuchWallet in readIsStakeKeyRegistered

cardano-foundation · Jul 3, 2020 · 97d9d7f · 97d9d7f
1 parent f29112d
commit 97d9d7f
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/lib/core/src/Cardano/Wallet/DB/Sqlite.hs b/lib/core/src/Cardano/Wallet/DB/Sqlite.hs
@@ -626,14 +626,15 @@ newDBLayer trace defaultFieldValues mDatabaseFile = do
                             (StakeKeyCertificate wid sl True)
 
         , readIsStakeKeyRegistered = \(PrimaryKey wid) -> ExceptT $ do
-            val <- fmap entityVal <$> selectFirst
-                    [StakeKeyCertWalletId ==. wid]
-                    [Desc StakeKeyCertSlot]
-
-            return $ case val of
-                Nothing -> Left $ ErrNoSuchWallet wid
-                Just (StakeKeyCertificate _ _ isReg) -> Right isReg
-
+              selectWallet wid >>= \case
+                Nothing -> pure $ Left $ ErrNoSuchWallet wid
+                Just{} -> do
+                    val <- fmap entityVal <$> selectFirst
+                        [StakeKeyCertWalletId ==. wid]
+                        [Desc StakeKeyCertSlot]
+                    return $ case val of
+                        Nothing -> Right False
+                        Just (StakeKeyCertificate _ _ isReg) -> Right isReg
 
         {-----------------------------------------------------------------------
                                      Tx History