@Nullable
 private static PsiClass findTargetClass(@NotNull Editor editor, @NotNull PsiFile file) {
   int offset = editor.getCaretModel().getOffset();
   PsiElement element = file.findElementAt(offset);
   return PsiTreeUtil.getParentOfType(element, PsiClass.class, false) == null
       ? null
       : TestIntegrationUtils.findOuterClass(element);
 }