intellij-plugin/Edu-Kotlin/testSrc/com/jetbrains/edu/kotlin/taskDescription/KtTaskDescriptionPsiLinksTest.kt

package com.jetbrains.edu.kotlin.taskDescription

import com.intellij.openapi.fileTypes.FileType
import com.jetbrains.edu.learning.taskToolWindow.links.TaskDescriptionPsiLinksTestBase
import org.jetbrains.kotlin.idea.KotlinFileType
import org.junit.Test

class KtTaskDescriptionPsiLinksTest : TaskDescriptionPsiLinksTestBase() {
  override val fileType: FileType = KotlinFileType.INSTANCE

  @Test
  fun `test navigate to class`() = doTest("psi_element://Bar", """
    class <caret>Bar {
      fun bar() {}
    }
  """) {
    kotlin("Foo.kt", """
      class Foo {
        fun foo() {}
      }
    """)
    kotlin("Bar.kt", """
      class Bar {
        fun bar() {}
      }
    """)
  }

  @Test
  fun `test navigate to method`() = doTest("psi_element://Foo#foo", """
    class Foo {
      fun <caret>foo() {}
    }
  """) {
    kotlin("Foo.kt", """
      class Foo {
        fun foo() {}
      }
    """)
    kotlin("Bar.kt", """
      class Bar {
        fun bar() {}
      }
    """)
  }

  @Test
  fun `test navigate to inner class`() = doTest("psi_element://Foo.FooBar", """
    class Foo {
      fun foo() {}

      data class <caret>FooBar(val x: Int)
    }
  """) {
    kotlin("Foo.kt", """
      class Foo {
        fun foo() {}

        data class FooBar(val x: Int)
      }
    """)
    kotlin("Bar.kt", """
      class Bar {
        fun bar() {}
      }
    """)
  }

  @Test
  fun `test navigate to property`() = doTest("psi_element://Foo#property", """
      class Foo {
        val <caret>property: Int = 1
        fun foo() {}
      }
  """) {
    kotlin("Foo.kt", """
      class Foo {
        val property: Int = 1
        fun foo() {}
      }
    """)
    kotlin("Bar.kt", """
      class Bar {
        fun bar() {}
      }
    """)
  }
}