From 6a2894282e29771c1a8fe8d44f66d1a2ec7380ea Mon Sep 17 00:00:00 2001 From: ArtRand Date: Mon, 26 Feb 2018 16:55:45 -0800 Subject: add nextNumericToken --- src/main/scala/xyz/driver/core/generators.scala | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) 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 -- cgit v1.2.3