package keeper_test import ( "context" "testing" "github.com/stretchr/testify/suite" "github.com/0glabs/0g-chain/x/committee/testutil" "github.com/0glabs/0g-chain/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)) }