Skip to content
GitLab
Switch branch/tag
Find file BlameHistoryPermalink
query_test.go 4.60 KiB
1 
package cli_test
2
3 
import (
4 
	gocontext "context"
5 
	"testing"
6 
	"time"
7
8 
	"github.com/stretchr/testify/suite"
9
10 
	"github.com/osmosis-labs/osmosis/v12/app/apptesting"
11 
	"github.com/osmosis-labs/osmosis/v12/x/lockup/types"
12
13 
	sdk "github.com/cosmos/cosmos-sdk/types"
14 
)
15
16 
type QueryTestSuite struct {
17 
	apptesting.KeeperTestHelper
18 
	queryClient types.QueryClient
19 
}
20
21 
func (s *QueryTestSuite) SetupSuite() {
22 
	s.Setup()
23 
	s.queryClient = types.NewQueryClient(s.QueryHelper)
24
25 
	// create a pool
26 
	s.PrepareBalancerPool()
27 
	// set up lock with id = 1
28 
	s.LockTokens(s.TestAccs[0], sdk.Coins{sdk.NewCoin("gamm/pool/1", sdk.NewInt(1000000))}, time.Hour*24)
29
30 
	s.Commit()
31 
}
32
33 
func (s *QueryTestSuite) TestQueriesNeverAlterState() {
34 
	testCases := []struct {
35 
		name   string
36 
		query  string
37 
		input  interface{}
38 
		output interface{}
39 
	}{
40 
		{
41 
			"Query account locked coins",
42 
			"/osmosis.lockup.Query/AccountLockedCoins",
43 
			&types.AccountLockedCoinsRequest{Owner: s.TestAccs[0].String()},
44 
			&types.AccountLockedCoinsResponse{},
45 
		},
46 
		{
47 
			"Query account locked by duration",
48 
			"/osmosis.lockup.Query/AccountLockedDuration",
49 
			&types.AccountLockedDurationRequest{Owner: s.TestAccs[0].String(), Duration: time.Hour * 24},
50 
			&types.AccountLockedDurationResponse{},
51 
		},
52 
		{
53 
			"Query account locked longer than given duration",
54 
			"/osmosis.lockup.Query/AccountLockedLongerDuration",
55 
			&types.AccountLockedLongerDurationRequest{Owner: s.TestAccs[0].String(), Duration: time.Hour},
56 
			&types.AccountLockedLongerDurationResponse{},
57 
		},
58 
		{
59 
			"Query account locked by denom that longer than given duration",
60 
			"/osmosis.lockup.Query/AccountLockedLongerDurationDenom",
61 
			&types.AccountLockedLongerDurationDenomRequest{Owner: s.TestAccs[0].String(), Duration: time.Hour, Denom: "gamm/pool/1"},
62 
			&types.AccountLockedLongerDurationDenomResponse{},
63 
		},
64 
		{
65 
			"Query account locked longer than given duration not unlocking",
66 
			"/osmosis.lockup.Query/AccountLockedLongerDurationNotUnlockingOnly",
67 
			&types.AccountLockedLongerDurationNotUnlockingOnlyRequest{Owner: s.TestAccs[0].String(), Duration: time.Hour},
68 
			&types.AccountLockedLongerDurationNotUnlockingOnlyResponse{},
69 
		},
70 
		{
7172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140
"Query account locked in past time", "/osmosis.lockup.Query/AccountLockedPastTime", &types.AccountLockedPastTimeRequest{Owner: s.TestAccs[0].String()}, &types.AccountLockedPastTimeResponse{}, }, { "Query account locked in past time by denom", "/osmosis.lockup.Query/AccountLockedPastTimeDenom", &types.AccountLockedPastTimeDenomRequest{Owner: s.TestAccs[0].String(), Denom: "gamm/pool/1"}, &types.AccountLockedPastTimeDenomResponse{}, }, { " Query account locked in past time that not unlocking", "/osmosis.lockup.Query/AccountLockedPastTimeNotUnlockingOnly", &types.AccountLockedPastTimeNotUnlockingOnlyRequest{Owner: s.TestAccs[0].String()}, &types.AccountLockedPastTimeNotUnlockingOnlyResponse{}, }, { "Query account unlockable coins", "/osmosis.lockup.Query/AccountUnlockableCoins", &types.AccountUnlockableCoinsRequest{Owner: s.TestAccs[0].String()}, &types.AccountUnlockableCoinsResponse{}, }, { "Query account unlocked before given time", "/osmosis.lockup.Query/AccountUnlockedBeforeTime", &types.AccountUnlockedBeforeTimeRequest{Owner: s.TestAccs[0].String()}, &types.AccountUnlockedBeforeTimeResponse{}, }, { "Query account unlocking coins", "/osmosis.lockup.Query/AccountUnlockingCoins", &types.AccountUnlockingCoinsRequest{Owner: s.TestAccs[0].String()}, &types.AccountUnlockingCoinsResponse{}, }, { "Query lock by id", "/osmosis.lockup.Query/LockedByID", &types.LockedRequest{LockId: 1}, &types.LockedResponse{}, }, { "Query lock by denom", "/osmosis.lockup.Query/LockedDenom", &types.LockedDenomRequest{Duration: time.Hour * 24, Denom: "gamm/pool/1"}, &types.LockedDenomResponse{}, }, { "Query module balances", "/osmosis.lockup.Query/ModuleBalance", &types.ModuleBalanceRequest{}, &types.ModuleBalanceResponse{}, }, { "Query module locked amount", "/osmosis.lockup.Query/ModuleLockedAmount", &types.ModuleLockedAmountRequest{}, &types.ModuleLockedAmountResponse{}, }, { "Query synthetic lock by id", "/osmosis.lockup.Query/SyntheticLockupsByLockupID", &types.SyntheticLockupsByLockupIDRequest{LockId: 1}, &types.SyntheticLockupsByLockupIDResponse{}, }, } for _, tc := range testCases { tc := tc
141142143144145146147148149150151152153
s.Run(tc.name, func() { s.SetupSuite() err := s.QueryHelper.Invoke(gocontext.Background(), tc.query, tc.input, tc.output) s.Require().NoError(err) s.StateNotAltered() }) } } func TestQueryTestSuite(t *testing.T) { suite.Run(t, new(QueryTestSuite)) }