github.com/spread-ai/gqlgen@v0.0.0-20221124102857-a6c8ef538a1d/internal/imports/prune_test.go (about)

     1  package imports
     2  
     3  import (
     4  	"os"
     5  	"strings"
     6  	"testing"
     7  
     8  	"github.com/spread-ai/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.ReplaceAll(string(mustReadFile("testdata/unused.expected.go")), "\r\n", "\n"), string(b))
    18  }
    19  
    20  func mustReadFile(filename string) []byte {
    21  	b, err := os.ReadFile(filename)
    22  	if err != nil {
    23  		panic(err)
    24  	}
    25  	return b
    26  }