File tree Expand file tree Collapse file tree 2 files changed +31
-4
lines changed
exts/database/src/test/kotlin/cc/unitmesh/database/util
scala/src/test/kotlin/cc/unitmesh/scala/context Expand file tree Collapse file tree 2 files changed +31
-4
lines changed Original file line number Diff line number Diff line change
1
+ package cc.unitmesh.database.util ;
2
+
3
+ import com.intellij.psi.PsiFileFactory
4
+ import com.intellij.sql.SqlFileType
5
+ import com.intellij.sql.dialects.base.SqlElementFactory
6
+ import com.intellij.sql.psi.SqlPsiFacade
7
+ import com.intellij.testFramework.LightPlatformTestCase
8
+
9
+ class SqlContextBuilderTest : LightPlatformTestCase () {
10
+
11
+ fun testShouldReturnCurrentNamespace () {
12
+ val factory = PsiFileFactory .getInstance(project)
13
+ val demoSqlCode = """
14
+ CREATE SCHEMA demo;
15
+ CREATE TABLE demo.users (
16
+ id INT PRIMARY KEY,
17
+ name VARCHAR(255) NOT NULL
18
+ );
19
+ """ .trimIndent()
20
+
21
+ val psiFile = factory.createFileFromText(" demo.sql" , SqlFileType .INSTANCE , demoSqlCode)
22
+ val sqlFile = psiFile as com.intellij.sql.psi.SqlFile
23
+ val currentNamespace = SqlContextBuilder .getCurrentNamespace(sqlFile)
24
+
25
+ assertEquals(currentNamespace, null )
26
+ }
27
+ }
28
+
Original file line number Diff line number Diff line change @@ -2,8 +2,9 @@ package cc.unitmesh.scala.context
2
2
3
3
import com.intellij.psi.PsiFileFactory
4
4
import com.intellij.testFramework.LightPlatformTestCase
5
+ import org.jetbrains.plugins.scala.ScalaFileType
5
6
6
- class ScalaClassContextBuilderTest : LightPlatformTestCase () {
7
+ class ScalaClassContextBuilderTest : LightPlatformTestCase () {
7
8
fun testShouldGetClassName () {
8
9
val fileFactory: PsiFileFactory = PsiFileFactory .getInstance(project)
9
10
@@ -18,12 +19,10 @@ class ScalaClassContextBuilderTest: LightPlatformTestCase() {
18
19
end Point
19
20
""" .trimIndent()
20
21
21
- val psiFile = fileFactory.createFileFromText(" Point.scala" , classCode)
22
+ val psiFile = fileFactory.createFileFromText(" Point.scala" , ScalaFileType . INSTANCE , classCode)
22
23
val psiElement = psiFile.children[0 ]
23
24
val classContext = ScalaClassContextBuilder ().getClassContext(psiElement, false )!!
24
25
25
26
assertEquals(classContext.name, " Point" )
26
- // TestCase.assertEquals(classContext.methods.size, 2)
27
- // TestCase.assertEquals(classContext.format(), """""")
28
27
}
29
28
}
You can’t perform that action at this time.
0 commit comments