Skip to content

Commit b41bd71

Browse files
committed
test(database): add SqlContextBuilderTest and ScalaClassContextBuilderTest
- Added `SqlContextBuilderTest` to test the `SqlContextBuilder` class. - Added `ScalaClassContextBuilderTest` to test the `ScalaClassContextBuilder` class.
1 parent 7b8eaff commit b41bd71

File tree

2 files changed

+31
-4
lines changed

2 files changed

+31
-4
lines changed
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
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+

scala/src/test/kotlin/cc/unitmesh/scala/context/ScalaClassContextBuilderTest.kt

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,9 @@ package cc.unitmesh.scala.context
22

33
import com.intellij.psi.PsiFileFactory
44
import com.intellij.testFramework.LightPlatformTestCase
5+
import org.jetbrains.plugins.scala.ScalaFileType
56

6-
class ScalaClassContextBuilderTest: LightPlatformTestCase() {
7+
class ScalaClassContextBuilderTest : LightPlatformTestCase() {
78
fun testShouldGetClassName() {
89
val fileFactory: PsiFileFactory = PsiFileFactory.getInstance(project)
910

@@ -18,12 +19,10 @@ class ScalaClassContextBuilderTest: LightPlatformTestCase() {
1819
end Point
1920
""".trimIndent()
2021

21-
val psiFile = fileFactory.createFileFromText("Point.scala", classCode)
22+
val psiFile = fileFactory.createFileFromText("Point.scala", ScalaFileType.INSTANCE, classCode)
2223
val psiElement = psiFile.children[0]
2324
val classContext = ScalaClassContextBuilder().getClassContext(psiElement, false)!!
2425

2526
assertEquals(classContext.name, "Point")
26-
// TestCase.assertEquals(classContext.methods.size, 2)
27-
// TestCase.assertEquals(classContext.format(), """""")
2827
}
2928
}

0 commit comments

Comments
 (0)