refactor querier test

x/slashing/keeper/querier_test.go

@@ -1,38 +1,53 @@
-package keeper
+package keeper_test
 
 import (
 	"testing"
 
 	"github.com/stretchr/testify/require"
+
 	abci "github.com/tendermint/tendermint/abci/types"
 
 	"github.com/cosmos/cosmos-sdk/codec"
+	"github.com/cosmos/cosmos-sdk/simapp"
+	"github.com/cosmos/cosmos-sdk/x/slashing/keeper"
 	"github.com/cosmos/cosmos-sdk/x/slashing/types"
 )
 
 func TestNewQuerier(t *testing.T) {
-	ctx, _, _, _, keeper := CreateTestInput(t, TestParams())
-	querier := NewQuerier(keeper)
+	app := simapp.Setup(false)
+	ctx := app.BaseApp.NewContext(false, abci.Header{})
+	app.SlashingKeeper.SetParams(ctx, keeper.TestParams())
+
+	querier := keeper.NewQuerier(app.SlashingKeeper)
 
 	query := abci.RequestQuery{
 		Path: "",
 		Data: []byte{},
 	}
 
-	_, err := querier(ctx, []string{"parameters"}, query)
+	_, err := querier(ctx, []string{types.QueryParameters}, query)
 	require.NoError(t, err)
 }
 
 func TestQueryParams(t *testing.T) {
 	cdc := codec.New()
-	ctx, _, _, _, keeper := CreateTestInput(t, TestParams())
+	app := simapp.Setup(false)
+	ctx := app.BaseApp.NewContext(false, abci.Header{})
+	app.SlashingKeeper.SetParams(ctx, keeper.TestParams())
+
+	querier := keeper.NewQuerier(app.SlashingKeeper)
+
+	query := abci.RequestQuery{
+		Path: "",
+		Data: []byte{},
+	}
 
 	var params types.Params
 
-	res, errRes := queryParams(ctx, keeper)
-	require.NoError(t, errRes)
-
-	err := cdc.UnmarshalJSON(res, &params)
+	res, err := querier(ctx, []string{types.QueryParameters}, query)
 	require.NoError(t, err)
-	require.Equal(t, keeper.GetParams(ctx), params)
+
+	err = cdc.UnmarshalJSON(res, &params)
+	require.NoError(t, err)
+	require.Equal(t, app.SlashingKeeper.GetParams(ctx), params)
 }