graphql: make updateGQLSchema always return the new schema

graphql/admin/schema.go

@@ -38,6 +38,10 @@ type updateSchemaResolver struct {
 
 	mutation schema.Mutation
 
+	// new GraphQL schema that is given as mutation input
+	newSchema string
+	// GraphQL schema that is generated from that input
+	generatedSchema string
 	// dgraph schema that is generated from the mutation input
 	newDgraphSchema string
 
@@ -75,10 +79,12 @@ func (asr *updateSchemaResolver) Rewrite(
 	// Disable subscription.
 	schHandler.DisableSubscription()
 
-	_, err = schema.FromString(schHandler.GQLSchema())
+	asr.generatedSchema = schHandler.GQLSchema()
+	_, err = schema.FromString(asr.generatedSchema)
 	if err != nil {
 		return nil, err
 	}
+	asr.newSchema = input.Set.Schema
 	asr.newDgraphSchema = schHandler.DGSchema()
 
 	// There will always be a graphql schema node present in Dgraph cluster. So, we just need to
@@ -109,7 +115,11 @@ func (asr *updateSchemaResolver) Execute(
 	if req == nil || (req.Query == "" && len(req.Mutations) == 0) {
 		// For schema updates, Execute will get called twice.  Once for the
 		// mutation and once for the following query.  This is the query case.
-		b, err := doQuery(asr.admin.schema, asr.mutation.QueryField())
+		b, err := doQuery(&gqlSchema{
+			ID:              asr.admin.schema.ID,
+			Schema:          asr.newSchema,
+			GeneratedSchema: asr.generatedSchema,
+		}, asr.mutation.QueryField())
 		return &dgoapi.Response{Json: b}, err
 	}
 

graphql/e2e/common/common.go

@@ -740,13 +740,18 @@ func addSchema(url, schema string) error {
 				}
 			}
 		}
+		Errors []interface{}
 	}
 
 	err = json.Unmarshal(resp, &addResult)
 	if err != nil {
 		return errors.Wrap(err, "error trying to unmarshal GraphQL mutation result")
 	}
 
+	if len(addResult.Errors) > 0 {
+		return errors.Errorf("%v", addResult.Errors)
+	}
+
 	if addResult.Data.UpdateGQLSchema.GQLSchema.Schema == "" {
 		return errors.New("GraphQL schema mutation failed")
 	}