Skip to content

Commit 476d7ff

Browse files
committed
add java enum test
1 parent 3a6651f commit 476d7ff

File tree

3 files changed

+20
-0
lines changed

3 files changed

+20
-0
lines changed

src/test/java/models/Level.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
package models;
2+
3+
enum Level {
4+
LOW,
5+
MEDIUM,
6+
HIGH
7+
}

src/test/scala/ScalaModelTest.scala

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -32,6 +32,17 @@ class ScalaModelTest extends AnyFlatSpec with Matchers {
3232
sp.getEnum().asScala.toSet shouldEqual Set("a", "b")
3333
}
3434

35+
it should "extract a java enum" in {
36+
val schemas = ModelConverters.getInstance().readAll(classOf[ModelWithJavaEnum]).asScala
37+
val userSchema = schemas("ModelWithJavaEnum")
38+
39+
val level = userSchema.getProperties().get("level")
40+
level shouldBe a[StringProperty]
41+
42+
val sp = level.asInstanceOf[StringProperty]
43+
sp.getEnum().asScala.toSet shouldEqual Set("LOW", "MEDIUM", "HIGH")
44+
}
45+
3546
it should "read a scala case class with properties" in {
3647
val schemas = ModelConverters.getInstance().readAll(classOf[SimpleUser]).asScala
3748
val userSchema = schemas("SimpleUser")

src/test/scala/models/EnumModels.scala

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,3 +26,5 @@ object TestEnum extends Enumeration {
2626
val AEnum = Value("a")
2727
val BEnum = Value("b")
2828
}
29+
30+
case class ModelWithJavaEnum(level: Level)

0 commit comments

Comments
 (0)