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