@@ -6,22 +6,24 @@ import (
66 "testing"
77
88 "github.com/jensneuse/diffview"
9+ "github.com/stretchr/testify/require"
910
1011 "github.com/wundergraph/graphql-go-tools/v2/pkg/astparser"
12+ "github.com/wundergraph/graphql-go-tools/v2/pkg/asttransform"
1113 "github.com/wundergraph/graphql-go-tools/v2/pkg/testing/goldie"
1214)
1315
1416func TestGenerator_Generate (t * testing.T ) {
1517 starwarsSchemaBytes , err := os .ReadFile ("./testdata/starwars.schema.graphql" )
16- if err != nil {
17- panic (err )
18- }
18+ require .NoError (t , err )
1919
2020 definition , report := astparser .ParseGraphqlDocumentBytes (starwarsSchemaBytes )
2121 if report .HasErrors () {
2222 t .Fatal (report )
2323 }
2424
25+ require .NoError (t , asttransform .MergeDefinitionWithBaseSchema (& definition ))
26+
2527 gen := NewGenerator ()
2628 var data Data
2729 gen .Generate (& definition , & report , & data )
@@ -30,16 +32,12 @@ func TestGenerator_Generate(t *testing.T) {
3032 }
3133
3234 outputPretty , err := json .MarshalIndent (data , "" , " " )
33- if err != nil {
34- t .Fatal (err )
35- }
35+ require .NoError (t , err )
3636
3737 goldie .Assert (t , "starwars_introspected" , outputPretty )
3838 if t .Failed () {
3939 fixture , err := os .ReadFile ("./fixtures/starwars_introspected.golden" )
40- if err != nil {
41- t .Fatal (err )
42- }
40+ require .NoError (t , err )
4341
4442 diffview .NewGoland ().DiffViewBytes ("startwars_introspected" , fixture , outputPretty )
4543 }
0 commit comments