1 files changed, 40 insertions, 16 deletions
diff --git a/plugins/InspectionGadgets/testsrc/com/siyeh/ig/abstraction/TypeMayBeWeakenedInspectionTest.java b/plugins/InspectionGadgets/testsrc/com/siyeh/ig/abstraction/TypeMayBeWeakenedInspectionTest.java
index fec91135e731..df0048a5218f 100644
--- a/plugins/InspectionGadgets/testsrc/com/siyeh/ig/abstraction/TypeMayBeWeakenedInspectionTest.java
+++ b/plugins/InspectionGadgets/testsrc/com/siyeh/ig/abstraction/TypeMayBeWeakenedInspectionTest.java
@@ -1,24 +1,48 @@
 package com.siyeh.ig.abstraction;
 
-import com.intellij.openapi.roots.LanguageLevelProjectExtension;
-import com.intellij.pom.java.LanguageLevel;
-import com.siyeh.ig.IGInspectionTestCase;
+import com.intellij.codeInspection.InspectionProfileEntry;
+import com.siyeh.ig.LightInspectionTestCase;
+import org.jetbrains.annotations.Nullable;
 
-public class TypeMayBeWeakenedInspectionTest extends IGInspectionTestCase {
+public class TypeMayBeWeakenedInspectionTest extends LightInspectionTestCase {
 
-  public void test() throws Exception {
-    final TypeMayBeWeakenedInspection inspection =
-      new TypeMayBeWeakenedInspection();
+  public void testTypeMayBeWeakened() { doTest(); }
+  public void testNumberAdderDemo() { doTest(); }
+  public void testAutoClosableTest() { doTest(); }
+
+  @Override
+  protected String[] getEnvironmentClasses() {
+    return new String[] {
+      "package weaken_type.sub;\n" +
+      "public class NumberAdderImpl implements NumberAdder {\n" +
+      "  public int doSomething() {\n" +
+      "    return getNumberOne() + 1;\n" +
+      "  }\n" +
+      "  protected int getNumberOne() {\n" +
+      "    return 1;\n" +
+      "  }\n" +
+      "}",
+      "package weaken_type.sub;\n" +
+      "public class NumberAdderExtension extends NumberAdderImpl {\n" +
+      "  @Override\n" +
+      "  public int getNumberOne() {\n" +
+      "    return super.getNumberOne();\n" +
+      "  }\n" +
+      "}"
+    };
+  }
+
+  @Override
+  protected String getBasePath() {
+    return "/plugins/InspectionGadgets/test/com/siyeh/igtest/abstraction/weaken_type";
+  }
+
+  @Nullable
+  @Override
+  protected InspectionProfileEntry getInspection() {
+    final TypeMayBeWeakenedInspection inspection = new TypeMayBeWeakenedInspection();
     inspection.doNotWeakenToJavaLangObject = false;
     inspection.onlyWeakentoInterface = false;
-    final LanguageLevelProjectExtension levelProjectExtension = LanguageLevelProjectExtension.getInstance(getProject());
-    final LanguageLevel level = levelProjectExtension.getLanguageLevel();
-    try {
-      levelProjectExtension.setLanguageLevel(LanguageLevel.JDK_1_7);
-      doTest("com/siyeh/igtest/abstraction/weaken_type", inspection);
-    }
-    finally {
-      levelProjectExtension.setLanguageLevel(level);
-    }
+    return inspection;
   }
 }
 \ No newline at end of file