Merge remote-tracking branch 'pr/56' into pull-56
authorAdrien Thebo <git@somethingsinistral.net>
Tue, 28 May 2013 18:03:51 +0000 (11:03 -0700)
committerAdrien Thebo <git@somethingsinistral.net>
Tue, 28 May 2013 18:07:41 +0000 (11:07 -0700)
Conflicts:
lib/puppet/parser/functions/range.rb
spec/unit/puppet/parser/functions/range_spec.rb

lib/puppet/parser/functions/range.rb
spec/unit/puppet/parser/functions/range_spec.rb

index 825617b..0849491 100644 (file)
@@ -27,6 +27,13 @@ Will return: ["a","b","c"]
     range("host01", "host10")
 
 Will return: ["host01", "host02", ..., "host09", "host10"]
+
+Passing a third argument will cause the generated range to step by that 
+interval, e.g.
+
+    range("0", "9", "2")
+
+Will return: [0,2,4,6,8]
     EOS
   ) do |arguments|
 
@@ -37,6 +44,7 @@ Will return: ["host01", "host02", ..., "host09", "host10"]
     if arguments.size > 1
       start = arguments[0]
       stop  = arguments[1]
+      step  = arguments[2].nil? ? 1 : arguments[2].to_i.abs
 
       type = '..' # We select simplest type for Range available in Ruby ...
 
@@ -71,7 +79,7 @@ Will return: ["host01", "host02", ..., "host09", "host10"]
         when /^(\.\.\.)$/  then (start ... stop) # Exclusive of last element ...
       end
 
-      result = range.collect { |i| i } # Get them all ... Pokemon ...
+      result = range.step(step).collect { |i| i } # Get them all ... Pokemon ...
 
     return result
   end
index 42751f4..5eb290f 100644 (file)
@@ -17,6 +17,21 @@ describe "the range function" do
     result.should(eq(['a','b','c','d']))
   end
 
+  it "should return a letter range given a step of 1" do
+    result = scope.function_range(["a","d","1"])
+    result.should(eq(['a','b','c','d']))
+  end
+
+  it "should return a stepped letter range" do
+    result = scope.function_range(["a","d","2"])
+    result.should(eq(['a','c']))
+  end
+
+  it "should return a stepped letter range given a negative step" do
+    result = scope.function_range(["a","d","-2"])
+    result.should(eq(['a','c']))
+  end
+
   it "should return a number range" do
     result = scope.function_range(["1","4"])
     result.should(eq([1,2,3,4]))
@@ -31,4 +46,20 @@ describe "the range function" do
     expected = (0..10).to_a
     scope.function_range(["00", "10"]).should eq expected
   end
+
+  it "should return a number range given a step of 1" do
+    result = scope.function_range(["1","4","1"])
+    result.should(eq([1,2,3,4]))
+  end
+
+  it "should return a stepped number range" do
+    result = scope.function_range(["1","4","2"])
+    result.should(eq([1,3]))
+  end
+
+  it "should return a stepped number range given a negative step" do
+    result = scope.function_range(["1","4","-2"])
+    result.should(eq([1,3]))
+  end
+
 end