UnsupportedOperationException when creating 'scratch file' in IntelliJ IDEA

Signed-off-by: Dmytro Nosan <[email protected]>

Author: nosan

spring-javaformat-intellij-idea/spring-javaformat-intellij-idea-plugin/src/main/java/io/spring/format/formatter/intellij/formatting/SpringJavaFormatFormattingService.java

@@ -78,7 +78,7 @@ public boolean canFormat(@NotNull PsiFile file) {
 	public void formatDocument(@NotNull Document document, @NotNull List<TextRange> formattingRanges,
 			@NotNull FormattingContext formattingContext, boolean canChangeWhiteSpaceOnly, boolean quickFormat) {
 		VirtualFile file = formattingContext.getVirtualFile();
-		Path path = (file != null) ? file.toNioPath() : null;
+		Path path = (file != null) ? file.getFileSystem().getNioPath(file) : null;
 		JavaFormatConfig config = JavaFormatConfig.findFrom(path);
 		Formatter formatter = new Formatter(config);
 		String source = document.getText();

spring-javaformat-intellij-idea/spring-javaformat-intellij-idea-plugin/src/test/java/io/spring/format/formatter/intellij/formatting/SpringJavaFormatFormattingServiceTests.java

@@ -16,6 +16,8 @@
 
 package io.spring.format.formatter.intellij.formatting;
 
+import java.nio.file.Files;
+import java.nio.file.Path;
 import java.util.Collections;
 
 import com.intellij.formatting.FormattingContext;
@@ -25,16 +27,21 @@
 import com.intellij.openapi.fileTypes.FileTypeManager;
 import com.intellij.openapi.fileTypes.PlainTextFileType;
 import com.intellij.openapi.project.Project;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.openapi.vfs.local.CoreLocalFileSystem;
+import com.intellij.openapi.vfs.local.CoreLocalVirtualFile;
 import com.intellij.psi.PsiFile;
+import com.intellij.testFramework.LightVirtualFile;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.io.TempDir;
 
 import io.spring.format.formatter.intellij.state.State;
 
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.BDDMockito.given;
+import static org.mockito.BDDMockito.willAnswer;
 import static org.mockito.Mockito.mock;
-import static org.mockito.Mockito.verify;
 
 /**
  * Tests for {@link SpringJavaFormatFormattingService}.
@@ -72,14 +79,49 @@ void canFormatWhenJavaFileAndActiveReturnsTrue() {
 		assertThat(this.service.canFormat(file)).isTrue();
 	}
 
+	@Test
+	void formatDocumentAppliesFormatting(@TempDir Path projectDir) throws Exception {
+		Files.writeString(projectDir.resolve(".springjavaformatconfig"), "indentation-style=spaces");
+		Document document = mockDocument("public class Hello{"
+				+ "\tpublic void hello() {"
+				+ "\tString value =\t\"Hello World\";}}");
+		FormattingContext formattingContext = mock(FormattingContext.class);
+		VirtualFile virtualFile = new CoreLocalVirtualFile(new CoreLocalFileSystem(), projectDir.resolve("Hello.java"));
+		given(formattingContext.getVirtualFile()).willReturn(virtualFile);
+		this.service.formatDocument(document, Collections.emptyList(), formattingContext, false, false);
+		assertThat(document.getText()).isEqualTo("public class Hello {\n\n"
+				+ "    public void hello() {\n"
+				+ "        String value = \"Hello World\";\n"
+				+ "    }\n\n"
+				+ "}");
+	}
+
 	@Test
 	void formatDocumentAppliesFormatting() {
-		Document document = mock(Document.class);
-		String text = "public class Hello {}";
-		given(document.getText()).willReturn(text);
+		Document document = mockDocument("public class Hello{"
+				+ "\tpublic void hello() {"
+				+ "\tString value =\t\"Hello World\";}}");
 		FormattingContext formattingContext = mock(FormattingContext.class);
+		VirtualFile virtualFile = new LightVirtualFile("Hello.java", document.getText());
+		given(formattingContext.getVirtualFile()).willReturn(virtualFile);
 		this.service.formatDocument(document, Collections.emptyList(), formattingContext, false, false);
-		verify(document).replaceString(20, 20, "\n\n");
+		assertThat(document.getText()).isEqualTo("public class Hello {\n\n"
+				+ "\tpublic void hello() {\n"
+				+ "\t\tString value = \"Hello World\";\n"
+				+ "\t}\n\n"
+				+ "}");
+	}
+
+
+	private Document mockDocument(String text) {
+		Document document = mock(Document.class);
+		StringBuilder documentText = new StringBuilder(text);
+		willAnswer((invocation) -> {
+			documentText.replace(invocation.getArgument(0), invocation.getArgument(1), invocation.getArgument(2));
+			return null;
+		}).given(document).replaceString(any(Integer.class), any(Integer.class), any(CharSequence.class));
+		given(document.getText()).willAnswer((invocation) -> documentText.toString());
+		return document;
 	}
 
 	private PsiFile mockFile(FileType fileType, State state) {