mirror of
				https://github.com/0glabs/0g-chain.git
				synced 2025-10-31 21:18:59 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			43 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			43 lines
		
	
	
		
			1.3 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package testutil
 | |
| 
 | |
| import (
 | |
| 	tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
 | |
| 	sdk "github.com/cosmos/cosmos-sdk/types"
 | |
| 	bankkeeper "github.com/cosmos/cosmos-sdk/x/bank/keeper"
 | |
| 	"github.com/stretchr/testify/suite"
 | |
| 
 | |
| 	"github.com/0glabs/0g-chain/app"
 | |
| 	"github.com/0glabs/0g-chain/chaincfg"
 | |
| 	"github.com/0glabs/0g-chain/x/committee/keeper"
 | |
| 	"github.com/0glabs/0g-chain/x/committee/types"
 | |
| )
 | |
| 
 | |
| // Suite implements a test suite for the module integration tests
 | |
| type Suite struct {
 | |
| 	suite.Suite
 | |
| 
 | |
| 	Keeper      keeper.Keeper
 | |
| 	BankKeeper  bankkeeper.Keeper
 | |
| 	App         app.TestApp
 | |
| 	Ctx         sdk.Context
 | |
| 	QueryClient types.QueryClient
 | |
| 	Addresses   []sdk.AccAddress
 | |
| }
 | |
| 
 | |
| // SetupTest instantiates a new app, keepers, and sets suite state
 | |
| func (suite *Suite) SetupTest() {
 | |
| 	chaincfg.SetSDKConfig()
 | |
| 	suite.App = app.NewTestApp()
 | |
| 	suite.Keeper = suite.App.GetCommitteeKeeper()
 | |
| 	suite.BankKeeper = suite.App.GetBankKeeper()
 | |
| 	suite.Ctx = suite.App.NewContext(true, tmproto.Header{})
 | |
| 	_, accAddresses := app.GeneratePrivKeyAddressPairs(10)
 | |
| 	suite.Addresses = accAddresses
 | |
| 
 | |
| 	// Set query client
 | |
| 	queryHelper := suite.App.NewQueryServerTestHelper(suite.Ctx)
 | |
| 	queryHandler := keeper.NewQueryServerImpl(suite.Keeper)
 | |
| 	types.RegisterQueryServer(queryHelper, queryHandler)
 | |
| 	suite.QueryClient = types.NewQueryClient(queryHelper)
 | |
| }
 | 
