fixup! test(e2e): consistent currentEpoch/nextEpoch/nextNextEpoch del…

…egatee assertions
input-output-hk · Jan 20, 2025 · 5eb960e · 5eb960e
1 parent b8d51f8
commit 5eb960e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/packages/e2e/test/wallet_epoch_3/PersonalWallet/delegation.test.ts b/packages/e2e/test/wallet_epoch_3/PersonalWallet/delegation.test.ts
@@ -33,7 +33,10 @@ const getWalletStateSnapshot = async (wallet: ObservableWallet) => {
   return {
     balance: { available: balanceAvailable, deposit, total: balanceTotal },
     epoch: epoch.epochNo,
-    isStakeKeyRegistered: rewardAccount.credentialStatus === Cardano.StakeCredentialStatus.Registered,
+    isStakeKeyRegistered: [
+      Cardano.StakeCredentialStatus.Registered,
+      Cardano.StakeCredentialStatus.Registering
+    ].includes(rewardAccount.credentialStatus),
     publicStakeKey,
     rewardAccount,
     rewardsBalance,

diff --git a/packages/e2e/test/wallet_epoch_3/SharedWallet/delegation.test.ts b/packages/e2e/test/wallet_epoch_3/SharedWallet/delegation.test.ts
@@ -32,7 +32,10 @@ const getWalletStateSnapshot = async (wallet: ObservableWallet) => {
   return {
     balance: { available: balanceAvailable, deposit, total: balanceTotal },
     epoch: epoch.epochNo,
-    isStakeCredentialRegistered: rewardAccount.credentialStatus === Cardano.StakeCredentialStatus.Registered,
+    isStakeCredentialRegistered: [
+      Cardano.StakeCredentialStatus.Registered,
+      Cardano.StakeCredentialStatus.Registering
+    ].includes(rewardAccount.credentialStatus),
     publicStakeKey,
     rewardAccount,
     rewardsBalance,