add tests

Signed-off-by: Shi Chen <[email protected]>
eclipse-jdtls · Feb 24, 2023 · dd96acb · dd96acb
1 parent beba7ca
commit dd96acb
Showing 1 changed file with 64 additions and 0 deletions.
diff --git a/....ls.tests/src/org/eclipse/jdt/ls/core/internal/handlers/DocumentLifeCycleHandlerTest.java b/....ls.tests/src/org/eclipse/jdt/ls/core/internal/handlers/DocumentLifeCycleHandlerTest.java
@@ -43,8 +43,10 @@
 import org.eclipse.core.runtime.jobs.Job;
 import org.eclipse.jdt.core.ICompilationUnit;
 import org.eclipse.jdt.core.IJavaProject;
+import org.eclipse.jdt.core.IMethod;
 import org.eclipse.jdt.core.IPackageFragment;
 import org.eclipse.jdt.core.IPackageFragmentRoot;
+import org.eclipse.jdt.core.IType;
 import org.eclipse.jdt.core.JavaCore;
 import org.eclipse.jdt.core.JavaModelException;
 import org.eclipse.jdt.core.compiler.IProblem;
@@ -204,6 +206,7 @@ private Preferences mockPreferences() {
 		Mockito.lenient().when(mockPreferences.getIncompleteClasspathSeverity()).thenReturn(Severity.ignore);
 		Mockito.lenient().when(this.preferenceManager.getClientPreferences()).thenReturn(clientPreferences);
 		Mockito.lenient().when(clientPreferences.isSupportedCodeActionKind(CodeActionKind.QuickFix)).thenReturn(true);
+		Mockito.lenient().when(preferenceManager.getPreferences().getRenameReferencesEnabled()).thenReturn(true);
 		return mockPreferences;
 	}
 
@@ -699,6 +702,67 @@ public void testNotExpectedPackage() throws Exception {
 		assertEquals("Unexpected number of errors", 0, problems.length);
 	}
 
+	@Test
+	public void testRenameReferences() throws Exception {
+		try {
+			Mockito.lenient().when(preferenceManager.getPreferences().getRenameReferencesEnabled()).thenReturn(true);
+			newDefaultProject();
+			// @formatter:off
+			String content =
+					"package org;\n" +
+					"public class Foo {\n" +
+					"\tpublic void test(){\n" +
+					"\t}\n" +
+					"\n" +
+					"\tpublic String getString(){\n" +
+					"\t}\n" +
+					"}";
+			// @formatter:on
+			temp = createTempFolder();
+			File file = createTempFile(temp, "Foo.java", content);
+			URI uri = file.toURI();
+			ICompilationUnit cu = JDTUtils.resolveCompilationUnit(uri);
+			openDocument(cu, cu.getSource(), 1);
+
+			// change 1
+			IType[] types = cu.getAllTypes();
+			assertEquals("Unexpected number of types", 1, types.length);
+			IMethod[] methods = types[0].getMethods();
+			assertEquals("Unexpected number of methods", 2, methods.length);
+			changeDocument(cu, "test1", 2, JDTUtils.toRange(cu, methods[0].getNameRange().getOffset(), methods[0].getNameRange().getLength()));
+			IMarker[] markers = cu.getResource().findMarkers(BaseDocumentLifeCycleHandler.RENAME_REFERENCE_MARKER_ID, false, IResource.DEPTH_ONE);
+			assertEquals("Unexpected number of markers", 1, markers.length);
+			String originalName = (String) markers[0].getAttribute("originalName");
+			assertEquals("Unexpected originalName", "test", originalName);
+			String newName = (String) markers[0].getAttribute("newName");
+			assertEquals("Unexpected newName", "test1", newName);
+
+			// change 2
+			types = cu.getAllTypes();
+			assertEquals("Unexpected number of types", 1, types.length);
+			methods = types[0].getMethods();
+			assertEquals("Unexpected number of methods", 2, methods.length);
+			changeDocument(cu, "test12", 3, JDTUtils.toRange(cu, methods[0].getNameRange().getOffset(), methods[0].getNameRange().getLength()));
+			markers = cu.getResource().findMarkers(BaseDocumentLifeCycleHandler.RENAME_REFERENCE_MARKER_ID, false, IResource.DEPTH_ONE);
+			assertEquals("Unexpected number of markers", 1, markers.length);
+			originalName = (String) markers[0].getAttribute("originalName");
+			assertEquals("Unexpected originalName", "test", originalName);
+			newName = (String) markers[0].getAttribute("newName");
+			assertEquals("Unexpected newName", "test12", newName);
+
+			// rollback change
+			types = cu.getAllTypes();
+			assertEquals("Unexpected number of types", 1, types.length);
+			methods = types[0].getMethods();
+			assertEquals("Unexpected number of methods", 2, methods.length);
+			changeDocument(cu, "test", 4, JDTUtils.toRange(cu, methods[0].getNameRange().getOffset(), methods[0].getNameRange().getLength()));
+			markers = cu.getResource().findMarkers(BaseDocumentLifeCycleHandler.RENAME_REFERENCE_MARKER_ID, false, IResource.DEPTH_ONE);
+			assertEquals("Unexpected number of markers", 0, markers.length);
+		} finally {
+			Mockito.lenient().when(preferenceManager.getPreferences().getRenameReferencesEnabled()).thenReturn(false);
+		}
+	}
+
 	@Test
 	public void testCreateCompilationUnit() throws Exception {
 		IJavaProject javaProject = newEmptyProject();