コード例 #1
0
 public RunnableHolder getWriteActionHolder() {
   final RunnableHolder runnableHolder = new RunnableHolder();
   doAnswer(
           invocation -> {
             runnableHolder.setRunnable((Runnable) invocation.getArguments()[0]);
             return null;
           })
       .when(application)
       .runWriteAction(any(Runnable.class));
   return runnableHolder;
 }
コード例 #2
0
  @Test
  public void testJavaFileIsEmpty() throws Exception {
    final PsiJavaFile psiJavaFile = mock(PsiJavaFile.class);
    final PsiClass toImport = mock(PsiClass.class);

    final PsiImportStatement psiImportStatement = mock(PsiImportStatement.class);
    final PsiImportList psiImportList = mock(PsiImportList.class);
    final RunnableHolder writeActionHolder = getWriteActionHolder();

    when(psiJavaFile.getProject()).thenReturn(project);
    when(psiImportStatement.getParent()).thenReturn(psiImportList);
    when(psiElementFactory.createImportStatement(toImport)).thenReturn(psiImportStatement);
    addImportService.addImportStatement(psiJavaFile, toImport);
    assertNotNull(writeActionHolder.getRunnable());
    verify(psiJavaFile, never()).add(any());
    writeActionHolder.getRunnable().run();
    verify(psiJavaFile).add(psiImportList);
  }