@@ -47,6 +47,7 @@ import org.jacodb.ets.dto.ValueDto
47
47
import org.jacodb.ets.dto.dtoModule
48
48
import org.jacodb.ets.dto.toEtsLocal
49
49
import org.jacodb.ets.dto.toEtsMethod
50
+ import org.jacodb.ets.model.EtsClassCategory
50
51
import org.jacodb.ets.model.EtsClassSignature
51
52
import org.jacodb.ets.model.EtsFile
52
53
import org.jacodb.ets.model.EtsFileSignature
@@ -478,8 +479,19 @@ class EtsFromJsonTest {
478
479
val path = " /samples/etsir/ast/lang/vararg.ts.json"
479
480
val file = loadEtsFileFromResource(path)
480
481
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)
484
496
}
485
497
}
0 commit comments