openide/python/python-markdown/test/com/jetbrains/python/markdown/PyCodeFenceTest.kt

package com.jetbrains.python.markdown

import com.jetbrains.python.fixtures.PyTestCase
import com.jetbrains.python.inspections.PyTypeCheckerInspection

/**
 * Tests for [PyCodeFenceLanguageProvider].
 */
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")
  }

  fun testPyConTypeEnter() {
    myFixture.configureByText("a.md", """
      ```pycon
      >>> "Hello, World!"<caret>
      'Hello, World!'

      ```
    """.trimIndent())
    myFixture.type("\n")
    myFixture.checkResult("""
      ```pycon
      >>> "Hello, World!"
      <caret>
      'Hello, World!'

      ```
    """.trimIndent())
  }

  fun testPyConReformat() {
    myFixture.configureByText("a.md", """
      ```pycon
      >>> 1
      >>> 2
      
      ```
    """.trimIndent())
    reformatFile()
    myFixture.checkResult("""
      ```pycon
      >>> 1
      >>> 2
      
      ```
    """.trimIndent())
  }

  fun testDoctestInjected() {
    myFixture.configureByText("a.md", """
      ```doctest
      foo = 1
      f<caret>

      ```
    """.trimIndent())
    myFixture.completeBasic()
    assertContainsElements(lookupStrings, "foo")
  }

  fun testDefaultPythonDialectsInCompletion() {
    myFixture.configureByText("a.md", """
      ```<caret>
      foo = 1
      f

      ```
    """.trimIndent())
    myFixture.completeBasic()
    assertContainsElements(lookupStrings, "doctest", "python", "pycon")
  }

  //PY-40313
  fun testStringsAndBytesResolvedCorrectlyInPythonFragment() {
    myFixture.configureByText("a.md", """
    ```python
    def expect_str(s:str):
        pass
    
    def expect_bytes(b:bytes):
        pass
    
    # Should not warn
    expect_str("abc")
    
    # Should warn
    expect_bytes(<warning descr="Expected type 'bytes', got 'LiteralString' instead">"abc"</warning>)
    ```
    """.trimIndent())
    myFixture.enableInspections(PyTypeCheckerInspection::class.java)
    myFixture.checkHighlighting(true, false, true)
    assertSdkRootsNotParsed(myFixture.file)
  }

  private val lookupStrings: List<String>
    get() = myFixture.lookupElementStrings ?: emptyList()
}