summaryrefslogtreecommitdiff
path: root/spec/unit/parser/methods
diff options
context:
space:
mode:
authorStig Sandbeck Mathisen <ssm@debian.org>2013-05-26 17:54:29 +0200
committerStig Sandbeck Mathisen <ssm@debian.org>2013-05-26 17:54:29 +0200
commit1a1e7395553ee2244cf3b9c75ac5ce84ebbc4775 (patch)
tree7072a41b5b20a4a134607052954310c69867ed7a /spec/unit/parser/methods
parent99de0b815d9c05804ddda33d5baa94b23ce1c39e (diff)
parent025f00d05226e74a8ae68b2b16122b17a9746f2c (diff)
downloadpuppet-upstream/3.2.1.tar.gz
Imported Upstream version 3.2.1upstream/3.2.1
Diffstat (limited to 'spec/unit/parser/methods')
-rw-r--r--spec/unit/parser/methods/collect_spec.rb110
-rw-r--r--spec/unit/parser/methods/each_spec.rb91
-rwxr-xr-xspec/unit/parser/methods/foreach_spec.rb91
-rw-r--r--spec/unit/parser/methods/reduce_spec.rb67
-rw-r--r--spec/unit/parser/methods/reject_spec.rb73
-rw-r--r--spec/unit/parser/methods/select_spec.rb79
-rw-r--r--spec/unit/parser/methods/shared.rb61
-rw-r--r--spec/unit/parser/methods/slice_spec.rb97
8 files changed, 669 insertions, 0 deletions
diff --git a/spec/unit/parser/methods/collect_spec.rb b/spec/unit/parser/methods/collect_spec.rb
new file mode 100644
index 000000000..13cd9eda6
--- /dev/null
+++ b/spec/unit/parser/methods/collect_spec.rb
@@ -0,0 +1,110 @@
+require 'puppet'
+require 'spec_helper'
+require 'puppet_spec/compiler'
+
+require 'unit/parser/methods/shared'
+
+describe 'the collect method' do
+ include PuppetSpec::Compiler
+
+ before :each do
+ Puppet[:parser] = "future"
+ end
+
+ context "using future parser" do
+ context "in Ruby style should be callable as" do
+ it 'collect on an array (multiplying each value by 2)' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ $a.collect {|$x| $x*2}.foreach {|$v|
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_4")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_6")['ensure'].should == 'present'
+ end
+
+ it 'collect on a hash selecting keys' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>1,'b'=>2,'c'=>3}
+ $a.collect {|$x| $x[0]}.foreach {|$k|
+ file { "/file_$k": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_a")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_b")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_c")['ensure'].should == 'present'
+ end
+ it 'foreach on a hash selecting value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>1,'b'=>2,'c'=>3}
+ $a.collect {|$x| $x[1]}.foreach {|$k|
+ file { "/file_$k": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+ end
+ context "in Java style should be callable as" do
+ shared_examples_for 'java style' do
+ it 'collect on an array (multiplying each value by 2)' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ $a.collect |$x| #{farr}{ $x*2}.foreach |$v| #{farr}{
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_4")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_6")['ensure'].should == 'present'
+ end
+
+ it 'collect on a hash selecting keys' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>1,'b'=>2,'c'=>3}
+ $a.collect |$x| #{farr}{ $x[0]}.foreach |$k| #{farr}{
+ file { "/file_$k": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_a")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_b")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_c")['ensure'].should == 'present'
+ end
+
+ it 'foreach on a hash selecting value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>1,'b'=>2,'c'=>3}
+ $a.collect |$x| #{farr} {$x[1]}.foreach |$k|#{farr}{
+ file { "/file_$k": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+ end
+ describe 'without fat arrow' do
+ it_should_behave_like 'java style' do
+ let(:farr) { '' }
+ end
+ end
+ describe 'with fat arrow' do
+ it_should_behave_like 'java style' do
+ let(:farr) { '=>' }
+ end
+ end
+ end
+ end
+
+ it_should_behave_like 'all iterative functions argument checks', 'collect'
+ it_should_behave_like 'all iterative functions hash handling', 'collect'
+end
diff --git a/spec/unit/parser/methods/each_spec.rb b/spec/unit/parser/methods/each_spec.rb
new file mode 100644
index 000000000..4d276e95d
--- /dev/null
+++ b/spec/unit/parser/methods/each_spec.rb
@@ -0,0 +1,91 @@
+require 'puppet'
+require 'spec_helper'
+require 'puppet_spec/compiler'
+require 'rubygems'
+
+describe 'the each method' do
+ include PuppetSpec::Compiler
+
+ before :each do
+ Puppet[:parser] = 'future'
+ end
+
+ context "should be callable as" do
+ it 'each on an array selecting each value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ $a.each |$v| {
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+ it 'each on an array selecting each value - function call style' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ each ($a) |$index, $v| => {
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+
+ it 'each on an array with index' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [present, absent, present]
+ $a.each |$k,$v| {
+ file { "/file_${$k+1}": ensure => $v }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+
+ it 'each on a hash selecting entries' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>'present','b'=>'absent','c'=>'present'}
+ $a.each |$e| {
+ file { "/file_${e[0]}": ensure => $e[1] }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_a")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_b")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_c")['ensure'].should == 'present'
+ end
+ it 'each on a hash selecting key and value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>present,'b'=>absent,'c'=>present}
+ $a.each |$k, $v| {
+ file { "/file_$k": ensure => $v }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_a")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_b")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_c")['ensure'].should == 'present'
+ end
+ end
+ context "should produce receiver" do
+ it 'each checking produced value using single expression' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1, 3, 2]
+ $b = $a.each |$x| { "unwanted" }
+ file { "/file_${b[1]}":
+ ensure => present
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+
+ end
+end
diff --git a/spec/unit/parser/methods/foreach_spec.rb b/spec/unit/parser/methods/foreach_spec.rb
new file mode 100755
index 000000000..72a9379a4
--- /dev/null
+++ b/spec/unit/parser/methods/foreach_spec.rb
@@ -0,0 +1,91 @@
+require 'puppet'
+require 'spec_helper'
+require 'puppet_spec/compiler'
+require 'rubygems'
+
+describe 'the foreach method' do
+ include PuppetSpec::Compiler
+
+ before :each do
+ Puppet[:parser] = 'future'
+ end
+
+ context "should be callable as" do
+ it 'foreach on an array selecting each value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ $a.foreach {|$v|
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+ it 'foreach on an array selecting each value - function call style' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ foreach ($a) {|$v|
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+
+ it 'foreach on an array with index' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [present, absent, present]
+ $a.foreach {|$k,$v|
+ file { "/file_${$k+1}": ensure => $v }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+
+ it 'foreach on a hash selecting entries' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>'present','b'=>'absent','c'=>'present'}
+ $a.foreach {|$e|
+ file { "/file_${e[0]}": ensure => $e[1] }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_a")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_b")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_c")['ensure'].should == 'present'
+ end
+ it 'foreach on a hash selecting key and value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'a'=>present,'b'=>absent,'c'=>present}
+ $a.foreach {|$k, $v|
+ file { "/file_$k": ensure => $v }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_a")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_b")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_c")['ensure'].should == 'present'
+ end
+ end
+ context "should produce receiver" do
+ it 'each checking produced value using single expression' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1, 3, 2]
+ $b = $a.each |$x| { $x }
+ file { "/file_${b[1]}":
+ ensure => present
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+
+ end
+end
diff --git a/spec/unit/parser/methods/reduce_spec.rb b/spec/unit/parser/methods/reduce_spec.rb
new file mode 100644
index 000000000..99ecfd18d
--- /dev/null
+++ b/spec/unit/parser/methods/reduce_spec.rb
@@ -0,0 +1,67 @@
+require 'puppet'
+require 'spec_helper'
+require 'puppet_spec/compiler'
+
+describe 'the reduce method' do
+ include PuppetSpec::Compiler
+
+ before :all do
+ # enable switching back
+ @saved_parser = Puppet[:parser]
+ # These tests only work with future parser
+ end
+ after :all do
+ # switch back to original
+ Puppet[:parser] = @saved_parser
+ end
+
+ before :each do
+ node = Puppet::Node.new("floppy", :environment => 'production')
+ @compiler = Puppet::Parser::Compiler.new(node)
+ @scope = Puppet::Parser::Scope.new(@compiler)
+ @topscope = @scope.compiler.topscope
+ @scope.parent = @topscope
+ Puppet[:parser] = 'future'
+ end
+
+ context "should be callable as" do
+ it 'reduce on an array' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ $b = $a.reduce {|$memo, $x| $memo + $x }
+ file { "/file_$b": ensure => present }
+ MANIFEST
+
+ catalog.resource(:file, "/file_6")['ensure'].should == 'present'
+ end
+ it 'reduce on an array with start value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1,2,3]
+ $b = $a.reduce(4) {|$memo, $x| $memo + $x }
+ file { "/file_$b": ensure => present }
+ MANIFEST
+
+ catalog.resource(:file, "/file_10")['ensure'].should == 'present'
+ end
+ it 'reduce on a hash' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {a=>1, b=>2, c=>3}
+ $start = [ignored, 4]
+ $b = $a.reduce {|$memo, $x| ['sum', $memo[1] + $x[1]] }
+ file { "/file_${$b[0]}_${$b[1]}": ensure => present }
+ MANIFEST
+
+ catalog.resource(:file, "/file_sum_6")['ensure'].should == 'present'
+ end
+ it 'reduce on a hash with start value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {a=>1, b=>2, c=>3}
+ $start = ['ignored', 4]
+ $b = $a.reduce($start) {|$memo, $x| ['sum', $memo[1] + $x[1]] }
+ file { "/file_${$b[0]}_${$b[1]}": ensure => present }
+ MANIFEST
+
+ catalog.resource(:file, "/file_sum_10")['ensure'].should == 'present'
+ end
+ end
+end
diff --git a/spec/unit/parser/methods/reject_spec.rb b/spec/unit/parser/methods/reject_spec.rb
new file mode 100644
index 000000000..e37eaebc5
--- /dev/null
+++ b/spec/unit/parser/methods/reject_spec.rb
@@ -0,0 +1,73 @@
+require 'puppet'
+require 'spec_helper'
+require 'puppet_spec/compiler'
+
+require 'unit/parser/methods/shared'
+
+describe 'the reject method' do
+ include PuppetSpec::Compiler
+
+ before :each do
+ Puppet[:parser] = 'future'
+ end
+
+ it 'rejects on an array (no berries)' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = ['strawberry','blueberry','orange']
+ $a.reject {|$x| $x =~ /berry$/}.foreach {|$v|
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_orange")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_strawberry").should == nil
+ end
+
+ it 'produces an array when acting on an array' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = ['strawberry','blueberry','orange']
+ $b = $a.reject {|$x| $x =~ /berry$/}
+ file { "/file_${b[0]}": ensure => present }
+
+ MANIFEST
+
+ catalog.resource(:file, "/file_orange")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_strawberry").should == nil
+ end
+
+ it 'rejects on a hash (all berries) by key' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'strawberry'=>'red','blueberry'=>'blue','orange'=>'orange'}
+ $a.reject {|$x| $x[0] =~ /berry$/}.foreach {|$v|
+ file { "/file_${v[0]}": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_orange")['ensure'].should == 'present'
+ end
+
+ it 'produces a hash when acting on a hash' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'strawberry'=>'red','blueberry'=>'blue','grape'=>'purple'}
+ $b = $a.reject {|$x| $x[0] =~ /berry$/}
+ file { "/file_${b[grape]}": ensure => present }
+
+ MANIFEST
+
+ catalog.resource(:file, "/file_purple")['ensure'].should == 'present'
+ end
+
+ it 'rejects on a hash (all berries) by value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'strawb'=>'red berry','blueb'=>'blue berry','orange'=>'orange fruit'}
+ $a.reject {|$x| $x[1] =~ /berry$/}.foreach {|$v|
+ file { "/file_${v[0]}": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_orange")['ensure'].should == 'present'
+ end
+
+ it_should_behave_like 'all iterative functions argument checks', 'reject'
+ it_should_behave_like 'all iterative functions hash handling', 'reject'
+end
diff --git a/spec/unit/parser/methods/select_spec.rb b/spec/unit/parser/methods/select_spec.rb
new file mode 100644
index 000000000..e61ee3a31
--- /dev/null
+++ b/spec/unit/parser/methods/select_spec.rb
@@ -0,0 +1,79 @@
+require 'puppet'
+require 'spec_helper'
+require 'puppet_spec/compiler'
+
+require 'unit/parser/methods/shared'
+
+describe 'the select method' do
+ include PuppetSpec::Compiler
+
+ before :each do
+ Puppet[:parser] = 'future'
+ end
+
+ it 'should select on an array (all berries)' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = ['strawberry','blueberry','orange']
+ $a.select {|$x| $x =~ /berry$/}.foreach {|$v|
+ file { "/file_$v": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_strawberry")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_blueberry")['ensure'].should == 'present'
+ end
+
+ it 'should produce an array when acting on an array' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = ['strawberry','blueberry','orange']
+ $b = $a.select {|$x| $x =~ /berry$/}
+ file { "/file_${b[0]}": ensure => present }
+ file { "/file_${b[1]}": ensure => present }
+ MANIFEST
+
+ catalog.resource(:file, "/file_strawberry")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_blueberry")['ensure'].should == 'present'
+ end
+
+ it 'selects on a hash (all berries) by key' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'strawberry'=>'red','blueberry'=>'blue','orange'=>'orange'}
+ $a.select {|$x| $x[0] =~ /berry$/}.foreach {|$v|
+ file { "/file_${v[0]}": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_strawberry")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_blueberry")['ensure'].should == 'present'
+ end
+
+ it 'should produce a hash when acting on a hash' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'strawberry'=>'red','blueberry'=>'blue','orange'=>'orange'}
+ $b = $a.select {|$x| $x[0] =~ /berry$/}
+ file { "/file_${b['strawberry']}": ensure => present }
+ file { "/file_${b['blueberry']}": ensure => present }
+ file { "/file_${b['orange']}": ensure => present }
+
+ MANIFEST
+
+ catalog.resource(:file, "/file_red")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_blue")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_")['ensure'].should == 'present'
+ end
+
+ it 'selects on a hash (all berries) by value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {'strawb'=>'red berry','blueb'=>'blue berry','orange'=>'orange fruit'}
+ $a.select {|$x| $x[1] =~ /berry$/}.foreach {|$v|
+ file { "/file_${v[0]}": ensure => present }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_strawb")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_blueb")['ensure'].should == 'present'
+ end
+
+ it_should_behave_like 'all iterative functions argument checks', 'select'
+ it_should_behave_like 'all iterative functions hash handling', 'select'
+end
diff --git a/spec/unit/parser/methods/shared.rb b/spec/unit/parser/methods/shared.rb
new file mode 100644
index 000000000..bf9c9f05f
--- /dev/null
+++ b/spec/unit/parser/methods/shared.rb
@@ -0,0 +1,61 @@
+
+shared_examples_for 'all iterative functions hash handling' do |func|
+ it 'passes a hash entry as an array of the key and value' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ {a=>1}.#{func} { |$v| notify { "${v[0]} ${v[1]}": } }
+ MANIFEST
+
+ catalog.resource(:notify, "a 1").should_not be_nil
+ end
+end
+
+shared_examples_for 'all iterative functions argument checks' do |func|
+
+ it 'raises an error when defined with more than 1 argument' do
+ expect do
+ compile_to_catalog(<<-MANIFEST)
+ [1].#{func} { |$x, $yikes| }
+ MANIFEST
+ end.to raise_error(Puppet::Error, /Too few arguments/)
+ end
+
+ it 'raises an error when defined with fewer than 1 argument' do
+ expect do
+ compile_to_catalog(<<-MANIFEST)
+ [1].#{func} { || }
+ MANIFEST
+ end.to raise_error(Puppet::Error, /Too many arguments/)
+ end
+
+ it 'raises an error when used against an unsupported type' do
+ expect do
+ compile_to_catalog(<<-MANIFEST)
+ "not correct".#{func} { |$v| }
+ MANIFEST
+ end.to raise_error(Puppet::Error, /must be an Array or a Hash/)
+ end
+
+ it 'raises an error when called with any parameters besides a block' do
+ expect do
+ compile_to_catalog(<<-MANIFEST)
+ [1].#{func}(1) { |$v| }
+ MANIFEST
+ end.to raise_error(Puppet::Error, /Wrong number of arguments/)
+ end
+
+ it 'raises an error when called without a block' do
+ expect do
+ compile_to_catalog(<<-MANIFEST)
+ [1].#{func}()
+ MANIFEST
+ end.to raise_error(Puppet::Error, /Wrong number of arguments/)
+ end
+
+ it 'raises an error when called without a block' do
+ expect do
+ compile_to_catalog(<<-MANIFEST)
+ [1].#{func}(1)
+ MANIFEST
+ end.to raise_error(Puppet::Error, /must be a parameterized block/)
+ end
+end
diff --git a/spec/unit/parser/methods/slice_spec.rb b/spec/unit/parser/methods/slice_spec.rb
new file mode 100644
index 000000000..b213415a1
--- /dev/null
+++ b/spec/unit/parser/methods/slice_spec.rb
@@ -0,0 +1,97 @@
+require 'puppet'
+require 'spec_helper'
+require 'puppet_spec/compiler'
+require 'rubygems'
+
+describe 'methods' do
+ include PuppetSpec::Compiler
+
+ before :all do
+ # enable switching back
+ @saved_parser = Puppet[:parser]
+ # These tests only work with future parser
+ Puppet[:parser] = 'future'
+ end
+ after :all do
+ # switch back to original
+ Puppet[:parser] = @saved_parser
+ end
+
+ before :each do
+ node = Puppet::Node.new("floppy", :environment => 'production')
+ @compiler = Puppet::Parser::Compiler.new(node)
+ @scope = Puppet::Parser::Scope.new(@compiler)
+ @topscope = @scope.compiler.topscope
+ @scope.parent = @topscope
+ Puppet[:parser] = 'future'
+ end
+
+ context "should be callable on array as" do
+
+ it 'slice with explicit parameters' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1, present, 2, absent, 3, present]
+ $a.slice(2) |$k,$v| {
+ file { "/file_${$k}": ensure => $v }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+ it 'slice with one parameter' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1, present, 2, absent, 3, present]
+ $a.slice(2) |$k| {
+ file { "/file_${$k[0]}": ensure => $k[1] }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+ end
+ it 'slice with shorter last slice' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1, present, 2, present, 3, absent]
+ $a.slice(4) |$a, $b, $c, $d| {
+ file { "/file_$a.$c": ensure => $b }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1.2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3.")['ensure'].should == 'absent'
+ end
+ end
+ context "should be callable on hash as" do
+
+ it 'slice with explicit parameters, missing are empty' do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = {1=>present, 2=>present, 3=>absent}
+ $a.slice(2) |$a,$b| {
+ file { "/file_${a[0]}.${b[0]}": ensure => $a[1] }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1.2")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_3.")['ensure'].should == 'absent'
+ end
+
+ end
+ context "when called without a block" do
+ it "should produce an array with the result" do
+ catalog = compile_to_catalog(<<-MANIFEST)
+ $a = [1, present, 2, absent, 3, present]
+ $a.slice(2).each |$k| {
+ file { "/file_${$k[0]}": ensure => $k[1] }
+ }
+ MANIFEST
+
+ catalog.resource(:file, "/file_1")['ensure'].should == 'present'
+ catalog.resource(:file, "/file_2")['ensure'].should == 'absent'
+ catalog.resource(:file, "/file_3")['ensure'].should == 'present'
+
+ end
+ end
+end