fix parsing of repeatable directives when cleaning up the sdl

mirumee · Nov 19, 2024 · 82ece57 · 82ece57
1 parent e4a8101
commit 82ece57
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ariadne/contrib/federation/utils.py b/ariadne/contrib/federation/utils.py
@@ -29,7 +29,7 @@
     f"{_i_token_delimiter}directive"
     f"(?:{_i_token_delimiter})?@({_i_token_name})"
     f"(?:(?:{_i_token_delimiter})?{_i_token_arguments})?"
-    f"{_i_token_delimiter}on"
+    f"{_i_token_delimiter}(?:repeatable)?(?:{_i_token_delimiter})?on"
     f"{_i_token_delimiter}(?:[|]{_i_token_delimiter})?{_i_token_location}"
     f"(?:{_i_token_delimiter}[|]{_i_token_delimiter}{_i_token_location})*"
     ")"

diff --git a/tests/federation/test_utils.py b/tests/federation/test_utils.py
@@ -63,6 +63,8 @@ def test_purge_directives_remove_custom_directives():
 
         directive @another on FIELD
 
+        directive @plural repeatable on FIELD
+
         type Query {
             field1: String @custom
             field2: String @other