add first test generated methods according to rearranger (IDEA-173880)

This commit is contained in:
Anna Kozlova
2017-06-06 19:07:57 +03:00
parent 2baec1d506
commit 9710edbc0a
@@ -219,7 +219,8 @@ public class JavaTestGenerator implements TestGenerator {
Editor editor,
@Nullable String name,
Set<String> existingNames, PsiMethod anchor) {
PsiMethod method = (PsiMethod)targetClass.addAfter(TestIntegrationUtils.createDummyMethod(targetClass), anchor);
PsiMethod dummyMethod = TestIntegrationUtils.createDummyMethod(targetClass);
PsiMethod method = (PsiMethod)(anchor == null ? targetClass.add(dummyMethod) : targetClass.addAfter(dummyMethod, anchor));
PsiDocumentManager.getInstance(targetClass.getProject()).doPostponedOperationsAndUnblockDocument(editor.getDocument());
TestIntegrationUtils.runTestMethodTemplate(methodKind, descriptor, editor, targetClass, sourceClass, method, name, true, existingNames);
return method;