Add now required import

openrewrite · Dec 7, 2024 · 0ad2c51 · 0ad2c51
1 parent 51579f4
commit 0ad2c51
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 11 deletions.
diff --git a/...ava/org/openrewrite/java/migrate/IllegalArgumentExceptionToAlreadyConnectedException.java b/...ava/org/openrewrite/java/migrate/IllegalArgumentExceptionToAlreadyConnectedException.java
@@ -54,6 +54,7 @@ public J.Try visitTry(J.Try tryStatement, ExecutionContext ctx) {
                 }
                 return try_.withCatches(ListUtils.map(try_.getCatches(), catch_ -> {
                     if (TypeUtils.isOfClassType(catch_.getParameter().getType(), ILLEGAL_ARGUMENT_EXCEPTION)) {
+                        maybeAddImport(ALREADY_CONNECTED_EXCEPTION);
                         return (J.Try.Catch) new ChangeType(ILLEGAL_ARGUMENT_EXCEPTION, ALREADY_CONNECTED_EXCEPTION, true)
                                 .getVisitor().visit(catch_, ctx);
                     }

diff --git a/...org/openrewrite/java/migrate/IllegalArgumentExceptionToAlreadyConnectedExceptionTest.java b/...org/openrewrite/java/migrate/IllegalArgumentExceptionToAlreadyConnectedExceptionTest.java
@@ -40,8 +40,8 @@ void catchException() {
               import java.net.SocketAddress;
               import java.nio.channels.DatagramChannel;
 
-              public class Test {
-                  public void sendDataCatch() {
+              class Test {
+                  void sendDataCatch() {
                       try {
                           DatagramChannel channel = DatagramChannel.open();
                           channel.send(ByteBuffer.allocate(1024), new java.net.InetSocketAddress("localhost", 8080));
@@ -53,11 +53,12 @@ public void sendDataCatch() {
               """,
             """
               import java.nio.ByteBuffer;
+              import java.nio.channels.AlreadyConnectedException;
               import java.net.SocketAddress;
               import java.nio.channels.DatagramChannel;
 
-              public class Test {
-                  public void sendDataCatch() {
+              class Test {
+                  void sendDataCatch() {
                       try {
                           DatagramChannel channel = DatagramChannel.open();
                           channel.send(ByteBuffer.allocate(1024), new java.net.InetSocketAddress("localhost", 8080));
@@ -81,8 +82,8 @@ void rethrowException() {
               import java.net.SocketAddress;
               import java.nio.channels.DatagramChannel;
 
-              public class Test {
-                  public void sendDataRethrow() {
+              class Test {
+                  void sendDataRethrow() {
                       try {
                           DatagramChannel channel = DatagramChannel.open();
                           channel.send(ByteBuffer.allocate(1024), new java.net.InetSocketAddress("localhost", 8080));
@@ -94,11 +95,12 @@ public void sendDataRethrow() {
               """,
             """
               import java.nio.ByteBuffer;
+              import java.nio.channels.AlreadyConnectedException;
               import java.net.SocketAddress;
               import java.nio.channels.DatagramChannel;
 
-              public class Test {
-                  public void sendDataRethrow() {
+              class Test {
+                  void sendDataRethrow() {
                       try {
                           DatagramChannel channel = DatagramChannel.open();
                           channel.send(ByteBuffer.allocate(1024), new java.net.InetSocketAddress("localhost", 8080));
@@ -118,12 +120,13 @@ void retainOtherCaughtExceptions() {
           //language=java
           java(
             """
-              import java.io.IOException;import java.nio.ByteBuffer;
+              import java.io.IOException;
+              import java.nio.ByteBuffer;
               import java.net.SocketAddress;
               import java.nio.channels.DatagramChannel;
 
-              public class Test {
-                  public void sendData() {
+              class Test {
+                  void sendData() {
                       try {
                           DatagramChannel channel = DatagramChannel.open();
                           channel.send(ByteBuffer.allocate(1024), new java.net.InetSocketAddress("localhost", 8080));