Injectable locked coins functionality

CHANGELOG.md

@@ -39,7 +39,9 @@ Ref: https://keepachangelog.com/en/1.0.0/
 
 ## Unreleased
 
-* nothing
+### Features
+
+* [#577](https://github.com/provenance-io/cosmos-sdk/pull/577) Add an injectable `GetLockedCoinsFn` to the bank module.
 
 ---
 

simapp/sim_test.go

@@ -254,16 +254,16 @@ func TestAppSimulationAfterImport(t *testing.T) {
 		PrintStats(db)
 	}
 
-	if stopEarly {
-		fmt.Println("can't export or import a zero-validator genesis, exiting test...")
-		return
-	}
-
 	fmt.Printf("exporting genesis...\n")
 
 	exported, err := app.ExportAppStateAndValidators(true, []string{})
 	require.NoError(t, err)
 
+	if stopEarly || len(exported.Validators) == 0 {
+		fmt.Println("can't import a zero-validator genesis, exiting test...")
+		return
+	}
+
 	fmt.Printf("importing genesis...\n")
 
 	_, newDB, newDir, _, _, err := SetupSimulation("leveldb-app-sim-2", "Simulation-2")

x/bank/keeper/export_test.go

@@ -5,10 +5,25 @@ import "github.com/cosmos/cosmos-sdk/x/bank/types"
 // This file exists in the keeper package to expose some private things
 // for the purpose of testing in the keeper_test package.
 
+// SetSendRestriction is a TEST ONLY method for overwriting the SendRestrictionFn.
 func (k BaseSendKeeper) SetSendRestriction(restriction types.SendRestrictionFn) {
 	k.sendRestriction.fn = restriction
 }
 
+// GetSendRestrictionFn is a TEST ONLY exposure of the currently defined SendRestrictionFn.
 func (k BaseSendKeeper) GetSendRestrictionFn() types.SendRestrictionFn {
 	return k.sendRestriction.fn
 }
+
+// SetLockedCoinsGetter is a TEST ONLY method for overwriting the GetLockedCoinsFn.
+func (k BaseSendKeeper) SetLockedCoinsGetter(getter types.GetLockedCoinsFn) {
+	k.lockedCoinsGetter.fn = getter
+}
+
+// GetLockedCoinsGetter is a TEST ONLY exposure of the currently defined GetLockedCoinsFn.
+func (k BaseViewKeeper) GetLockedCoinsGetter() types.GetLockedCoinsFn {
+	return k.lockedCoinsGetter.fn
+}
+
+// GetLockedCoinsFnWrapper is a TEST ONLY exposure of the getLockedCoinsFnWrapper function.
+var GetLockedCoinsFnWrapper = getLockedCoinsFnWrapper

x/bank/keeper/keeper.go

@@ -161,6 +161,10 @@ func (k BaseKeeper) DelegateCoins(ctx sdk.Context, delegatorAddr, moduleAccAddr
 		}
 	}
 
+	if _, err := k.sendRestriction.apply(ctx, delegatorAddr, moduleAccAddr, amt); err != nil {
+		return err
+	}
+
 	if err := k.trackDelegation(ctx, delegatorAddr, balances, amt); err != nil {
 		return sdkerrors.Wrap(err, "failed to track delegation")
 	}