fixed quickdoc test
authorEkaterina Tuzova <Ekaterina.Tuzova@jetbrains.com>
Mon, 13 Jan 2014 08:37:15 +0000 (12:37 +0400)
committerEkaterina Tuzova <Ekaterina.Tuzova@jetbrains.com>
Mon, 13 Jan 2014 08:37:15 +0000 (12:37 +0400)
python/testSrc/com/jetbrains/python/PyQuickDocTest.java

index 0b78bd5021b90e5964c3b21a2d902264daddc45b..fb45faeeec0902744f8adac6fd13eac6b2ed0645 100644 (file)
@@ -20,6 +20,8 @@ import com.intellij.openapi.vfs.VfsUtil;
 import com.intellij.openapi.vfs.VirtualFile;
 import com.intellij.psi.PsiElement;
 import com.intellij.testFramework.TestDataFile;
+import com.jetbrains.python.documentation.DocStringFormat;
+import com.jetbrains.python.documentation.PyDocumentationSettings;
 import com.jetbrains.python.documentation.PythonDocumentationProvider;
 import com.jetbrains.python.fixtures.LightMarkedTestCase;
 import com.jetbrains.python.fixtures.PyTestCase;
@@ -35,12 +37,23 @@ import java.util.Map;
  */
 public class PyQuickDocTest extends LightMarkedTestCase {
   private PythonDocumentationProvider myProvider;
+  private String myFormat;
 
   @Override
   protected void setUp() throws Exception {
     super.setUp();
     // the provider is stateless, can be reused, as in real life
     myProvider = new PythonDocumentationProvider();
+    final PyDocumentationSettings documentationSettings = PyDocumentationSettings.getInstance(myFixture.getModule());
+    myFormat = documentationSettings.getFormat();
+    documentationSettings.setFormat(DocStringFormat.PLAIN);
+  }
+
+  @Override
+  public void tearDown() throws Exception {
+    final PyDocumentationSettings documentationSettings = PyDocumentationSettings.getInstance(myFixture.getModule());
+    documentationSettings.setFormat(myFormat);
+    super.tearDown();
   }
 
   private void checkByHTML(String text) {