Merge pull request #1051 from 99designs/has-operation-context

Add function to check presense of operation context

graphql/context_operation.go

@@ -62,6 +62,14 @@ func WithOperationContext(ctx context.Context, rc *OperationContext) context.Con
 	return context.WithValue(ctx, operationCtx, rc)
 }
 
+// HasOperationContext checks if the given context is part of an ongoing operation
+//
+// Some errors can happen outside of an operation, eg json unmarshal errors.
+func HasOperationContext(ctx context.Context) bool {
+	_, ok := ctx.Value(operationCtx).(*OperationContext)
+	return ok
+}
+
 // This is just a convenient wrapper method for CollectFields
 func CollectFieldsCtx(ctx context.Context, satisfies []string) []CollectedField {
 	resctx := GetFieldContext(ctx)

graphql/context_operation_test.go

@@ -10,7 +10,22 @@ import (
 
 func TestGetOperationContext(t *testing.T) {
 	rc := &OperationContext{}
-	require.Equal(t, rc, GetOperationContext(WithOperationContext(context.Background(), rc)))
+
+	t.Run("with operation context", func(t *testing.T) {
+		ctx := WithOperationContext(context.Background(), rc)
+
+		require.True(t, HasOperationContext(ctx))
+		require.Equal(t, rc, GetOperationContext(ctx))
+	})
+
+	t.Run("without operation context", func(t *testing.T) {
+		ctx := context.Background()
+
+		require.False(t, HasOperationContext(ctx))
+		require.Panics(t, func() {
+			GetOperationContext(ctx)
+		})
+	})
 }
 
 func TestCollectAllFields(t *testing.T) {