github.com/segakazzz/buffalo@v0.16.22-0.20210119082501-1f52048d3feb/genny/build/transform_main_test.go (about)

     1  package build
     2  
     3  import (
     4  	"strings"
     5  	"testing"
     6  
     7  	"github.com/gobuffalo/genny/v2"
     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  	run := gentest.NewRunner()
    16  	run.Disk.Add(genny.NewFile("main.go", strings.NewReader(coke.String("main.go"))))
    17  
    18  	opts := &Options{}
    19  	run.WithRun(transformMain(opts))
    20  
    21  	r.NoError(run.Run())
    22  
    23  	res := run.Results()
    24  	r.Len(res.Files, 1)
    25  	f := res.Files[0]
    26  	r.Contains(f.String(), "func originalMain()")
    27  }