public void testListenerNotifiedWhenOperationsFail() { addLibrary(); // no annotation roots: all operations should fail myFixture.configureByFiles("lib/p/Test.java"); final PsiMethod method = ((PsiJavaFile) myFixture.getFile()).getClasses()[0].getMethods()[0]; startListening(method, AnnotationUtil.NOT_NULL, false); new WriteCommandAction(myProject) { @Override protected void run(final Result result) throws Throwable { ExternalAnnotationsManager.getInstance(myProject) .annotateExternally(method, AnnotationUtil.NOT_NULL, myFixture.getFile(), null); } }.execute(); stopListeningAndCheckEvents(); startListening(method, AnnotationUtil.NOT_NULL, false); new WriteCommandAction(myProject) { @Override protected void run(final Result result) throws Throwable { ExternalAnnotationsManager.getInstance(myProject) .editExternalAnnotation(method, AnnotationUtil.NOT_NULL, null); } }.execute(); stopListeningAndCheckEvents(); startListening(method, AnnotationUtil.NOT_NULL, false); new WriteCommandAction(myProject) { @Override protected void run(final Result result) throws Throwable { ExternalAnnotationsManager.getInstance(myProject) .deannotate(method, AnnotationUtil.NOT_NULL); } }.execute(); stopListeningAndCheckEvents(); }
public static void doTest(@NotNull final CodeInsightTestFixture fixture) { final List<CaretPositionInfo> caretPositions = DartTestUtils.extractPositionMarkers( fixture.getProject(), fixture.getEditor().getDocument()); fixture.doHighlighting(); for (CaretPositionInfo caretPositionInfo : caretPositions) { final int line = fixture.getEditor().getDocument().getLineNumber(caretPositionInfo.caretOffset); final int column = caretPositionInfo.caretOffset - fixture.getEditor().getDocument().getLineStartOffset(line); final String fileNameAndPosition = fixture.getFile().getName() + ":" + (line + 1) + ":" + (column + 1); final PsiReference reference = TargetElementUtil.findReference(fixture.getEditor(), caretPositionInfo.caretOffset); assertNotNull("No reference in " + fileNameAndPosition, reference); final PsiElement resolve = reference.resolve(); final String actualElementPosition = getPresentableElementPosition(fixture, resolve); assertEquals( "Incorrect resolution for reference in " + fileNameAndPosition, caretPositionInfo.expected, actualElementPosition); } }
private void doDeannotate(@NonNls final String testPath, String hint1, String hint2) throws Throwable { myFixture.configureByFile(testPath); final PsiFile file = myFixture.getFile(); final Editor editor = myFixture.getEditor(); assertNotAvailable(hint1); assertNotAvailable(hint2); final DeannotateIntentionAction deannotateFix = new DeannotateIntentionAction(); assertTrue(deannotateFix.isAvailable(myProject, editor, file)); final PsiModifierListOwner container = DeannotateIntentionAction.getContainer(editor, file); startListening(container, AnnotationUtil.NOT_NULL, true); new WriteCommandAction(myProject) { @Override protected void run(final Result result) throws Throwable { ExternalAnnotationsManager.getInstance(myProject) .deannotate(container, AnnotationUtil.NOT_NULL); } }.execute(); stopListeningAndCheckEvents(); FileDocumentManager.getInstance().saveAllDocuments(); IntentionAction fix = myFixture.findSingleIntention(hint1); assertNotNull(fix); fix = myFixture.findSingleIntention(hint2); assertNotNull(fix); assertFalse(deannotateFix.isAvailable(myProject, editor, file)); }
public void testAnnotated() throws Throwable { PsiFile psiFile = myFixture.configureByFile("lib/p/TestAnnotated.java"); PsiTestUtil.addSourceRoot(myModule, psiFile.getVirtualFile().getParent()); final PsiFile file = myFixture.getFile(); final Editor editor = myFixture.getEditor(); assertNotAvailable("Annotate method 'get' as @NotNull"); assertNotAvailable("Annotate method 'get' as @Nullable"); final DeannotateIntentionAction deannotateFix = new DeannotateIntentionAction(); assertFalse(deannotateFix.isAvailable(myProject, editor, file)); }
public void testEditingMultiRootAnnotations() { addLibrary("/content/annoMultiRoot/root1", "/content/annoMultiRoot/root2"); myFixture.configureByFiles( "/content/annoMultiRoot/root1/multiRoot/annotations.xml", "/content/annoMultiRoot/root2/multiRoot/annotations.xml"); myFixture.configureByFiles("lib/multiRoot/Test.java"); final ExternalAnnotationsManager manager = ExternalAnnotationsManager.getInstance(myProject); final PsiMethod method = ((PsiJavaFile) myFixture.getFile()).getClasses()[0].getMethods()[0]; final PsiParameter parameter = method.getParameterList().getParameters()[0]; assertMethodAndParameterAnnotationsValues(manager, method, parameter, "\"foo\""); final PsiAnnotation annotationFromText = JavaPsiFacade.getElementFactory(myProject) .createAnnotationFromText("@Annotation(value=\"bar\")", null); startListening(method, AnnotationUtil.NULLABLE, true); new WriteCommandAction(myProject) { @Override protected void run(final Result result) throws Throwable { manager.editExternalAnnotation( method, AnnotationUtil.NULLABLE, annotationFromText.getParameterList().getAttributes()); } }.execute(); stopListeningAndCheckEvents(); startListening(parameter, AnnotationUtil.NOT_NULL, true); new WriteCommandAction(myProject) { @Override protected void run(final Result result) throws Throwable { manager.editExternalAnnotation( parameter, AnnotationUtil.NOT_NULL, annotationFromText.getParameterList().getAttributes()); } }.execute(); stopListeningAndCheckEvents(); assertMethodAndParameterAnnotationsValues(manager, method, parameter, "\"bar\""); myFixture.checkResultByFile( "content/annoMultiRoot/root1/multiRoot/annotations.xml", "content/annoMultiRoot/root1/multiRoot/annotations_after.xml", false); myFixture.checkResultByFile( "content/annoMultiRoot/root2/multiRoot/annotations.xml", "content/annoMultiRoot/root2/multiRoot/annotations_after.xml", false); }
public void testAnnotateLibrary() throws Throwable { addDefaultLibrary(); myFixture.configureByFiles("lib/p/TestPrimitive.java", "content/anno/p/annotations.xml"); myFixture.configureByFiles("lib/p/Test.java"); final PsiFile file = myFixture.getFile(); final Editor editor = myFixture.getEditor(); final IntentionAction fix = myFixture.findSingleIntention("Annotate method 'get' as @NotNull"); assertTrue(fix.isAvailable(myProject, editor, file)); // expecting other @Nullable annotations to be removed, and default @NotNull to be added List<Trinity<PsiModifierListOwner, String, Boolean>> expectedSequence = new ArrayList<Trinity<PsiModifierListOwner, String, Boolean>>(); for (String notNull : NullableNotNullManager.getInstance(myProject).getNullables()) { expectedSequence.add(Trinity.create(getOwner(), notNull, false)); } expectedSequence.add(Trinity.create(getOwner(), AnnotationUtil.NOT_NULL, true)); startListening(expectedSequence); new WriteCommandAction(myProject) { @Override protected void run(final Result result) throws Throwable { fix.invoke(myProject, editor, file); } }.execute(); FileDocumentManager.getInstance().saveAllDocuments(); final PsiElement psiElement = file.findElementAt(editor.getCaretModel().getOffset()); assertNotNull(psiElement); final PsiModifierListOwner listOwner = PsiTreeUtil.getParentOfType(psiElement, PsiModifierListOwner.class); assertNotNull(listOwner); assertNotNull( ExternalAnnotationsManager.getInstance(myProject) .findExternalAnnotation(listOwner, AnnotationUtil.NOT_NULL)); stopListeningAndCheckEvents(); myFixture.checkResultByFile( "content/anno/p/annotations.xml", "content/anno/p/annotationsAnnotateLibrary_after.xml", false); }
public static void doTest( @NotNull final CodeInsightTestFixture fixture, @NotNull final String text) { PsiFile file = fixture.getFile(); if (file == null) { file = fixture.addFileToProject("file.dart", text); fixture.openFileInEditor(file.getVirtualFile()); } else { final Document document = FileDocumentManager.getInstance().getDocument(file.getVirtualFile()); assert document != null; ApplicationManager.getApplication() .runWriteAction( new Runnable() { @Override public void run() { document.setText(text); } }); } doTest(fixture); }
public static PsiMethodCallExpressionExtractor createExpressionExtractor( CodeInsightTestFixture fixture) { Caret currentCaret = getCaret(fixture); return ExtractorFactory.createMethodCallExpressionExtractor( fixture.getFile(), currentCaret.getSelectionStart(), currentCaret.getSelectionEnd()); }
@NotNull private PsiModifierListOwner getOwner() { return ObjectUtils.assertNotNull( AddAnnotationPsiFix.getContainer(myFixture.getFile(), myFixture.getCaretOffset())); }