Added support to update Test.class in UpdateTestAnnotation.

fixes #284

Author: traceyyoshima

src/main/java/org/openrewrite/java/testing/junit5/UpdateTestAnnotation.java

@@ -29,6 +29,7 @@
 
 import java.time.Duration;
 import java.util.Comparator;
+import java.util.Set;
 
 public class UpdateTestAnnotation extends Recipe {
 
@@ -61,6 +62,13 @@ private static class UpdateTestAnnotationVisitor extends JavaIsoVisitor<Executio
         @Override
         public J.CompilationUnit visitCompilationUnit(J.CompilationUnit cu, ExecutionContext executionContext) {
             J.CompilationUnit c = super.visitCompilationUnit(cu, executionContext);
+            Set<NameTree> nameTreeSet = c.findType("org.junit.Test");
+            if (!nameTreeSet.isEmpty()) {
+                // Update other references like `Test.class`.
+                c = (J.CompilationUnit) new ChangeType("org.junit.Test", "org.junit.jupiter.api.Test", true)
+                        .getVisitor().visit(cu, executionContext);
+            }
+
             maybeRemoveImport("org.junit.Test");
             doAfterVisit(new JavaIsoVisitor<ExecutionContext>() {
                 @Override

src/test/java/org/openrewrite/java/testing/junit5/JUnit5MigrationTest.java

@@ -37,6 +37,33 @@ public void defaults(RecipeSpec spec) {
             .activateRecipes("org.openrewrite.java.testing.junit5.JUnit4to5Migration"));
     }
 
+    @Issue("https://github.com/openrewrite/rewrite-testing-frameworks/issues/284")
+    @Test
+    void classReference() {
+        rewriteRun(
+          java(
+            """
+              import org.junit.Test;
+              
+              public class Sample {
+                  void method() {
+                      Class<Test> c = Test.class;
+                  }
+              }
+            """,
+            """
+              import org.junit.jupiter.api.Test;
+              
+              public class Sample {
+                  void method() {
+                      Class<Test> c = Test.class;
+                  }
+              }
+            """
+          )
+        );
+    }
+
     @Test
     @Issue("https://github.com/openrewrite/rewrite-testing-frameworks/issues/145")
     void assertThatReceiver() {