1 files changed, 35 insertions, 0 deletions
diff --git a/spec/functions/ensure_resource_spec.rb b/spec/functions/ensure_resource_spec.rb
index 611666e..abd8719 100644
--- a/spec/functions/ensure_resource_spec.rb
+++ b/spec/functions/ensure_resource_spec.rb
@@ -37,4 +37,39 @@ describe 'ensure_resource' do
                             ).and_raise_error(Puppet::Error)
     end
   end
+
+  describe 'when an array of new resources are passed in' do
+    it do
+      should run.with_params('User', ['dan', 'alex'], {})
+      compiler.catalog.resource('User[dan]').to_s.should == 'User[dan]'
+      compiler.catalog.resource('User[alex]').to_s.should == 'User[alex]'
+    end
+  end
+
+  describe 'when an array of existing resources is compared against existing resources' do
+    let :pre_condition do
+      'user { "dan": ensure => present; "alex": ensure => present }'
+    end
+    it do
+      should run.with_params('User', ['dan', 'alex'], {})
+      compiler.catalog.resource('User[dan]').to_s.should == 'User[dan]'
+      compiler.catalog.resource('User[alex]').to_s.should == 'User[alex]'
+    end
+  end
+
+  describe 'when compared against existing resources with attributes' do
+    let :pre_condition do
+      'user { "dan": ensure => present; "alex": ensure => present }'
+    end
+    it do
+      # These should not fail
+      should run.with_params('User', ['dan', 'alex'], {})
+      should run.with_params('User', ['dan', 'alex'], '')
+      should run.with_params('User', ['dan', 'alex'], {'ensure' => 'present'})
+      # This should fail
+      should run.with_params('User', ['dan', 'alex'],
+                             {'ensure' => 'absent'}
+                            ).and_raise_error(Puppet::Error)
+    end
+  end
 end