/*
* Copyright 2000-2017 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.intellij.refactoring;
import com.intellij.JavaTestUtil;
import com.intellij.psi.JavaCodeFragment;
import com.intellij.psi.JavaCodeFragmentFactory;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiElement;
import com.intellij.refactoring.extractMethodObject.ExtractLightMethodObjectHandler;
import org.jetbrains.annotations.NotNull;
public class ExtractMethodObject4DebuggerTest extends LightRefactoringTestCase {
@NotNull
@Override
protected String getTestDataPath() {
return JavaTestUtil.getJavaTestDataPath();
}
private void doTest(String evaluatedText, String expectedCallSite, String expectedClass) throws Exception {
doTest(evaluatedText, expectedCallSite, expectedClass, true);
}
private void doTest(String evaluatedText,
String expectedCallSite,
String expectedClass,
boolean codeBlock) throws Exception {
final String testName = getTestName(false);
configureByFile("/refactoring/extractMethodObject4Debugger/" + testName + ".java");
final int offset = getEditor().getCaretModel().getOffset();
final PsiElement context = getFile().findElementAt(offset);
final JavaCodeFragmentFactory fragmentFactory = JavaCodeFragmentFactory.getInstance(getProject());
final JavaCodeFragment fragment = codeBlock ? fragmentFactory.createCodeBlockCodeFragment(evaluatedText, context, false) : fragmentFactory.createExpressionCodeFragment(evaluatedText, context, null, false);
final ExtractLightMethodObjectHandler.ExtractedData extractedData =
ExtractLightMethodObjectHandler.extractLightMethodObject(getProject(), context, fragment, "test");
assertNotNull(extractedData);
assertEquals(expectedCallSite, extractedData.getGeneratedCallText());
final PsiClass innerClass = extractedData.getGeneratedInnerClass();
assertEquals(expectedClass, innerClass.getText());
}
public void testSimpleGeneration() throws Exception {
doTest("int i = 0; int j = 0;", "Test test = new Test().invoke();int i = test.getI();int j = test.getJ();",
"static class Test {\n" +
" private int i;\n" +
" private int j;\n" +
"\n" +
" public int getI() {\n" +
" return i;\n" +
" }\n" +
"\n" +
" public int getJ() {\n" +
" return j;\n" +
" }\n" +
"\n" +
" public Test invoke() {\n" +
" i = 0;\n" +
" j = 0;\n" +
" return this;\n" +
" }\n" +
" }");
}
public void testInvokeReturnType() throws Exception {
doTest("x = 6; y = 6;", "Test test = new Test().invoke();x = test.getX();y = test.getY();",
"static class Test {\n" +
" private int x;\n" +
" private int y;\n" +
"\n" +
" public int getX() {\n" +
" return x;\n" +
" }\n" +
"\n" +
" public int getY() {\n" +
" return y;\n" +
" }\n" +
"\n" +
" public Test invoke() {\n" +
" x = 6;\n" +
" y = 6;\n" +
" return this;\n" +
" }\n" +
" }");
}
public void testAnonymousClassParams() throws Exception {
doTest("new I() {public void foo(int i) {i++;}};", "I result = Test.invoke();",
"static class Test {\n" +
" static I invoke() {\n" +
" return new I() {\n" +
" public void foo(int i) {\n" +
" i++;\n" +
" }\n" +
" };\n" +
" }\n" +
" }");
}
public void testInnerClass() throws Exception {
doTest(" new I(2).foo()", "new Test().invoke();",
"class Test {\n" +
" public void invoke() {\n" +
" new Sample.I(2).foo();\n" +
" }\n" +
" }");
}
public void testResultExpr() throws Exception {
doTest(" foo()", "int result = new Test().invoke();",
"class Test {\n" +
" public int invoke() {\n" +
" return foo();\n" +
" }\n" +
" }");
}
public void testResultStatements() throws Exception {
doTest("int i = 0;\nfoo()", "Test test = new Test().invoke();int i = test.getI();int result = test.getResult();",
"class Test {\n" +
" private int i;\n" +
" private int result;\n" +
"\n" +
" public int getI() {\n" +
" return i;\n" +
" }\n" +
"\n" +
" public int getResult() {\n" +
" return result;\n" +
" }\n" +
"\n" +
" public Test invoke() {\n" +
" i = 0;\n" +
" result = foo();\n" +
" return this;\n" +
" }\n" +
" }");
}
public void testOffsetsAtCallSite() throws Exception {
doTest("map.entrySet().stream().filter((a) -> (a.getKey()>0));",
"Stream<Map.Entry<Integer,Integer>> result = new Test(map).invoke();",
"static class Test {\n" +
" private Map<Integer, Integer> map;\n" +
"\n" +
" public Test(Map<Integer, Integer> map) {\n" +
" this.map = map;\n" +
" }\n" +
"\n" +
" public Stream<Map.Entry<Integer, Integer>> invoke() {\n" +
" return map.entrySet().stream().filter((a) -> (a.getKey() > 0));\n" +
" }\n" +
" }");
}
public void testHangingFunctionalExpressions() throws Exception {
doTest("() -> {}", "Test.invoke();", "static class Test {\n" +
" static void invoke() {\n" +
" () -> {\n" +
" };\n" +
" }\n" +
" }");
}
public void testArrayInitializer() throws Exception {
doTest("{new Runnable() {public void run(){} } }",
"Runnable[] result = Test.invoke();",
"static class Test {\n" +
" static Runnable[] invoke() {\n" +
" return new Runnable[]{new Runnable() {\n" +
" public void run() {\n" +
" }\n" +
" }};\n" +
" }\n" +
" }", false);
}
public void testNewArrayInitializer() throws Exception {
doTest("new Runnable[] {new Runnable() {public void run(){} } }",
"Runnable[] result = Test.invoke();",
"static class Test {\n" +
" static Runnable[] invoke() {\n" +
" return new Runnable[]{new Runnable() {\n" +
" public void run() {\n" +
" }\n" +
" }};\n" +
" }\n" +
" }", false);
}
public void testThisAndSuperReferences() throws Exception {
doTest("list.forEach(i -> {\n" +
" new Runnable() {\n" +
" int xxx = 0;\n" +
" @Override\n" +
" public void run() {\n" +
" this.xxx = 5; // this stays the same\n" +
" }\n" +
" }.run();\n" +
" this.a++; // have to be qualified\n" +
" super.foo(); // have to be qualified\n" +
" a++;\n" +
" foo();\n" +
" });",
"new Test(list).invoke();",
"class Test {\n" +
" private List<Integer> list;\n" +
"\n" +
" public Test(List<Integer> list) {\n" +
" this.list = list;\n" +
" }\n" +
"\n" +
" public void invoke() {\n" +
" list.forEach(i -> {\n" +
" new Runnable() {\n" +
" int xxx = 0;\n" +
"\n" +
" @Override\n" +
" public void run() {\n" +
" this.xxx = 5; // this stays the same\n" +
" }\n" +
" }.run();\n" +
" Sample.this.a++; // have to be qualified\n" +
" Sample.super.foo(); // have to be qualified\n" +
" a++;\n" +
" foo();\n" +
" });\n" +
" }\n" +
" }", false);
}
public void testOnClosingBrace() throws Exception {
doTest(" foo()", "int result = new Test().invoke();",
"class Test {\n" +
" public int invoke() {\n" +
" return foo();\n" +
" }\n" +
" }");
}
public void testOnClosingBraceLocalClass() throws Exception {
doTest(" foo()", "int result = new Test().invoke();",
"class Test {\n" +
" public int invoke() {\n" +
" return foo();\n" +
" }\n" +
" }");
}
public void testOnFieldInitialization() throws Exception {
doTest(" foo()", "int result = new Test().invoke();",
"class Test {\n" +
" public int invoke() {\n" +
" return foo();\n" +
" }\n" +
" }");
}
public void testOnEmptyMethod() throws Exception {
doTest(" foo()", "int result = Test.invoke();",
"static class Test {\n" +
" static int invoke() {\n" +
" return foo();\n" +
" }\n" +
" }");
}
public void testOnSuperConstructorCall() throws Exception {
doTest(" foo()", "int result = new Test().invoke();",
"class Test {\n" +
" public int invoke() {\n" +
" return foo();\n" +
" }\n" +
" }");
}
public void testOnPrivateField() throws Exception {
doTest(" foo()", "int result = new Test().invoke();",
"class Test {\n" +
" public int invoke() {\n" +
" return foo();\n" +
" }\n" +
" }");
}
}