github.com/tiagovtristao/plz@v13.4.0+incompatible/src/gc/rewrite_test.go (about)

     1  package gc
     2  
     3  import (
     4  	"io/ioutil"
     5  	"os"
     6  	"path"
     7  	"testing"
     8  
     9  	"github.com/stretchr/testify/assert"
    10  
    11  	"github.com/thought-machine/please/src/core"
    12  	"github.com/thought-machine/please/src/fs"
    13  )
    14  
    15  func TestRewriteFile(t *testing.T) {
    16  	state := core.NewDefaultBuildState()
    17  	// Copy file to avoid any issues with links etc.
    18  	wd, _ := os.Getwd()
    19  	err := fs.CopyFile("src/gc/test_data/before.build", path.Join(wd, "test.build"), 0644)
    20  	assert.NoError(t, err)
    21  	assert.NoError(t, RewriteFile(state, "test.build", []string{"prometheus", "cover"}))
    22  	rewritten, err := ioutil.ReadFile("test.build")
    23  	assert.NoError(t, err)
    24  	after, err := ioutil.ReadFile("src/gc/test_data/after.build")
    25  	assert.NoError(t, err)
    26  	assert.EqualValues(t, string(after), string(rewritten))
    27  }