Make Source objects comparable

Author: Cito

src/graphql/language/source.py

@@ -56,3 +56,11 @@ def get_location(self, position: int) -> SourceLocation:
 
     def __repr__(self):
         return f"<{self.__class__.__name__} name={self.name!r}>"
+
+    def __eq__(self, other):
+        return (isinstance(other, Source) and other.body == self.body) or (
+            isinstance(other, str) and other == self.body
+        )
+
+    def __ne__(self, other):
+        return not self == other

tests/execution/test_sync.py

@@ -7,13 +7,14 @@
 from graphql.execution import execute
 from graphql.language import parse
 from graphql.type import GraphQLField, GraphQLObjectType, GraphQLSchema, GraphQLString
+from graphql.validation import validate
 
 
 def describe_execute_synchronously_when_possible():
-    def _resolve_sync(root_value, info_):
+    def _resolve_sync(root_value, _info):
         return root_value
 
-    async def _resolve_async(root_value, info_):
+    async def _resolve_async(root_value, _info):
         return root_value
 
     schema = GraphQLSchema(
@@ -77,20 +78,9 @@ def does_not_return_a_promise_for_syntax_errors():
 
         def does_not_return_a_promise_for_validation_errors():
             doc = "fragment Example on Query { unknownField }"
-            assert graphql_sync(schema, doc) == (
-                None,
-                [
-                    {
-                        "message": "Cannot query field 'unknownField' on type 'Query'."
-                        " Did you mean 'asyncField' or 'syncField'?",
-                        "locations": [(1, 29)],
-                    },
-                    {
-                        "message": "Fragment 'Example' is never used.",
-                        "locations": [(1, 1)],
-                    },
-                ],
-            )
+            validation_errors = validate(schema, parse(doc))
+            result = graphql_sync(schema, doc)
+            assert result == (None, validation_errors)
 
         def raises_a_type_error_when_no_query_is_passed():
             with raises(TypeError) as exc_info:

tests/language/test_source.py

@@ -11,6 +11,21 @@ def can_be_stringified():
         source = Source("", "Custom source name")
         assert str(source) == "<Source name='Custom source name'>"
 
+    def can_be_compared():
+        source = Source("foo")
+        assert source == source
+        assert not source != source
+        assert source == "foo"
+        assert not source != "foo"
+        same_source = Source("foo")
+        assert source == same_source
+        assert not source != same_source
+        different_source = Source("bar")
+        assert not source == different_source
+        assert source != different_source
+        assert not source == "bar"
+        assert source != "bar"
+
     def rejects_invalid_location_offset():
         def create_source(location_offset):
             return Source("", "", location_offset)