From 1888ae4386fd812b610c9a3bfa572152085bb9ac Mon Sep 17 00:00:00 2001
From: Simon Warta <simon@warta.it>
Date: Thu, 7 Jul 2022 08:39:53 +0200
Subject: [PATCH] Improve validatorDelegations testing

---
 .../stargate/src/modules/staking/queries.spec.ts    | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/packages/stargate/src/modules/staking/queries.spec.ts b/packages/stargate/src/modules/staking/queries.spec.ts
index 3c4a6268..b8ce620b 100644
--- a/packages/stargate/src/modules/staking/queries.spec.ts
+++ b/packages/stargate/src/modules/staking/queries.spec.ts
@@ -223,8 +223,17 @@ describe("StakingExtension", () => {
       const [client, tmClient] = await makeClientWithStaking(simapp.tendermintUrl);
 
       const response = await client.staking.validatorDelegations(validator.validatorAddress);
-      expect(response.delegationResponses).toBeDefined();
-      expect(response.delegationResponses).not.toBeNull();
+      expect(response.delegationResponses.length).toBeGreaterThanOrEqual(2);
+      // Find the self-delegation
+      expect(response.delegationResponses).toContain({
+        delegation: {
+          delegatorAddress: validator.delegatorAddress,
+          validatorAddress: validator.validatorAddress,
+          shares: "3000000000000000000000000",
+        },
+        balance: { denom: "ustake", amount: "3000000" },
+      });
+      expect(response.pagination?.total.toNumber()).toBeGreaterThanOrEqual(2);
 
       tmClient.disconnect();
     });