Merge branch 'master' into TM-1431

1 files changed, 20 insertions, 0 deletions

diff --git a/src/test/scala/xyz/driver/core/GeneratorsTest.scala b/src/test/scala/xyz/driver/core/GeneratorsTest.scala
index 62ba7ae..53a3aa9 100644
--- a/src/test/scala/xyz/driver/core/GeneratorsTest.scala
+++ b/src/test/scala/xyz/driver/core/GeneratorsTest.scala
@@ -2,6 +2,8 @@ package xyz.driver.core
 
 import org.scalatest.{Assertions, FlatSpec, Matchers}
 
+import scala.collection.immutable.IndexedSeq
+
 class GeneratorsTest extends FlatSpec with Matchers with Assertions {
   import generators._
 
@@ -175,6 +177,24 @@ class GeneratorsTest extends FlatSpec with Matchers with Assertions {
     Set(pick1, pick2, pick3, pick4, pick5, pick6).size should be >= 1
   }
 
+  it should "be able to generate a specific value from an enumeratum enum" in {
+
+    import enumeratum._
+    sealed trait TestEnumValue extends EnumEntry
+    object TestEnum extends Enum[TestEnumValue] {
+      case object Value1 extends TestEnumValue
+      case object Value2 extends TestEnumValue
+      case object Value3 extends TestEnumValue
+      case object Value4 extends TestEnumValue
+      val values: IndexedSeq[TestEnumValue] = findValues
+    }
+
+    val picks = (1 to 100).map(_ => generators.oneOf(TestEnum))
+
+    TestEnum.values should contain allElementsOf picks
+    picks.toSet.size should be >= 1
+  }
+
   it should "be able to generate array with values generated by generators" in {
 
     val arrayOfTimes = arrayOf(nextTime(), 16)