package keeper_test

import (
	"github.com/0glabs/0g-chain/x/incentive/keeper"
	"github.com/0glabs/0g-chain/x/incentive/types"
	sdk "github.com/cosmos/cosmos-sdk/types"
)

// TestKeeper is a test wrapper for the keeper which contains useful methods for testing
type TestKeeper struct {
	keeper.Keeper
}

func (keeper TestKeeper) storeGlobalBorrowIndexes(ctx sdk.Context, indexes types.MultiRewardIndexes) {
	for _, i := range indexes {
		keeper.SetHardBorrowRewardIndexes(ctx, i.CollateralType, i.RewardIndexes)
	}
}

func (keeper TestKeeper) storeGlobalSupplyIndexes(ctx sdk.Context, indexes types.MultiRewardIndexes) {
	for _, i := range indexes {
		keeper.SetHardSupplyRewardIndexes(ctx, i.CollateralType, i.RewardIndexes)
	}
}

func (keeper TestKeeper) storeGlobalDelegatorIndexes(ctx sdk.Context, multiRewardIndexes types.MultiRewardIndexes) {
	// Hardcoded to use bond denom
	multiRewardIndex, _ := multiRewardIndexes.GetRewardIndex(types.BondDenom)
	keeper.SetDelegatorRewardIndexes(ctx, types.BondDenom, multiRewardIndex.RewardIndexes)
}

func (keeper TestKeeper) storeGlobalSwapIndexes(ctx sdk.Context, indexes types.MultiRewardIndexes) {
	for _, i := range indexes {
		keeper.SetSwapRewardIndexes(ctx, i.CollateralType, i.RewardIndexes)
	}
}

func (keeper TestKeeper) storeGlobalSavingsIndexes(ctx sdk.Context, indexes types.MultiRewardIndexes) {
	for _, i := range indexes {
		keeper.SetSavingsRewardIndexes(ctx, i.CollateralType, i.RewardIndexes)
	}
}

func (keeper TestKeeper) storeGlobalEarnIndexes(ctx sdk.Context, indexes types.MultiRewardIndexes) {
	for _, i := range indexes {
		keeper.SetEarnRewardIndexes(ctx, i.CollateralType, i.RewardIndexes)
	}
}