Add remaining bond module keeper tests

tests/integration/bond/keeper/query_server_test.go

@@ -67,6 +67,133 @@ func (kts *KeeperTestSuite) TestGrpcQueryBondsList() {
 	}
 }
 
+func (kts *KeeperTestSuite) TestGrpcQueryBondByBondId() {
+	testCases := []struct {
+		msg         string
+		req         *types.QueryGetBondByIdRequest
+		createBonds bool
+		errResponse bool
+		bondId      string
+	}{
+		{
+			"empty request",
+			&types.QueryGetBondByIdRequest{},
+			false,
+			true,
+			"",
+		},
+		{
+			"Get Bond By ID",
+			&types.QueryGetBondByIdRequest{},
+			true,
+			false,
+			"",
+		},
+	}
+
+	for _, test := range testCases {
+		kts.Run(fmt.Sprintf("Case %s ", test.msg), func() {
+			if test.createBonds {
+				bond, err := kts.createBond()
+				kts.Require().NoError(err)
+				test.req.Id = bond.Id
+			}
+			resp, err := kts.queryClient.GetBondById(context.Background(), test.req)
+			if !test.errResponse {
+				kts.Require().Nil(err)
+				kts.Require().NotNil(resp.GetBond())
+				kts.Require().Equal(test.req.Id, resp.GetBond().GetId())
+			} else {
+				kts.Require().NotNil(err)
+				kts.Require().Error(err)
+			}
+		})
+	}
+}
+
+func (kts *KeeperTestSuite) TestGrpcGetBondsByOwner() {
+	testCases := []struct {
+		msg         string
+		req         *types.QueryGetBondsByOwnerRequest
+		noOfBonds   int
+		createBonds bool
+		errResponse bool
+		bondId      string
+	}{
+		{
+			"empty request",
+			&types.QueryGetBondsByOwnerRequest{},
+			0,
+			false,
+			true,
+			"",
+		},
+		{
+			"Get Bond By Owner",
+			&types.QueryGetBondsByOwnerRequest{},
+			1,
+			true,
+			false,
+			"",
+		},
+	}
+
+	for _, test := range testCases {
+		kts.Run(fmt.Sprintf("Case %s ", test.msg), func() {
+			if test.createBonds {
+				bond, err := kts.createBond()
+				kts.Require().NoError(err)
+				test.req.Owner = bond.Owner
+			}
+			resp, err := kts.queryClient.GetBondsByOwner(context.Background(), test.req)
+			if !test.errResponse {
+				kts.Require().Nil(err)
+				kts.Require().NotNil(resp.GetBonds())
+				kts.Require().Equal(test.noOfBonds, len(resp.GetBonds()))
+			} else {
+				kts.Require().NotNil(err)
+				kts.Require().Error(err)
+			}
+		})
+	}
+}
+
+func (kts *KeeperTestSuite) TestGrpcGetModuleBalance() {
+	testCases := []struct {
+		msg         string
+		req         *types.QueryGetBondModuleBalanceRequest
+		noOfBonds   int
+		createBonds bool
+		errResponse bool
+	}{
+		{
+			"empty request",
+			&types.QueryGetBondModuleBalanceRequest{},
+			0,
+			true,
+			false,
+		},
+	}
+
+	for _, test := range testCases {
+		kts.Run(fmt.Sprintf("Case %s ", test.msg), func() {
+			if test.createBonds {
+				_, err := kts.createBond()
+				kts.Require().NoError(err)
+			}
+			resp, err := kts.queryClient.GetBondsModuleBalance(context.Background(), test.req)
+			if !test.errResponse {
+				kts.Require().Nil(err)
+				kts.Require().NotNil(resp.GetBalance())
+				kts.Require().Equal(resp.GetBalance(), sdk.NewCoins(sdk.NewCoin(sdk.DefaultBondDenom, math.NewInt(10))))
+			} else {
+				kts.Require().NotNil(err)
+				kts.Require().Error(err)
+			}
+		})
+	}
+}
+
 func (kts *KeeperTestSuite) createBond() (*types.Bond, error) {
 	ctx, k := kts.SdkCtx, kts.BondKeeper
 	accCount := 1

tests/integration/common.go

@@ -98,7 +98,7 @@ func (tf *TestFixture) Setup() error {
 
 	auctionKeeper := auctionkeeper.NewKeeper(cdc, runtime.NewKVStoreService(keys[auctionTypes.StoreKey]), accountKeeper, bankKeeper)
 
-	bondKeeper := bondkeeper.NewKeeper(cdc, runtime.NewKVStoreService(keys[auctionTypes.StoreKey]), accountKeeper, bankKeeper)
+	bondKeeper := bondkeeper.NewKeeper(cdc, runtime.NewKVStoreService(keys[bondTypes.StoreKey]), accountKeeper, bankKeeper)
 
 	authModule := auth.NewAppModule(cdc, accountKeeper, authsims.RandomGenesisAccounts, nil)
 	bankModule := bank.NewAppModule(cdc, bankKeeper, accountKeeper, nil)