github.com/JarrahG/buffalocli@v0.0.0-20230801092127-b85bfd5d395a/internal/genny/build/transform_main_test.go (about)

     1  package build
     2  
     3  import (
     4  	"testing"
     5  
     6  	"github.com/JarrahG/buffalocli/internal/genny/newapp/web"
     7  	"github.com/JarrahG/buffalocli/internal/genny/testrunner"
     8  	"github.com/gobuffalo/genny/v2/gentest"
     9  	"github.com/stretchr/testify/require"
    10  )
    11  
    12  func Test_transformMain(t *testing.T) {
    13  	r := require.New(t)
    14  
    15  	ref, err := testrunner.WebApp(&web.Options{})
    16  	r.NoError(err)
    17  	main, err := ref.Disk.Find("cmd/app/main.go")
    18  	r.NoError(err)
    19  
    20  	run := gentest.NewRunner()
    21  	run.Disk.Add(main)
    22  
    23  	opts := &Options{}
    24  	run.WithRun(transformMain(opts))
    25  	r.NoError(run.Run())
    26  
    27  	res := run.Results()
    28  	r.Len(res.Files, 1)
    29  	f := res.Files[0]
    30  	r.Contains(f.String(), "func originalMain()")
    31  }