mirror of
				https://github.com/0glabs/0g-chain.git
				synced 2025-11-04 11:17:28 +00:00 
			
		
		
		
	* Append votes to queryResults, remove unused GetVotesByProposal * Add Votes grpc_query test
		
			
				
	
	
		
			54 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			54 lines
		
	
	
		
			1.4 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
package keeper_test
 | 
						|
 | 
						|
import (
 | 
						|
	"context"
 | 
						|
	"testing"
 | 
						|
 | 
						|
	"github.com/stretchr/testify/suite"
 | 
						|
 | 
						|
	"github.com/kava-labs/kava/x/committee/testutil"
 | 
						|
	"github.com/kava-labs/kava/x/committee/types"
 | 
						|
)
 | 
						|
 | 
						|
type grpcQueryTestSuite struct {
 | 
						|
	testutil.Suite
 | 
						|
}
 | 
						|
 | 
						|
func (suite *grpcQueryTestSuite) SetupTest() {
 | 
						|
	suite.Suite.SetupTest()
 | 
						|
}
 | 
						|
 | 
						|
func (suite *grpcQueryTestSuite) TestVote() {
 | 
						|
	ctx, keeper, queryClient := suite.Ctx, suite.Keeper, suite.QueryClient
 | 
						|
	vote := types.Vote{
 | 
						|
		ProposalID: 1,
 | 
						|
		Voter:      suite.Addresses[0],
 | 
						|
		VoteType:   types.VOTE_TYPE_ABSTAIN,
 | 
						|
	}
 | 
						|
	keeper.SetVote(ctx, vote)
 | 
						|
 | 
						|
	req := types.QueryVoteRequest{
 | 
						|
		ProposalId: vote.ProposalID,
 | 
						|
		Voter:      vote.Voter.String(),
 | 
						|
	}
 | 
						|
	res, err := queryClient.Vote(context.Background(), &req)
 | 
						|
	suite.Require().NoError(err)
 | 
						|
	suite.Require().Equal(vote.ProposalID, res.ProposalID)
 | 
						|
	suite.Require().Equal(vote.VoteType, res.VoteType)
 | 
						|
	suite.Require().Equal(vote.Voter.String(), res.Voter)
 | 
						|
 | 
						|
	queryRes, err := queryClient.Votes(context.Background(), &types.QueryVotesRequest{
 | 
						|
		ProposalId: vote.ProposalID,
 | 
						|
	})
 | 
						|
 | 
						|
	suite.Require().NoError(err)
 | 
						|
	suite.Require().Len(queryRes.Votes, 1)
 | 
						|
	suite.Require().Equal(vote.ProposalID, queryRes.Votes[0].ProposalID)
 | 
						|
	suite.Require().Equal(vote.VoteType, queryRes.Votes[0].VoteType)
 | 
						|
	suite.Require().Equal(vote.Voter.String(), queryRes.Votes[0].Voter)
 | 
						|
}
 | 
						|
 | 
						|
func TestGrpcQueryTestSuite(t *testing.T) {
 | 
						|
	suite.Run(t, new(grpcQueryTestSuite))
 | 
						|
}
 |