aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/xyz/driver/core/generators.scala
diff options
context:
space:
mode:
authorAleksandr <ognelisar@gmail.com>2018-04-04 16:52:10 +0700
committerAleksandr <ognelisar@gmail.com>2018-04-04 16:52:10 +0700
commit223628e3c756701309ba8d33ecc886f43857fc15 (patch)
tree7cd251276898464c4745cc69f3c833e4ff567181 /src/main/scala/xyz/driver/core/generators.scala
parent9ffa65c1ccdc5fdea4ccec26c4b39557d45867f7 (diff)
parentbdf9ec57f213eb652ba5fb3b21973d028034d40e (diff)
downloaddriver-core-223628e3c756701309ba8d33ecc886f43857fc15.tar.gz
driver-core-223628e3c756701309ba8d33ecc886f43857fc15.tar.bz2
driver-core-223628e3c756701309ba8d33ecc886f43857fc15.zip
Merge branch 'master' into TM-1431
Diffstat (limited to 'src/main/scala/xyz/driver/core/generators.scala')
-rw-r--r--src/main/scala/xyz/driver/core/generators.scala3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/main/scala/xyz/driver/core/generators.scala b/src/main/scala/xyz/driver/core/generators.scala
index 143044c..3c85447 100644
--- a/src/main/scala/xyz/driver/core/generators.scala
+++ b/src/main/scala/xyz/driver/core/generators.scala
@@ -1,5 +1,6 @@
package xyz.driver.core
+import enumeratum._
import java.math.MathContext
import java.util.UUID
@@ -91,6 +92,8 @@ object generators {
def oneOf[T](items: Set[T]): T = items.toSeq(nextInt(items.size))
+ def oneOf[T <: EnumEntry](enum: Enum[T]): T = oneOf(enum.values: _*)
+
def arrayOf[T: ClassTag](generator: => T, maxLength: Int = DefaultMaxLength, minLength: Int = 0): Array[T] =
Array.fill(nextInt(maxLength, minLength))(generator)