github.com/99designs/gqlgen@v0.17.45/internal/imports/prune_test.go (about)

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