From 949549ffbdcfd3bb233835e7e870537dbc8c2261 Mon Sep 17 00:00:00 2001
From: peter <peter@jetbrains.com>
Date: Mon, 17 Nov 2014 12:44:17 +0100
Subject: [PATCH] fixture-based OverrideImplementTest

---
 .../codeInsight/OverrideImplementTest.java    | 22 +++++++++++--------
 1 file changed, 13 insertions(+), 9 deletions(-)

diff --git a/java/java-tests/testSrc/com/intellij/codeInsight/OverrideImplementTest.java b/java/java-tests/testSrc/com/intellij/codeInsight/OverrideImplementTest.java
index 46f40b9bc706..acec8be9ea0e 100644
--- a/java/java-tests/testSrc/com/intellij/codeInsight/OverrideImplementTest.java
+++ b/java/java-tests/testSrc/com/intellij/codeInsight/OverrideImplementTest.java
@@ -15,18 +15,23 @@
  */
 package com.intellij.codeInsight;
 
+import com.intellij.JavaTestUtil;
 import com.intellij.codeInsight.generation.OverrideImplementUtil;
 import com.intellij.psi.PsiClass;
-import com.intellij.psi.PsiElement;
 import com.intellij.psi.util.PsiTreeUtil;
-import com.intellij.testFramework.LightCodeInsightTestCase;
+import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase;
 
 /**
  * @author ven
  */
-public class OverrideImplementTest extends LightCodeInsightTestCase {
+public class OverrideImplementTest extends LightCodeInsightFixtureTestCase {
   private static final String BASE_DIR = "/codeInsight/overrideImplement/";
 
+  @Override
+  protected String getBasePath() {
+    return JavaTestUtil.getRelativeJavaTestDataPath();
+  }
+
   public void testImplementExtensionMethods() { doTest(true); }
   public void testOverrideExtensionMethods() { doTest(false); }
   public void testDoNotImplementExtensionMethods() { doTest(true); }
@@ -37,12 +42,11 @@ public class OverrideImplementTest extends LightCodeInsightTestCase {
 
   private void doTest(boolean toImplement) {
     String name = getTestName(false);
-    configureByFile(BASE_DIR + "before" + name + ".java");
-    int offset = getEditor().getCaretModel().getOffset();
-    PsiElement context = getFile().findElementAt(offset);
-    PsiClass psiClass = PsiTreeUtil.getParentOfType(context, PsiClass.class);
+    myFixture.configureByFile(BASE_DIR + "before" + name + ".java");
+    int offset = myFixture.getEditor().getCaretModel().getOffset();
+    PsiClass psiClass = PsiTreeUtil.findElementOfClassAtOffset(myFixture.getFile(), offset, PsiClass.class, false);
     assert psiClass != null;
-    OverrideImplementUtil.chooseAndOverrideOrImplementMethods(getProject(), getEditor(), psiClass, toImplement);
-    checkResultByFile(BASE_DIR + "after" + name + ".java");
+    OverrideImplementUtil.chooseAndOverrideOrImplementMethods(getProject(), myFixture.getEditor(), psiClass, toImplement);
+    myFixture.checkResultByFile(BASE_DIR + "after" + name + ".java");
   }
 }
\ No newline at end of file