feat(templates): unit tests for indexed type

starport/templates/typed/indexed/stargate/component/x/{{moduleName}}/keeper/grpc_query_{{typeName}}_test.go.plush

@@ -0,0 +1,104 @@
+package keeper
+
+import (
+	"testing"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/types/query"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+
+	"<%= ModulePath %>/x/<%= ModuleName %>/types"
+)
+
+func Test<%= title(TypeName) %>QuerySingle(t *testing.T) {
+	keeper, ctx := setupKeeper(t)
+	wctx := sdk.WrapSDKContext(ctx)
+	msgs := createN<%= title(TypeName) %>(keeper, ctx, 2)
+	for _, tc := range []struct {
+		desc     string
+		request  *types.QueryGet<%= title(TypeName) %>Request
+		response *types.QueryGet<%= title(TypeName) %>Response
+		err      error
+	}{
+		{
+			desc:     "First",
+			request:  &types.QueryGet<%= title(TypeName) %>Request{Index: msgs[0].Index},
+			response: &types.QueryGet<%= title(TypeName) %>Response{<%= title(TypeName) %>: &msgs[0]},
+		},
+		{
+			desc:     "Second",
+			request:  &types.QueryGet<%= title(TypeName) %>Request{Index: msgs[1].Index},
+			response: &types.QueryGet<%= title(TypeName) %>Response{<%= title(TypeName) %>: &msgs[1]},
+		},
+		{
+			desc:    "KeyNotFound",
+			request: &types.QueryGet<%= title(TypeName) %>Request{Index: "missing"},
+			err:     status.Error(codes.InvalidArgument, "not found"),
+		},
+		{
+			desc: "InvalidRequest",
+			err:  status.Error(codes.InvalidArgument, "invalid request"),
+		},
+	} {
+		tc := tc
+		t.Run(tc.desc, func(t *testing.T) {
+			response, err := keeper.<%= title(TypeName) %>(wctx, tc.request)
+			if tc.err != nil {
+				require.ErrorIs(t, err, tc.err)
+			} else {
+				require.Equal(t, tc.response, response)
+			}
+		})
+	}
+}
+
+func Test<%= title(TypeName) %>QueryPaginated(t *testing.T) {
+	keeper, ctx := setupKeeper(t)
+	wctx := sdk.WrapSDKContext(ctx)
+	msgs := createN<%= title(TypeName) %>(keeper, ctx, 5)
+
+	request := func(next []byte, offset, limit uint64, total bool) *types.QueryAll<%= title(TypeName) %>Request {
+		return &types.QueryAll<%= title(TypeName) %>Request{
+			Pagination: &query.PageRequest{
+				Key:        next,
+				Offset:     offset,
+				Limit:      limit,
+				CountTotal: total,
+			},
+		}
+	}
+	t.Run("ByOffset", func(t *testing.T) {
+		step := 2
+		for i := 0; i < len(msgs); i += step {
+			resp, err := keeper.<%= title(TypeName) %>All(wctx, request(nil, uint64(i), uint64(step), false))
+			require.NoError(t, err)
+			for j := i; j < len(msgs) && j < i+step; j++ {
+				assert.Equal(t, &msgs[j], resp.<%= title(TypeName) %>[j-i])
+			}
+		}
+	})
+	t.Run("ByKey", func(t *testing.T) {
+		step := 2
+		var next []byte
+		for i := 0; i < len(msgs); i += step {
+			resp, err := keeper.<%= title(TypeName) %>All(wctx, request(next, 0, uint64(step), false))
+			require.NoError(t, err)
+			for j := i; j < len(msgs) && j < i+step; j++ {
+				assert.Equal(t, &msgs[j], resp.<%= title(TypeName) %>[j-i])
+			}
+			next = resp.Pagination.NextKey
+		}
+	})
+	t.Run("Total", func(t *testing.T) {
+		resp, err := keeper.<%= title(TypeName) %>All(wctx, request(nil, 0, 0, true))
+		require.NoError(t, err)
+		require.Equal(t, len(msgs), int(resp.Pagination.Total))
+	})
+	t.Run("InvalidRequest", func(t *testing.T) {
+		_, err := keeper.<%= title(TypeName) %>All(wctx, nil)
+		require.ErrorIs(t, err, status.Error(codes.InvalidArgument, "invalid request"))
+	})
+}

starport/templates/typed/indexed/stargate/component/x/{{moduleName}}/keeper/{{typeName}}_test.go.plush

@@ -0,0 +1,46 @@
+package keeper
+
+import (
+	"fmt"
+	"testing"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/stretchr/testify/assert"
+
+	"<%= ModulePath %>/x/<%= ModuleName %>/types"
+)
+
+func createN<%= title(TypeName) %>(keeper *Keeper, ctx sdk.Context, n int) []types.<%= title(TypeName) %> {
+	items := make([]types.<%= title(TypeName) %>, n)
+	for i := range items {
+		items[i].Creator = "any"
+		items[i].Index = fmt.Sprintf("%d", i)
+		keeper.Set<%= title(TypeName) %>(ctx, items[i])
+	}
+	return items
+}
+
+func Test<%= title(TypeName) %>Get(t *testing.T) {
+	keeper, ctx := setupKeeper(t)
+	items := createN<%= title(TypeName) %>(keeper, ctx, 10)
+	for _, item := range items {
+		rst, found := keeper.Get<%= title(TypeName) %>(ctx, item.Index)
+		assert.True(t, found)
+		assert.Equal(t, item, rst)
+	}
+}
+func Test<%= title(TypeName) %>Remove(t *testing.T) {
+	keeper, ctx := setupKeeper(t)
+	items := createN<%= title(TypeName) %>(keeper, ctx, 10)
+	for _, item := range items {
+		keeper.Remove<%= title(TypeName) %>(ctx, item.Index)
+		_, found := keeper.Get<%= title(TypeName) %>(ctx, item.Index)
+		assert.False(t, found)
+	}
+}
+
+func Test<%= title(TypeName) %>GetAll(t *testing.T) {
+	keeper, ctx := setupKeeper(t)
+	items := createN<%= title(TypeName) %>(keeper, ctx, 10)
+	assert.Equal(t, items, keeper.GetAll<%= title(TypeName) %>(ctx))
+}

starport/templates/typed/indexed/stargate/messages/x/{{moduleName}}/keeper/msg_server_{{typeName}}_test.go.plush

@@ -0,0 +1,119 @@
+package keeper
+
+import (
+	"fmt"
+	"testing"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	sdkerrors "github.com/cosmos/cosmos-sdk/types/errors"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+
+    "<%= ModulePath %>/x/<%= ModuleName %>/types"
+)
+
+func Test<%= title(TypeName) %>MsgServerCreate(t *testing.T) {
+	keeper, ctx := setupKeeper(t)
+	srv := NewMsgServerImpl(*keeper)
+	wctx := sdk.WrapSDKContext(ctx)
+	creator := "A"
+	for i := 0; i < 5; i++ {
+		idx := fmt.Sprintf("%d", i)
+		expected := &types.MsgCreate<%= title(TypeName) %>{Creator: creator, Index: idx}
+		_, err := srv.Create<%= title(TypeName) %>(wctx, expected)
+		require.NoError(t, err)
+		rst, found := keeper.Get<%= title(TypeName) %>(ctx, expected.Index)
+		require.True(t, found)
+		assert.Equal(t, expected.Creator, rst.Creator)
+	}
+}
+
+func Test<%= title(TypeName) %>MsgServerUpdate(t *testing.T) {
+	creator := "A"
+	index := "any"
+
+	for _, tc := range []struct {
+		desc    string
+		request *types.MsgUpdate<%= title(TypeName) %>
+		err     error
+	}{
+		{
+			desc:    "Completed",
+			request: &types.MsgUpdate<%= title(TypeName) %>{Creator: creator, Index: index},
+		},
+		{
+			desc:    "Unauthorized",
+			request: &types.MsgUpdate<%= title(TypeName) %>{Creator: "B", Index: index},
+			err:     sdkerrors.ErrUnauthorized,
+		},
+		{
+			desc:    "KeyNotFound",
+			request: &types.MsgUpdate<%= title(TypeName) %>{Creator: creator, Index: "missing"},
+			err:     sdkerrors.ErrKeyNotFound,
+		},
+	} {
+		tc := tc
+		t.Run(tc.desc, func(t *testing.T) {
+			keeper, ctx := setupKeeper(t)
+			srv := NewMsgServerImpl(*keeper)
+			wctx := sdk.WrapSDKContext(ctx)
+			expected := &types.MsgCreate<%= title(TypeName) %>{Creator: creator, Index: index}
+			_, err := srv.Create<%= title(TypeName) %>(wctx, expected)
+			require.NoError(t, err)
+
+			_, err = srv.Update<%= title(TypeName) %>(wctx, tc.request)
+			if tc.err != nil {
+				require.ErrorIs(t, err, tc.err)
+			} else {
+				require.NoError(t, err)
+				rst, found := keeper.Get<%= title(TypeName) %>(ctx, expected.Index)
+				require.True(t, found)
+				assert.Equal(t, expected.Creator, rst.Creator)
+			}
+		})
+	}
+}
+
+func Test<%= title(TypeName) %>MsgServerDelete(t *testing.T) {
+	creator := "A"
+	index := "any"
+
+	for _, tc := range []struct {
+		desc    string
+		request *types.MsgDelete<%= title(TypeName) %>
+		err     error
+	}{
+		{
+			desc:    "Completed",
+			request: &types.MsgDelete<%= title(TypeName) %>{Creator: creator, Index: index},
+		},
+		{
+			desc:    "Unauthorized",
+			request: &types.MsgDelete<%= title(TypeName) %>{Creator: "B", Index: index},
+			err:     sdkerrors.ErrUnauthorized,
+		},
+		{
+			desc:    "KeyNotFound",
+			request: &types.MsgDelete<%= title(TypeName) %>{Creator: creator, Index: "missing"},
+			err:     sdkerrors.ErrKeyNotFound,
+		},
+	} {
+		tc := tc
+		t.Run(tc.desc, func(t *testing.T) {
+			keeper, ctx := setupKeeper(t)
+			srv := NewMsgServerImpl(*keeper)
+			wctx := sdk.WrapSDKContext(ctx)
+
+			_, err := srv.Create<%= title(TypeName) %>(wctx, &types.MsgCreate<%= title(TypeName) %>{Creator: creator, Index: index})
+			require.NoError(t, err)
+			_, err = srv.Delete<%= title(TypeName) %>(wctx, tc.request)
+			if tc.err != nil {
+				require.ErrorIs(t, err, tc.err)
+			} else {
+				require.NoError(t, err)
+				_, found := keeper.Get<%= title(TypeName) %>(ctx, tc.request.Index)
+				require.False(t, found)
+			}
+		})
+	}
+}