Address code review comments

coinbase · Nov 26, 2024 · b28b12b · b28b12b
1 parent 9da8cc5
commit b28b12b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/tests/e2e.ts b/src/tests/e2e.ts
@@ -183,7 +183,8 @@ describe("Coinbase SDK Stake E2E Test", () => {
 
       expect(rewards).toBeDefined();
       // TODO(rohit): Fix code. This should ideally return 20.
-      expect(rewards.length).toEqual(19);
+      expect(rewards.length).toBeGreaterThanOrEqual(19);
+      expect(rewards.length).toBeLessThanOrEqual(20);
     });
 
     it("should list shared eth staking rewards via ExternalAddress", async () => {
@@ -220,7 +221,8 @@ describe("Coinbase SDK Stake E2E Test", () => {
 
       expect(stakingBalances).toBeDefined();
       // TODO(rohit): Fix code. This should ideally return 20.
-      expect(stakingBalances.length).toEqual(19);
+      expect(stakingBalances.length).toBeGreaterThanOrEqual(19);
+      expect(stakingBalances.length).toBeLessThanOrEqual(20);
     });
   });