/******************************************************************************* * Copyright (c) 2020 Microsoft Corporation and others. * All rights reserved. This program and the accompanying materials * are made available under the terms of the Eclipse Public License 2.0 * which accompanies this distribution, and is available at * https://www.eclipse.org/legal/epl-2.0/ * * SPDX-License-Identifier: EPL-2.0 * * Contributors: * Microsoft Corporation - initial API and implementation *******************************************************************************/ package org.eclipse.jdt.ls.core.internal.handlers; import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertNotNull; import static org.junit.Assert.assertTrue; import static org.mockito.Mockito.when; import java.util.ArrayList; import java.util.Arrays; import java.util.List; import org.eclipse.core.runtime.NullProgressMonitor; import org.eclipse.jdt.core.ICompilationUnit; import org.eclipse.jdt.core.IJavaProject; import org.eclipse.jdt.core.IPackageFragment; import org.eclipse.jdt.core.IPackageFragmentRoot; import org.eclipse.jdt.core.JavaModelException; import org.eclipse.jdt.ls.core.internal.JDTUtils; import org.eclipse.jdt.ls.core.internal.TextEditUtil; import org.eclipse.jdt.ls.core.internal.handlers.FileEventHandler.FileRenameEvent; import org.eclipse.jdt.ls.core.internal.handlers.FileEventHandler.FileRenameParams; import org.eclipse.jdt.ls.core.internal.managers.AbstractProjectsManagerBasedTest; import org.eclipse.jdt.ls.core.internal.preferences.ClientPreferences; import org.eclipse.jface.text.BadLocationException; import org.eclipse.lsp4j.ResourceOperation; import org.eclipse.lsp4j.TextDocumentEdit; import org.eclipse.lsp4j.TextEdit; import org.eclipse.lsp4j.WorkspaceEdit; import org.eclipse.lsp4j.jsonrpc.messages.Either; import org.junit.Before; import org.junit.Test; public class FileEventHandlerTest extends AbstractProjectsManagerBasedTest { private ClientPreferences clientPreferences; private IPackageFragmentRoot sourceFolder; @Before public void setup() throws Exception { IJavaProject javaProject = newEmptyProject(); sourceFolder = javaProject.getPackageFragmentRoot(javaProject.getProject().getFolder("src")); clientPreferences = preferenceManager.getClientPreferences(); } @Test public void testDidRenameFiles_fileNameRenamed() throws JavaModelException, BadLocationException { when(clientPreferences.isResourceOperationSupported()).thenReturn(true); IPackageFragment pack1 = sourceFolder.createPackageFragment("test1", false, null); StringBuilder builderA = new StringBuilder(); builderA.append("package test1;\n"); builderA.append("public class A {\n"); builderA.append(" public void foo() {\n"); builderA.append(" }\n"); builderA.append("}\n"); ICompilationUnit cuA = pack1.createCompilationUnit("ANew.java", builderA.toString(), false, null); StringBuilder builderB = new StringBuilder(); builderB.append("package test1;\n"); builderB.append("public class B {\n"); builderB.append(" public void foo() {\n"); builderB.append(" A a = new A();\n"); builderB.append(" a.foo();\n"); builderB.append(" }\n"); builderB.append("}\n"); ICompilationUnit cuB = pack1.createCompilationUnit("B.java", builderB.toString(), false, null); String uriA = JDTUtils.toURI(cuA); String oldUriA = uriA.replace("ANew", "A"); WorkspaceEdit edit = FileEventHandler.handleRenameFiles(new FileRenameParams(Arrays.asList(new FileRenameEvent(oldUriA, uriA))), new NullProgressMonitor()); assertNotNull(edit); assertEquals(2, edit.getDocumentChanges().size()); assertTrue(edit.getDocumentChanges().get(0).isLeft()); assertEquals(edit.getDocumentChanges().get(0).getLeft().getTextDocument().getUri(), uriA); assertEquals(TextEditUtil.apply(builderA.toString(), edit.getDocumentChanges().get(0).getLeft().getEdits()), "package test1;\n" + "public class ANew {\n" + " public void foo() {\n" + " }\n" + "}\n" ); assertTrue(edit.getDocumentChanges().get(1).isLeft()); assertEquals(edit.getDocumentChanges().get(1).getLeft().getTextDocument().getUri(), JDTUtils.toURI(cuB)); assertEquals(TextEditUtil.apply(builderB.toString(), edit.getDocumentChanges().get(1).getLeft().getEdits()), "package test1;\n" + "public class B {\n" + " public void foo() {\n" + " ANew a = new ANew();\n" + " a.foo();\n" + " }\n" + "}\n" ); } @Test public void testRenamePackage() throws JavaModelException, BadLocationException { when(clientPreferences.isResourceOperationSupported()).thenReturn(true); IPackageFragment pack1 = sourceFolder.createPackageFragment("parent.pack1", false, null); IPackageFragment pack2 = sourceFolder.createPackageFragment("parent.pack2", false, null); StringBuilder codeA = new StringBuilder(); codeA.append("package parent.pack1;\n"); codeA.append("import parent.pack2.B;\n"); codeA.append("public class A {\n"); codeA.append(" public void foo() {\n"); codeA.append(" B b = new B();\n"); codeA.append(" b.foo();\n"); codeA.append(" }\n"); codeA.append("}\n"); StringBuilder codeB = new StringBuilder(); codeB.append("package parent.pack2;\n"); codeB.append("public class B {\n"); codeB.append(" public B() {}\n"); codeB.append(" public void foo() {}\n"); codeB.append("}\n"); ICompilationUnit cuA = pack1.createCompilationUnit("A.java", codeA.toString(), false, null); ICompilationUnit cuB = pack2.createCompilationUnit("B.java", codeB.toString(), false, null); String pack2Uri = JDTUtils.getFileURI(pack2.getResource()); String newPack2Uri = pack2Uri.replace("pack2", "newpack2"); WorkspaceEdit edit = FileEventHandler.handleWillRenameFiles(new FileRenameParams(Arrays.asList(new FileRenameEvent(pack2Uri, newPack2Uri))), new NullProgressMonitor()); assertNotNull(edit); List<Either<TextDocumentEdit, ResourceOperation>> documentChanges = edit.getDocumentChanges(); assertEquals(2, documentChanges.size()); assertTrue(documentChanges.get(0).isLeft()); assertEquals(documentChanges.get(0).getLeft().getTextDocument().getUri(), JDTUtils.toURI(cuA)); assertEquals(TextEditUtil.apply(codeA.toString(), documentChanges.get(0).getLeft().getEdits()), "package parent.pack1;\n" + "import parent.newpack2.B;\n" + "public class A {\n" + " public void foo() {\n" + " B b = new B();\n" + " b.foo();\n" + " }\n" + "}\n" ); assertTrue(documentChanges.get(1).isLeft()); assertEquals(documentChanges.get(1).getLeft().getTextDocument().getUri(), JDTUtils.toURI(cuB)); assertEquals(TextEditUtil.apply(codeB.toString(), documentChanges.get(1).getLeft().getEdits()), "package parent.newpack2;\n" + "public class B {\n" + " public B() {}\n" + " public void foo() {}\n" + "}\n" ); } @Test public void testRenameSubPackage() throws JavaModelException, BadLocationException { when(clientPreferences.isResourceOperationSupported()).thenReturn(true); IPackageFragment parentPack = sourceFolder.createPackageFragment("parent", false, null); IPackageFragment pack1 = sourceFolder.createPackageFragment("parent.pack1", false, null); IPackageFragment pack2 = sourceFolder.createPackageFragment("parent.pack2", false, null); StringBuilder codeA = new StringBuilder(); codeA.append("package parent.pack1;\n"); codeA.append("import parent.pack2.B;\n"); codeA.append("public class A {\n"); codeA.append(" public void foo() {\n"); codeA.append(" B b = new B();\n"); codeA.append(" b.foo();\n"); codeA.append(" }\n"); codeA.append("}\n"); StringBuilder codeB = new StringBuilder(); codeB.append("package parent.pack2;\n"); codeB.append("public class B {\n"); codeB.append(" public B() {}\n"); codeB.append(" public void foo() {}\n"); codeB.append("}\n"); ICompilationUnit cuA = pack1.createCompilationUnit("A.java", codeA.toString(), false, null); ICompilationUnit cuB = pack2.createCompilationUnit("B.java", codeB.toString(), false, null); String parentPackUri = JDTUtils.getFileURI(parentPack.getResource()); String newParentPackUri = parentPackUri.replace("parent", "newparent"); WorkspaceEdit edit = FileEventHandler.handleWillRenameFiles(new FileRenameParams(Arrays.asList(new FileRenameEvent(parentPackUri, newParentPackUri))), new NullProgressMonitor()); assertNotNull(edit); List<Either<TextDocumentEdit, ResourceOperation>> documentChanges = edit.getDocumentChanges(); assertEquals(3, documentChanges.size()); assertTrue(documentChanges.get(0).isLeft()); assertEquals(documentChanges.get(0).getLeft().getTextDocument().getUri(), JDTUtils.toURI(cuA)); assertTrue(documentChanges.get(1).isLeft()); assertEquals(documentChanges.get(1).getLeft().getTextDocument().getUri(), JDTUtils.toURI(cuA)); List<TextEdit> edits = new ArrayList<>(); edits.addAll(documentChanges.get(0).getLeft().getEdits()); edits.addAll(documentChanges.get(1).getLeft().getEdits()); assertEquals(TextEditUtil.apply(codeA.toString(), edits), "package newparent.pack1;\n" + "import newparent.pack2.B;\n" + "public class A {\n" + " public void foo() {\n" + " B b = new B();\n" + " b.foo();\n" + " }\n" + "}\n" ); assertTrue(documentChanges.get(2).isLeft()); assertEquals(documentChanges.get(2).getLeft().getTextDocument().getUri(), JDTUtils.toURI(cuB)); assertEquals(TextEditUtil.apply(codeB.toString(), documentChanges.get(2).getLeft().getEdits()), "package newparent.pack2;\n" + "public class B {\n" + " public B() {}\n" + " public void foo() {}\n" + "}\n" ); } }