@Override
  public @Nullable PsiFile getPsiFile(@NotNull Document document) {
    final PsiFile psiFile = super.getPsiFile(document);
    if (myUnitTestMode) {
      VirtualFile virtualFile = FileDocumentManager.getInstance().getFile(document);
      if (virtualFile != null) {
        assertFileIsFromCorrectProject(virtualFile);
      }
    }
    return psiFile;
  }
@Override
  public @Nullable PsiFile getPsiFile(@NotNull Document document)