add nextNumericToken

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