add support for custom ranges

2 files changed, 15 insertions, 0 deletions

diff --git a/src/test/scala/spray/boilerplate/GeneratorSpecs.scala b/src/test/scala/spray/boilerplate/GeneratorSpecs.scala
index d88010b..d1f6af8 100644
--- a/src/test/scala/spray/boilerplate/GeneratorSpecs.scala
+++ b/src/test/scala/spray/boilerplate/GeneratorSpecs.scala
@@ -28,6 +28,18 @@ class GeneratorSpecs extends Specification {
     "support custom separator" in {
       gen4("[#a1#.]") === "a1.a2.a3.a4"
     }
+    "support custom range start" in {
+      gen4("[2..#a1#.]") === "a2.a3.a4"
+    }
+    "support custom range end" in {
+      gen4("[..6#a1#.]") === "a1.a2.a3.a4.a5.a6"
+    }
+    "support custom range" in {
+      gen4("[5..10#a1#.]") === "a5.a6.a7.a8.a9.a10"
+    }
+    "support inner custom range" in {
+      gen4("[#a1([2..#T1#])#]") === "a1(), a2(T2), a3(T2, T3), a4(T2, T3, T4)"
+    }
   }
 
   def gen4(template: String): String = Generator.generateFromTemplate(template, 4)
diff --git a/src/test/scala/spray/boilerplate/TemplateParserSpecs.scala b/src/test/scala/spray/boilerplate/TemplateParserSpecs.scala
index 85b99af..fbf6ee2 100644
--- a/src/test/scala/spray/boilerplate/TemplateParserSpecs.scala
+++ b/src/test/scala/spray/boilerplate/TemplateParserSpecs.scala
@@ -25,5 +25,8 @@ class TemplateParserSpecs extends Specification {
     "a quoted number" in {
       parse("[#T##1#]") === Expand(LiteralString("T1"))
     }
+    "custom range" in {
+      parse("[0..11#T1#]") === Expand(LiteralString("T") ~ Offset(1), range = Range(start = Some(0), end = Some(11)))
+    }
   }
 }