PY-46321 Tests for Python and Python REPL in Markdown code fences

vlasovskikh · intellij-monorepo-bot · commit ac36f11225bb · 2021-04-27T17:31:59.000Z
GitOrigin-RevId: f6d7f212c363f7852f69a4a54993b36774d9fed6
diff --git a/python/python-markdown/intellij.python.markdown.iml b/python/python-markdown/intellij.python.markdown.iml
@@ -5,12 +5,14 @@
     <content url="file://$MODULE_DIR$">
       <sourceFolder url="file://$MODULE_DIR$/resources" type="java-resource" />
       <sourceFolder url="file://$MODULE_DIR$/src" isTestSource="false" />
+      <sourceFolder url="file://$MODULE_DIR$/test" isTestSource="true" />
     </content>
     <orderEntry type="inheritedJdk" />
     <orderEntry type="sourceFolder" forTests="false" />
     <orderEntry type="library" name="kotlin-stdlib-jdk8" level="project" />
     <orderEntry type="module" module-name="intellij.markdown" />
     <orderEntry type="module" module-name="intellij.python.psi.impl" />
     <orderEntry type="module" module-name="intellij.python.psi" />
+    <orderEntry type="module" module-name="intellij.python.community.tests" scope="TEST" />
   </component>
 </module>
diff --git a/python/python-markdown/test/com/jetbrains/python/markdown/PyCodeFenceTest.kt b/python/python-markdown/test/com/jetbrains/python/markdown/PyCodeFenceTest.kt
@@ -0,0 +1,68 @@
+package com.jetbrains.python.markdown
+
+import com.jetbrains.python.fixtures.PyTestCase
+
+class PyCodeFenceTest : PyTestCase() {
+  fun testPythonInjected() {
+    myFixture.configureByText("a.md", """
+      ```python
+      foo = 1
+      f<caret>
+
+      ```
+    """.trimIndent())
+    myFixture.completeBasic()
+    assertContainsElements(lookupStrings, "foo")
+  }
+
+  fun testPyInjected() {
+    myFixture.configureByText("a.md", """
+      ```py
+      foo = 1
+      f<caret>
+
+      ```
+    """.trimIndent())
+    myFixture.completeBasic()
+    assertContainsElements(lookupStrings, "foo")
+  }
+
+  fun testPyConInjected() {
+    myFixture.configureByText("a.md", """
+      ```pycon
+      >>> foo = 1
+      >>> f<caret>
+
+      ```
+    """.trimIndent())
+    myFixture.completeBasic()
+    assertContainsElements(lookupStrings, "foo")
+  }
+
+  fun testPyConNoCompletionForUndefinedName() {
+    myFixture.configureByText("a.md", """
+      ```pycon
+      >>> print(foo)
+      >>> f<caret>
+
+      ```
+    """.trimIndent())
+    myFixture.completeBasic()
+    assertDoesntContain(lookupStrings, "foo")
+  }
+
+  fun testPythonReplInjected() {
+    myFixture.configureByText("a.md", """
+      ```python-repl
+      >>> foo = 1
+      >>> f<caret>
+
+      ```
+    """.trimIndent())
+    myFixture.completeBasic()
+    assertContainsElements(lookupStrings, "foo")
+  }
+
+  private val lookupStrings: List<String>
+    get() = myFixture.lookupElementStrings ?: emptyList()
+}
diff --git a/python/src/com/jetbrains/python/spellchecker/pythonExtras.dic b/python/src/com/jetbrains/python/spellchecker/pythonExtras.dic
@@ -7,6 +7,7 @@ tuples
 teardown
 doctest
 doctests
+pycon
 unittest
 unittests
 virtualenv
diff --git a/python/testSrc/META-INF/PyCharmCorePlugin.xml b/python/testSrc/META-INF/PyCharmCorePlugin.xml
@@ -5,6 +5,7 @@
   <xi:include href="python-copyright-plugin.xml" xpointer="xpointer(/idea-plugin/*)"/>
   <xi:include href="intellilang-python-support.xml" xpointer="xpointer(/idea-plugin/*)"/>
   <xi:include href="python-rest-plugin.xml" xpointer="xpointer(/idea-plugin/*)"/>
+  <xi:include href="python-markdown-plugin.xml" xpointer="xpointer(/idea-plugin/*)"/>
   <extensions defaultExtensionNs="com.intellij">
     <applicationService serviceInterface="com.jetbrains.python.console.PythonConsoleRunnerFactory"
                         serviceImplementation="com.jetbrains.python.console.PydevConsoleRunnerFactory" />
diff --git a/python/testSrc/intellij.python.community.tests.iml b/python/testSrc/intellij.python.community.tests.iml
@@ -40,5 +40,6 @@
     <orderEntry type="library" scope="TEST" name="StreamEx" level="project" />
     <orderEntry type="module" module-name="intellij.platform.core.ui" scope="TEST" />
     <orderEntry type="module" module-name="intellij.platform.codeStyle.impl" scope="TEST" />
+    <orderEntry type="module" module-name="intellij.python.markdown" scope="RUNTIME" />
   </component>
 </module>
