git.sr.ht/~sircmpwn/gqlgen@v0.0.0-20200522192042-c84d29a1c940/internal/imports/prune_test.go (about)

     1  package imports
     2  
     3  import (
     4  	"io/ioutil"
     5  	"testing"
     6  
     7  	"git.sr.ht/~sircmpwn/gqlgen/internal/code"
     8  	"github.com/stretchr/testify/require"
     9  )
    10  
    11  func TestPrune(t *testing.T) {
    12  	// prime the packages cache so that it's not considered uninitialized
    13  
    14  	b, err := Prune("testdata/unused.go", mustReadFile("testdata/unused.go"), &code.Packages{})
    15  	require.NoError(t, err)
    16  	require.Equal(t, string(mustReadFile("testdata/unused.expected.go")), string(b))
    17  }
    18  
    19  func mustReadFile(filename string) []byte {
    20  	b, err := ioutil.ReadFile(filename)
    21  	if err != nil {
    22  		panic(err)
    23  	}
    24  	return b
    25  }