Added missing keeper dependencies

sentinel-official · Jul 27, 2023 · 7d07576 · 7d07576
1 parent e7f524b
commit 7d07576
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/x/node/keeper/keeper.go b/x/node/keeper/keeper.go
@@ -38,6 +38,10 @@ func (k *Keeper) WithProviderKeeper(keeper expected.ProviderKeeper) {
 	k.provider = keeper
 }
 
+func (k *Keeper) WithSubscriptionKeeper(keeper expected.SubscriptionKeeper) {
+	k.subscription = keeper
+}
+
 func (k *Keeper) Logger(ctx sdk.Context) log.Logger {
 	return ctx.Logger().With("module", "x/"+types.ModuleName)
 }

diff --git a/x/vpn/keeper/keeper.go b/x/vpn/keeper/keeper.go
@@ -51,10 +51,13 @@ func NewKeeper(
 	)
 	k.Node.WithDistributionKeeper(distributionKeeper)
 	k.Node.WithProviderKeeper(&k.Provider)
+	k.Node.WithSubscriptionKeeper(&k.Subscription)
 
 	k.Plan = plankeeper.NewKeeper(cdc, key)
+	k.Plan.WithBankKeeper(bankKeeper)
 	k.Plan.WithProviderKeeper(&k.Provider)
 	k.Plan.WithNodeKeeper(&k.Node)
+	k.Plan.WithSubscriptionKeeper(&k.Subscription)
 
 	k.Subscription = subscriptionkeeper.NewKeeper(
 		cdc, key, paramsKeeper.Subspace(fmt.Sprintf("%s/%s", types.ModuleName, subscriptiontypes.ModuleName)),