From c973ec02374412cd37ec548d5b1b675fee94f2a1 Mon Sep 17 00:00:00 2001
From: Anna Kozlova <anna.kozlova@jetbrains.com>
Date: Thu, 6 Sep 2018 15:11:37 +0300
Subject: [PATCH] select open/closing brace of class works as on method
 (IDEA-198213)

---
 .../wordSelection/CodeBlockOrInitializerSelectioner.java    | 2 +-
 .../editorActions/wordSelection/JavaTokenSelectioner.java   | 6 +++---
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/CodeBlockOrInitializerSelectioner.java b/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/CodeBlockOrInitializerSelectioner.java
index ed46781f6921..d880291b597b 100644
--- a/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/CodeBlockOrInitializerSelectioner.java
+++ b/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/CodeBlockOrInitializerSelectioner.java
@@ -26,7 +26,7 @@ import java.util.List;
 public class CodeBlockOrInitializerSelectioner extends BasicSelectioner {
   @Override
   public boolean canSelect(@NotNull PsiElement e) {
-    return e instanceof PsiCodeBlock || e instanceof PsiArrayInitializerExpression;
+    return e instanceof PsiCodeBlock || e instanceof PsiArrayInitializerExpression || e instanceof PsiClass;
   }
 
   @Override
diff --git a/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/JavaTokenSelectioner.java b/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/JavaTokenSelectioner.java
index 1837d203a6cb..9869910511f2 100644
--- a/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/JavaTokenSelectioner.java
+++ b/java/java-impl/src/com/intellij/codeInsight/editorActions/wordSelection/JavaTokenSelectioner.java
@@ -15,9 +15,9 @@
  */
 package com.intellij.codeInsight.editorActions.wordSelection;
 
-import com.intellij.psi.*;
-import com.intellij.openapi.util.TextRange;
 import com.intellij.openapi.editor.Editor;
+import com.intellij.openapi.util.TextRange;
+import com.intellij.psi.*;
 import org.jetbrains.annotations.NotNull;
 
 import java.util.List;
@@ -25,7 +25,7 @@ import java.util.List;
 public class JavaTokenSelectioner extends BasicSelectioner {
   @Override
   public boolean canSelect(@NotNull PsiElement e) {
-    return e instanceof PsiJavaToken && !(e instanceof PsiKeyword) && !(e.getParent()instanceof PsiCodeBlock);
+    return e instanceof PsiJavaToken && !(e instanceof PsiKeyword) && !(e.getParent()instanceof PsiCodeBlock) && !(e.getParent() instanceof PsiClass);
   }
 
   @Override