Skip to content

Commit fd26f36

Browse files
committed
Add tests for class category
1 parent 469ab48 commit fd26f36

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

jacodb-ets/src/test/kotlin/org/jacodb/ets/test/EtsFromJsonTest.kt

+15-3
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,7 @@ import org.jacodb.ets.dto.ValueDto
4747
import org.jacodb.ets.dto.dtoModule
4848
import org.jacodb.ets.dto.toEtsLocal
4949
import org.jacodb.ets.dto.toEtsMethod
50+
import org.jacodb.ets.model.EtsClassCategory
5051
import org.jacodb.ets.model.EtsClassSignature
5152
import org.jacodb.ets.model.EtsFile
5253
import org.jacodb.ets.model.EtsFileSignature
@@ -478,8 +479,19 @@ class EtsFromJsonTest {
478479
val path = "/samples/etsir/ast/lang/vararg.ts.json"
479480
val file = loadEtsFileFromResource(path)
480481
val method = file.classes.flatMap { it.methods }.first { it.name == "f" }
481-
assertEquals(method.parameters.size, 2)
482-
assertEquals(method.parameters[0].isRest, false)
483-
assertEquals(method.parameters[1].isRest, true)
482+
assertEquals(2, method.parameters.size)
483+
assertEquals(false, method.parameters[0].isRest)
484+
assertEquals(true, method.parameters[1].isRest)
485+
}
486+
487+
@Test
488+
fun testClassCategory() {
489+
val path = "/samples/etsir/ast/lang/enum.ts.json"
490+
val file = loadEtsFileFromResource(path)
491+
val cls = file.classes.first { it.name == "Animal" }
492+
assertEquals(EtsClassCategory.ENUM, cls.category)
493+
assertEquals(2, cls.fields.size)
494+
assertEquals("Cat", cls.fields[0].name)
495+
assertEquals("Dog", cls.fields[1].name)
484496
}
485497
}

0 commit comments

Comments
 (0)