cleanup

Author: suejung-sentry

graphql_api/tests/test_validation.py

@@ -12,8 +12,13 @@
     create_max_depth_rule,
 )
 
+
+def resolve_field(*args):
+    return "test"
+
+
 QueryType = GraphQLObjectType(
-    "Query", {"field": GraphQLField(GraphQLString, resolve=lambda *args: "test")}
+    "Query", {"field": GraphQLField(GraphQLString, resolve=resolve_field)}
 )
 schema = GraphQLSchema(query=QueryType)
 
@@ -50,7 +55,16 @@ def test_max_depth_rule_rejects_exceeding_depth():
     )
 
 
-# Tests for MaxAliasesRule
+def test_max_depth_rule_exact_depth():
+    query = """
+    query {
+        field
+    }
+    """
+    errors = validate_query(query, create_max_depth_rule(2))
+    assert not errors, "Expected no errors when query depth matches the limit"
+
+
 def test_max_aliases_rule_allows_within_alias_limit():
     query = """
     query {
@@ -75,16 +89,6 @@ def test_max_aliases_rule_rejects_exceeding_alias_limit():
     assert any("Query uses too many aliases" in str(e) for e in errors)
 
 
-def test_max_depth_rule_exact_depth():
-    query = """
-    query {
-        field
-    }
-    """
-    errors = validate_query(query, create_max_depth_rule(2))
-    assert not errors, "Expected no errors when query depth matches the limit"
-
-
 def test_max_aliases_rule_exact_alias_limit():
     query = """
     query {