Add test cases for Walmart MockedConstruction pattern (#895)

Added two test cases from customer-requests#1731:
- Test for fully qualified Mockito.mockConstruction() call
- Test for multiple when() statements in lambda

Author: jkschneider

src/test/java/org/openrewrite/java/testing/mockito/MockConstructionToTryWithResourcesTest.java

@@ -18,6 +18,7 @@
 import org.junit.jupiter.api.Test;
 import org.openrewrite.DocumentExample;
 import org.openrewrite.InMemoryExecutionContext;
+import org.openrewrite.Issue;
 import org.openrewrite.java.JavaParser;
 import org.openrewrite.test.RecipeSpec;
 import org.openrewrite.test.RewriteTest;
@@ -267,4 +268,137 @@ void test() {
           )
         );
     }
+
+    @Issue("https://github.com/moderneinc/customer-requests/issues/1731")
+    @Test
+    void shouldHandleFullyQualifiedMockitoMockConstruction() {
+        //language=java
+        rewriteRun(
+          java(
+            """
+              import org.junit.jupiter.api.Test;
+              import org.mockito.MockedConstruction;
+              import org.mockito.Mockito;
+
+              import static org.junit.jupiter.api.Assertions.assertEquals;
+              import static org.mockito.Mockito.when;
+              import static org.mockito.ArgumentMatchers.any;
+
+              class TestClass {
+                  @Test
+                  void test() {
+                      MockedConstruction<A> aMockedConstruction = Mockito.mockConstruction(A.class, (mock, context) -> {
+                          when(mock.method(any())).thenReturn("XYZ");
+                      });
+                      A instance = new A();
+                      assertEquals("XYZ", instance.method("test"));
+                      aMockedConstruction.close();
+                  }
+              }
+              """,
+            """
+              import org.junit.jupiter.api.Test;
+              import org.mockito.MockedConstruction;
+              import org.mockito.Mockito;
+
+              import static org.junit.jupiter.api.Assertions.assertEquals;
+              import static org.mockito.Mockito.when;
+              import static org.mockito.ArgumentMatchers.any;
+
+              class TestClass {
+                  @Test
+                  void test() {
+                      try (MockedConstruction<A> aMockedConstruction = Mockito.mockConstruction(A.class, (mock, context) -> {
+                          when(mock.method(any())).thenReturn("XYZ");
+                      })) {
+                          A instance = new A();
+                          assertEquals("XYZ", instance.method("test"));
+                      }
+                  }
+              }
+              """
+          )
+        );
+    }
+
+    @Issue("https://github.com/moderneinc/customer-requests/issues/1731")
+    @Test
+    void shouldHandleMultipleWhensInLambda() {
+        //language=java
+        rewriteRun(
+          spec -> spec.parser(JavaParser.fromJavaVersion()
+            .classpathFromResources(new InMemoryExecutionContext(), "junit-jupiter-api-5", "mockito-core-5")
+            //language=java
+            .dependsOn(
+              """
+                import java.util.List;
+                public class ItemInflator {
+                    public List<String> inflateNodesWithItemStore(List<Object> items) {
+                        return null;
+                    }
+                    public List<String> inflateWithIRO(List<Object> items) {
+                        return null;
+                    }
+                }
+                """
+            )
+          ),
+          java(
+            """
+              import org.junit.jupiter.api.Test;
+              import org.mockito.MockedConstruction;
+              import org.mockito.Mockito;
+
+              import java.util.List;
+
+              import static org.junit.jupiter.api.Assertions.assertNotNull;
+              import static org.mockito.Mockito.when;
+              import static org.mockito.ArgumentMatchers.anyList;
+
+              class TestClass {
+                  private static final List<String> productList = List.of("product1", "product2");
+
+                  @Test
+                  void test() {
+                      MockedConstruction<ItemInflator> itemInflatorMockedConstruction = Mockito.mockConstruction(ItemInflator.class, (mock, context) -> {
+                          when(mock.inflateNodesWithItemStore(anyList())).thenReturn(productList);
+                          when(mock.inflateWithIRO(anyList())).thenReturn(productList);
+                      });
+                      ItemInflator inflator = new ItemInflator();
+                      assertNotNull(inflator.inflateNodesWithItemStore(List.of()));
+                      assertNotNull(inflator.inflateWithIRO(List.of()));
+                      itemInflatorMockedConstruction.close();
+                  }
+              }
+              """,
+            """
+              import org.junit.jupiter.api.Test;
+              import org.mockito.MockedConstruction;
+              import org.mockito.Mockito;
+
+              import java.util.List;
+
+              import static org.junit.jupiter.api.Assertions.assertNotNull;
+              import static org.mockito.Mockito.when;
+              import static org.mockito.ArgumentMatchers.anyList;
+
+              class TestClass {
+                  private static final List<String> productList = List.of("product1", "product2");
+
+                  @Test
+                  void test() {
+                      try (MockedConstruction<ItemInflator> itemInflatorMockedConstruction = Mockito.mockConstruction(ItemInflator.class, (mock, context) -> {
+                          when(mock.inflateNodesWithItemStore(anyList())).thenReturn(productList);
+                          when(mock.inflateWithIRO(anyList())).thenReturn(productList);
+                      })) {
+                          ItemInflator inflator = new ItemInflator();
+                          assertNotNull(inflator.inflateNodesWithItemStore(List.of()));
+                          assertNotNull(inflator.inflateWithIRO(List.of()));
+                      }
+                  }
+              }
+              """
+          )
+        );
+    }
 }