1 files changed, 46 insertions, 0 deletions
diff --git a/spec/unit/puppet/parser/functions/has_key_spec.rb b/spec/unit/puppet/parser/functions/has_key_spec.rb
new file mode 100644
index 0000000..d1dcd15
--- /dev/null
+++ b/spec/unit/puppet/parser/functions/has_key_spec.rb
@@ -0,0 +1,46 @@
+require 'puppet'
+require 'mocha'
+describe Puppet::Parser::Functions.function(:has_key) do
+
+  # Pulled from Dan's create_resources function
+  # TODO - this should be moved to spec_helper since the
+  # logic is likely to be applied to multiple rspec files.
+  let(:compiler) {
+    topscope = Puppet::Parser::Scope.new
+    # This is necessary so we don't try to use the compiler to discover our parent.
+    topscope.parent = nil
+    my_scope = Puppet::Parser::Scope.new
+    my_scope.compiler = Puppet::Parser::Compiler.new(Puppet::Node.new("floppy", :environment => 'production'))
+    my_scope.parent = topscope
+    compiler = my_scope.compiler
+  }
+  let(:scope) {
+    scope = Puppet::Parser::Scope.new
+    scope.stubs(:environment).returns(Puppet::Node::Environment.new('production'))
+    scope
+  }
+
+  describe 'when calling has_key from puppet' do
+    it "should not compile when no arguments are passed" do
+      Puppet[:code] = 'has_key()'
+      expect { compiler.compile }.should raise_error(Puppet::ParseError, /wrong number of arguments/)
+    end
+    it "should not compile when 1 argument is passed" do
+      Puppet[:code] = "has_key('foo')"
+      expect { compiler.compile }.should raise_error(Puppet::ParseError, /wrong number of arguments/)
+    end
+    it "should require the first value to be a Hash" do
+      Puppet[:code] = "has_key('foo', 'bar')"
+      expect { compiler.compile }.should raise_error(Puppet::ParseError, /expects the first argument to be a hash/)
+    end
+  end
+  describe 'when calling the function has_key from a scope instance' do
+    it 'should detect existing keys' do
+      scope.function_has_key([{'one' => 1}, 'one']).should be_true
+    end
+    it 'should detect existing keys' do
+      scope.function_has_key([{'one' => 1}, 'two']).should be_false
+    end
+  end
+
+end