feat(template): scaffold genesis validation tests

starport/services/scaffolder/component.go

@@ -8,10 +8,7 @@ import (
 	"os"
 	"path/filepath"
 
-	"github.com/gobuffalo/genny"
 	"github.com/tendermint/starport/starport/pkg/multiformatname"
-	"github.com/tendermint/starport/starport/pkg/placeholder"
-	modulecreate "github.com/tendermint/starport/starport/templates/module/create"
 )
 
 const (
@@ -21,40 +18,6 @@ const (
 	componentPacket  = "packet"
 )
 
-// supportMsgServer checks if the module supports the MsgServer convention
-// if not, the module codebase is modified to support it
-// https://github.com/cosmos/cosmos-sdk/blob/master/docs/architecture/adr-031-msg-service.md
-func supportMsgServer(
-	replacer placeholder.Replacer,
-	appPath string,
-	opts *modulecreate.MsgServerOptions,
-) (*genny.Generator, error) {
-	// Check if convention used
-	msgServerDefined, err := isMsgServerDefined(appPath, opts.ModuleName)
-	if err != nil {
-		return nil, err
-	}
-	if !msgServerDefined {
-		// Patch the module to support the convention
-		return modulecreate.AddMsgServerConventionToLegacyModule(replacer, opts)
-	}
-	return nil, nil
-}
-
-// isMsgServerDefined checks if the module uses the MsgServer convention for transactions
-// this is checked by verifying the existence of the tx.proto file
-func isMsgServerDefined(appPath, moduleName string) (bool, error) {
-	txProto, err := filepath.Abs(filepath.Join(appPath, "proto", moduleName, "tx.proto"))
-	if err != nil {
-		return false, err
-	}
-
-	if _, err := os.Stat(txProto); os.IsNotExist(err) {
-		return false, nil
-	}
-	return true, err
-}
-
 // checkComponentValidity performs various checks common to all components to verify if it can be scaffolded
 func checkComponentValidity(appPath, moduleName string, compName multiformatname.Name, noMessage bool) error {
 	ok, err := moduleExists(appPath, moduleName)

starport/services/scaffolder/message.go

@@ -116,7 +116,8 @@ func (s *Scaffolder) AddMessage(
 
 	// Check and support MsgServer convention
 	var gens []*genny.Generator
-	g, err = supportMsgServer(
+	gens, err = supportMsgServer(
+		gens,
 		tracer,
 		s.path,
 		&modulecreate.MsgServerOptions{
@@ -129,9 +130,6 @@ func (s *Scaffolder) AddMessage(
 	if err != nil {
 		return sm, err
 	}
-	if g != nil {
-		gens = append(gens, g)
-	}
 
 	// Scaffold
 	g, err = message.NewStargate(tracer, opts)

starport/services/scaffolder/patch.go

@@ -0,0 +1,72 @@
+package scaffolder
+
+import (
+	"os"
+	"path/filepath"
+
+	"github.com/gobuffalo/genny"
+	"github.com/tendermint/starport/starport/pkg/placeholder"
+	modulecreate "github.com/tendermint/starport/starport/templates/module/create"
+)
+
+// supportGenesisTests checks if types/genesis_test.go exists
+// appends the generator to create the file if it doesn't
+func supportGenesisTests(
+	gens []*genny.Generator,
+	appPath,
+	appName,
+	modulePath,
+	moduleName string,
+) ([]*genny.Generator, error) {
+	filepath, err := filepath.Abs(filepath.Join(appPath, "x", moduleName, "types", "genesis_test.go"))
+	if err != nil {
+		return nil, err
+	}
+	if _, err := os.Stat(filepath); os.IsNotExist(err) {
+		g, err := modulecreate.AddGenesisTest(appName, modulePath, moduleName)
+		if err != nil {
+			return nil, err
+		}
+		gens = append(gens, g)
+	}
+	return gens, err
+}
+
+// supportMsgServer checks if the module supports the MsgServer convention
+// appends the generator to support it if it doesn't
+// https://github.com/cosmos/cosmos-sdk/blob/master/docs/architecture/adr-031-msg-service.md
+func supportMsgServer(
+	gens []*genny.Generator,
+	replacer placeholder.Replacer,
+	appPath string,
+	opts *modulecreate.MsgServerOptions,
+) ([]*genny.Generator, error) {
+	// Check if convention used
+	msgServerDefined, err := isMsgServerDefined(appPath, opts.ModuleName)
+	if err != nil {
+		return nil, err
+	}
+	if !msgServerDefined {
+		// Patch the module to support the convention
+		g, err := modulecreate.AddMsgServerConventionToLegacyModule(replacer, opts)
+		if err != nil {
+			return nil, err
+		}
+		gens = append(gens, g)
+	}
+	return gens, nil
+}
+
+// isMsgServerDefined checks if the module uses the MsgServer convention for transactions
+// this is checked by verifying the existence of the tx.proto file
+func isMsgServerDefined(appPath, moduleName string) (bool, error) {
+	txProto, err := filepath.Abs(filepath.Join(appPath, "proto", moduleName, "tx.proto"))
+	if err != nil {
+		return false, err
+	}
+
+	if _, err := os.Stat(txProto); os.IsNotExist(err) {
+		return false, nil
+	}
+	return true, err
+}

starport/services/scaffolder/type.go

@@ -144,7 +144,8 @@ func (s *Scaffolder) AddType(
 		gens []*genny.Generator
 	)
 	// Check and support MsgServer convention
-	g, err = supportMsgServer(
+	gens, err = supportMsgServer(
+		gens,
 		tracer,
 		s.path,
 		&modulecreate.MsgServerOptions{
@@ -157,8 +158,16 @@ func (s *Scaffolder) AddType(
 	if err != nil {
 		return sm, err
 	}
-	if g != nil {
-		gens = append(gens, g)
+
+	gens, err = supportGenesisTests(
+		gens,
+		opts.AppPath,
+		opts.AppName,
+		opts.ModulePath,
+		opts.ModuleName,
+	)
+	if err != nil {
+		return sm, err
 	}
 
 	// create the type generator depending on the model

starport/templates/module/create/genesistest.go

@@ -0,0 +1,23 @@
+package modulecreate
+
+import (
+	"github.com/gobuffalo/genny"
+	"github.com/gobuffalo/plush"
+	"github.com/gobuffalo/plushgen"
+)
+
+// AddGenesisTest returns the generator to generate genesis_test.go
+func AddGenesisTest(appName, modulePath, moduleName string) (*genny.Generator, error) {
+	g := genny.New()
+	if err := g.Box(genesisTestTemplate); err != nil {
+		return g, err
+	}
+	ctx := plush.NewContext()
+	ctx.Set("moduleName", moduleName)
+	ctx.Set("modulePath", modulePath)
+	ctx.Set("appName", appName)
+
+	g.Transformer(plushgen.Transformer(ctx))
+	g.Transformer(genny.Replace("{{moduleName}}", moduleName))
+	return g, nil
+}

starport/templates/module/create/genesistest/x/{{moduleName}}/types/genesis_test.go.plush

@@ -0,0 +1,39 @@
+package types_test
+
+import (
+	"testing"
+	"github.com/stretchr/testify/require"
+	"<%= modulePath %>/x/<%= moduleName %>/types"
+)
+
+func TestGenesisState_Validate(t *testing.T) {
+    for _, tc := range []struct {
+    		desc          string
+    		genState      *types.GenesisState
+    		valid bool
+    } {
+        {
+            desc:     "default is valid",
+            genState: types.DefaultGenesis(),
+            valid:    true,
+        },
+        {
+            desc:     "valid genesis state",
+            genState: &types.GenesisState{
+                // this line is used by starport scaffolding # types/genesis/validField
+            },
+            valid:    true,
+        },
+        // this line is used by starport scaffolding # types/genesis/testcase
+    } {
+        tc := tc
+        t.Run(tc.desc, func(t *testing.T) {
+            err := tc.genState.Validate()
+            if tc.valid {
+                require.NoError(t, err)
+            } else {
+                require.Error(t, err)
+            }
+        })
+    }
+}

starport/templates/module/create/stargate.go

@@ -18,6 +18,9 @@ func NewStargate(opts *CreateOptions) (*genny.Generator, error) {
 	if err := g.Box(msgServerTemplate); err != nil {
 		return g, err
 	}
+	if err := g.Box(genesisTestTemplate); err != nil {
+		return g, err
+	}
 	if err := g.Box(stargateTemplate); err != nil {
 		return g, err
 	}

starport/templates/module/create/templates.go

@@ -16,7 +16,11 @@ var (
 	//go:embed msgserver/* msgserver/**/*
 	fsMsgServer embed.FS
 
-	stargateTemplate  = xgenny.NewEmbedWalker(fsStargate, "stargate/")
-	ibcTemplate       = xgenny.NewEmbedWalker(fsIBC, "ibc/")
-	msgServerTemplate = xgenny.NewEmbedWalker(fsMsgServer, "msgserver/")
+	//go:embed genesistest/* genesistest/**/*
+	fsGenesisTest embed.FS
+
+	stargateTemplate    = xgenny.NewEmbedWalker(fsStargate, "stargate/")
+	ibcTemplate         = xgenny.NewEmbedWalker(fsIBC, "ibc/")
+	msgServerTemplate   = xgenny.NewEmbedWalker(fsMsgServer, "msgserver/")
+	genesisTestTemplate = xgenny.NewEmbedWalker(fsGenesisTest, "genesistest/")
 )

starport/templates/module/placeholders.go

@@ -44,4 +44,8 @@ const (
 	PlaceholderIBCAppScopedKeeperDefinition  = "// this line is used by starport scaffolding # ibc/app/scopedKeeper/definition"
 	PlaceholderIBCAppKeeperArgument          = "// this line is used by starport scaffolding # ibc/app/keeper/argument"
 	PlaceholderIBCAppRouter                  = "// this line is used by starport scaffolding # ibc/app/router"
+
+	// Genesis test
+	PlaceholderTypesGenesisTestcase   = "// this line is used by starport scaffolding # types/genesis/testcase"
+	PlaceholderTypesGenesisValidField = "// this line is used by starport scaffolding # types/genesis/validField"
 )

starport/templates/typed/genesis.go

@@ -6,12 +6,14 @@ import (
 
 	"github.com/gobuffalo/genny"
 	"github.com/tendermint/starport/starport/pkg/placeholder"
+	"github.com/tendermint/starport/starport/templates/module"
 )
 
 func (t *typedStargate) genesisModify(replacer placeholder.Replacer, opts *Options, g *genny.Generator) {
 	g.RunFn(t.genesisProtoModify(replacer, opts))
 	g.RunFn(t.genesisTypesModify(replacer, opts))
 	g.RunFn(t.genesisModuleModify(replacer, opts))
+	g.RunFn(t.genesisTestsModify(replacer, opts))
 }
 
 func (t *typedStargate) genesisProtoModify(replacer placeholder.Replacer, opts *Options) genny.RunFn {
@@ -146,6 +148,71 @@ genesis.%[2]vCount = k.Get%[2]vCount(ctx)
 	}
 }
 
+func (t *typedStargate) genesisTestsModify(replacer placeholder.Replacer, opts *Options) genny.RunFn {
+	return func(r *genny.Runner) error {
+		path := fmt.Sprintf("x/%s/types/genesis_test.go", opts.ModuleName)
+		f, err := r.Disk.Find(path)
+		if err != nil {
+			return err
+		}
+
+		templateValid := `%[1]v
+%[2]vList: []types.%[2]v{
+	{
+		Id: 0,
+	},
+	{
+		Id: 1,
+	},
+},
+%[2]vCount: 2,`
+		replacementValid := fmt.Sprintf(
+			templateValid,
+			module.PlaceholderTypesGenesisValidField,
+			opts.TypeName.UpperCamel,
+		)
+		content := replacer.Replace(f.String(), module.PlaceholderTypesGenesisValidField, replacementValid)
+
+		templateTests := `%[1]v
+{
+	desc:     "duplicated %[2]v",
+	genState: &types.GenesisState{
+		%[3]vList: []types.%[3]v{
+			{
+				Id: 0,
+			},
+			{
+				Id: 0,
+			},
+		},
+	},
+	valid:    false,
+},
+{
+	desc:     "invalid %[2]v count",
+	genState: &types.GenesisState{
+		%[3]vList: []types.%[3]v{
+			{
+				Id: 1,
+			},
+		},
+		%[3]vCount: 0,
+	},
+	valid:    false,
+},`
+		replacementTests := fmt.Sprintf(
+			templateTests,
+			module.PlaceholderTypesGenesisTestcase,
+			opts.TypeName.LowerCamel,
+			opts.TypeName.UpperCamel,
+		)
+		content = replacer.Replace(content, module.PlaceholderTypesGenesisTestcase, replacementTests)
+
+		newFile := genny.NewFileS(path, content)
+		return r.File(newFile)
+	}
+}
+
 // PatchGenesisTypeImport patches types/genesis.go content from the issue:
 // https://github.com/tendermint/starport/issues/992
 func PatchGenesisTypeImport(replacer placeholder.Replacer, content string) string {