From ef23eaf22d2b776d4d77ff858ab9a60d6791082b Mon Sep 17 00:00:00 2001
From: Tim te Beek <timtebeek@gmail.com>
Date: Sat, 10 Feb 2024 20:43:43 +0100
Subject: [PATCH] Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
---
 .../java/spring/batch/MigrateStepBuilderFactory.java        | 6 +-----
 .../java/spring/batch/MigrateStepBuilderFactoryTest.java    | 2 ++
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/src/main/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactory.java b/src/main/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactory.java
index d7bbc5af..405eab7d 100644
--- a/src/main/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactory.java
+++ b/src/main/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactory.java
@@ -22,11 +22,7 @@
 import org.openrewrite.internal.ListUtils;
 import org.openrewrite.java.*;
 import org.openrewrite.java.search.UsesMethod;
-import org.openrewrite.java.tree.J;
-import org.openrewrite.java.tree.J.ClassDeclaration;
-import org.openrewrite.java.tree.J.MethodDeclaration;
-import org.openrewrite.java.tree.Statement;
-import org.openrewrite.java.tree.TypeUtils;
+import org.openrewrite.java.tree.*;
 
 import java.util.List;
 import java.util.stream.Collectors;
diff --git a/src/testWithSpringBoot_3_0/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactoryTest.java b/src/testWithSpringBoot_3_0/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactoryTest.java
index ff44b337..dc82cbd4 100644
--- a/src/testWithSpringBoot_3_0/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactoryTest.java
+++ b/src/testWithSpringBoot_3_0/java/org/openrewrite/java/spring/batch/MigrateStepBuilderFactoryTest.java
@@ -16,6 +16,7 @@
 package org.openrewrite.java.spring.batch;
 
 import org.junit.jupiter.api.Test;
+import org.openrewrite.DocumentExample;
 import org.openrewrite.InMemoryExecutionContext;
 import org.openrewrite.java.JavaParser;
 import org.openrewrite.test.RecipeSpec;
@@ -37,6 +38,7 @@ public void defaults(RecipeSpec spec) {
           ));
     }
 
+    @DocumentExample
     @Test
     void replaceStepBuilderFactoryWithTasket() {
         // language=java