chore: add gateway keeper dep

pokt-network · Feb 19, 2025 · 49ee23f · 49ee23f
1 parent 10208dc
commit 49ee23f
Show file tree

Hide file tree

Showing 6 changed files with 23 additions and 1 deletion.
diff --git a/testutil/integration/app.go b/testutil/integration/app.go
@@ -526,6 +526,7 @@ func NewCompleteIntegrationApp(t *testing.T, opts ...IntegrationAppOptionFn) *Ap
 		authority.String(),
 		accountKeeper,
 		bankKeeper,
+		gatewayKeeper,
 		applicationKeeper,
 		supplierKeeper,
 	)

diff --git a/testutil/keeper/migration.go b/testutil/keeper/migration.go
@@ -61,6 +61,14 @@ func MigrationKeeper(t testing.TB) (keeper.Keeper, sdk.Context) {
 
 	mockAccountKeeper := mocks.NewMockAccountKeeper(ctrl)
 
+	mockGatewayKeeper := mocks.NewMockGatewayKeeper(ctrl)
+	mockGatewayKeeper.EXPECT().
+		GetGateway(gomock.Any(), gomock.Any()).
+		AnyTimes()
+	mockGatewayKeeper.EXPECT().
+		SetGateway(gomock.Any(), gomock.Any()).
+		AnyTimes()
+
 	mockAppKeeper := mocks.NewMockApplicationKeeper(ctrl)
 	mockAppKeeper.EXPECT().
 		GetApplication(gomock.Any(), gomock.Any()).
@@ -84,6 +92,7 @@ func MigrationKeeper(t testing.TB) (keeper.Keeper, sdk.Context) {
 		authority.String(),
 		mockAccountKeeper,
 		mockBankKeeper,
+		mockGatewayKeeper,
 		mockAppKeeper,
 		mockSupplierKeeper,
 	)

diff --git a/testutil/keeper/tokenomics.go b/testutil/keeper/tokenomics.go
@@ -534,6 +534,7 @@ func NewTokenomicsModuleKeepers(
 		authority.String(),
 		accountKeeper,
 		bankKeeper,
+		gatewayKeeper,
 		appKeeper,
 		supplierKeeper,
 	)

diff --git a/x/migration/keeper/keeper.go b/x/migration/keeper/keeper.go
@@ -23,6 +23,7 @@ type (
 
 		accountKeeper  types.AccountKeeper
 		bankKeeper     types.BankKeeper
+		gatewayKeeper  types.GatewayKeeper
 		appKeeper      types.ApplicationKeeper
 		supplierKeeper types.SupplierKeeper
 	}
@@ -35,6 +36,7 @@ func NewKeeper(
 	authority string,
 	accountKeeper types.AccountKeeper,
 	bankKeeper types.BankKeeper,
+	gatewayKeeper types.GatewayKeeper,
 	appKeeper types.ApplicationKeeper,
 	supplierKeeper types.SupplierKeeper,
 ) Keeper {
@@ -49,6 +51,7 @@ func NewKeeper(
 		logger:         logger,
 		accountKeeper:  accountKeeper,
 		bankKeeper:     bankKeeper,
+		gatewayKeeper:  gatewayKeeper,
 		appKeeper:      appKeeper,
 		supplierKeeper: supplierKeeper,
 	}

diff --git a/x/migration/module/module.go b/x/migration/module/module.go
@@ -188,6 +188,7 @@ type ModuleInputs struct {
 
 	AccountKeeper     types.AccountKeeper
 	BankKeeper        types.BankKeeper
+	GatewayKeeper     types.GatewayKeeper
 	ApplicationKeeper types.ApplicationKeeper
 	SupplierKeeper    types.SupplierKeeper
 }
@@ -212,6 +213,7 @@ func ProvideModule(in ModuleInputs) ModuleOutputs {
 		authority.String(),
 		in.AccountKeeper,
 		in.BankKeeper,
+		in.GatewayKeeper,
 		in.ApplicationKeeper,
 		in.SupplierKeeper,
 	)

diff --git a/x/migration/types/expected_keepers.go b/x/migration/types/expected_keepers.go
@@ -1,4 +1,4 @@
-//go:generate go run go.uber.org/mock/mockgen -destination ../../../testutil/migration/mocks/expected_keepers_mock.go -package mocks . AccountKeeper,BankKeeper,ApplicationKeeper,SupplierKeeper
+//go:generate go run go.uber.org/mock/mockgen -destination ../../../testutil/migration/mocks/expected_keepers_mock.go -package mocks . AccountKeeper,BankKeeper,GatewayKeeper,ApplicationKeeper,SupplierKeeper
 
 package types
 
@@ -8,6 +8,7 @@ import (
 	sdk "github.com/cosmos/cosmos-sdk/types"
 
 	apptypes "github.com/pokt-network/poktroll/x/application/types"
+	"github.com/pokt-network/poktroll/x/gateway/types"
 	sharedtypes "github.com/pokt-network/poktroll/x/shared/types"
 )
 
@@ -25,6 +26,11 @@ type BankKeeper interface {
 	// Methods imported from bank should be defined here
 }
 
+type GatewayKeeper interface {
+	GetGateway(ctx context.Context, address string) (gateway types.Gateway, found bool)
+	SetGateway(ctx context.Context, gateway types.Gateway)
+}
+
 type ApplicationKeeper interface {
 	GetApplication(ctx context.Context, appAddr string) (app apptypes.Application, found bool)
 	SetApplication(ctx context.Context, application apptypes.Application)