before_recipe hooks fix (#6735)

Author: greg-at-moderne

rewrite-python/rewrite/src/rewrite/test/rewrite_test.py

@@ -178,6 +178,12 @@ def rewrite_run(self, *source_specs: SourceSpec) -> None:
                 self._expect_parse_print_idempotence(parsed)
             all_parsed.extend(parsed)
 
+        # Apply before_recipe hooks (after idempotence check, before recipe execution)
+        all_parsed = [
+            (spec, spec.before_recipe(sf) or sf) if spec.before_recipe else (spec, sf)
+            for spec, sf in all_parsed
+        ]
+
         # Run the recipe
         source_files = [sf for _, sf in all_parsed]
         lss = InMemoryLargeSourceSet(source_files)
@@ -221,12 +227,6 @@ def _parse(self, specs: List[SourceSpec]) -> List[Tuple[SourceSpec, SourceFile]]
             source = dedent(spec.before)
             parsed = self._parse_python(source, source_path)
 
-            # Call before_recipe hook if provided
-            if spec.before_recipe:
-                modified = spec.before_recipe(parsed)
-                if modified is not None:
-                    parsed = modified
-
             result.append((spec, parsed))
 
         return result

rewrite-python/rewrite/tests/test_rewrite_test.py

@@ -216,7 +216,6 @@ def after_hook(cu: CompilationUnit) -> None:
         assert len(called) == 1
         assert called[0][0] == "after"
 
-    @pytest.mark.skip(reason="The test has been failing for unknown reason, disabling on 2026-02-10")
     def test_before_recipe_can_modify_ast(self):
         """Test that before_recipe can modify the parsed AST."""
         from rewrite import random_id