Function TestQueryRewriting

graphql/resolve/query_test.go:37–73

(t *testing.T)

}

func TestQueryRewriting(t *testing.T) {
	b, err := os.ReadFile("query_test.yaml")
	require.NoError(t, err, "Unable to read test file")

	var tests []QueryRewritingCase
	err = yaml.Unmarshal(b, &tests)
	require.NoError(t, err, "Unable to unmarshal tests to yaml.")

	gqlSchema := test.LoadSchemaFromFile(t, "schema.graphql")

	testRewriter := NewQueryRewriter()

	for _, tcase := range tests {
		t.Run(tcase.Name, func(t *testing.T) {
			var vars map[string]interface{}
			if tcase.GQLVariables != "" {
				require.NoError(t, json.Unmarshal([]byte(tcase.GQLVariables), &vars))
			}
			op, err := gqlSchema.Operation(
				&schema.Request{
					Query:     tcase.GQLQuery,
					Variables: vars,
				})
			require.NoError(t, err)
			gqlQuery := test.GetQuery(t, op)

			dgQuery, dgVars, err := testRewriter.Rewrite(context.Background(), gqlQuery)
			require.NoError(t, err)
			require.Equal(t, tcase.DGQuery, dgraph.AsString(dgQuery))
			if len(tcase.DGVars) == 0 {
				require.Empty(t, dgVars)
			} else {
				require.Equal(t, tcase.DGVars, dgVars)
			}
		})
	}
}

type HTTPRewritingCase struct {
	Name             string