diff options
-rw-r--r-- | src/main/scala/xyz/driver/core/generators.scala | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/main/scala/xyz/driver/core/generators.scala b/src/main/scala/xyz/driver/core/generators.scala index e2a4d5a..e842dd2 100644 --- a/src/main/scala/xyz/driver/core/generators.scala +++ b/src/main/scala/xyz/driver/core/generators.scala @@ -20,8 +20,14 @@ object generators { private val DefaultMaxLength = 10 private val StringLetters = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ ".toSet private val NonAmbigiousCharacters = "abcdefghijkmnpqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ23456789".toSet + private val Numbers = "0123456789".toSet - def nextToken(length: Int): String = List.fill(length)(oneOf(NonAmbigiousCharacters)).mkString + private def nextTokenString(length: Int, charSet: Set[Char]): String = + List.fill(length)(oneOf(charSet)).mkString + + def nextToken(length: Int): String = nextTokenString(length, NonAmbigiousCharacters) + + def nextNumericToken(length: Int): String = nextTokenString(length, Numbers) def nextInt(maxValue: Int, minValue: Int = 0): Int = random.nextInt(maxValue - minValue) + minValue |