github.com/maeglindeveloper/gqlgen@v0.13.1-0.20210413081235-57808b12a0a0/internal/imports/prune_test.go (about)

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