Added type checks for dirname(), and additional tests

1 files changed, 14 insertions, 0 deletions

diff --git a/spec/functions/dirname_spec.rb b/spec/functions/dirname_spec.rb
index 8a3bcab..4261144 100755
--- a/spec/functions/dirname_spec.rb
+++ b/spec/functions/dirname_spec.rb
@@ -12,6 +12,10 @@ describe "the dirname function" do
     expect { scope.function_dirname([]) }.to( raise_error(Puppet::ParseError))
   end
 
+  it "should raise a ParseError if there is more than 1 argument" do
+    expect { scope.function_dirname(['a', 'b']) }.to( raise_error(Puppet::ParseError))
+  end
+
   it "should return dirname for an absolute path" do
     result = scope.function_dirname(['/path/to/a/file.ext'])
     expect(result).to(eq('/path/to/a'))
@@ -21,4 +25,14 @@ describe "the dirname function" do
     result = scope.function_dirname(['path/to/a/file.ext'])
     expect(result).to(eq('path/to/a'))
   end
+
+  it "should complain about hash argument" do
+    expect { scope.function_dirname([{}]) }.to( raise_error(Puppet::ParseError))
+  end
+  it "should complain about list argument" do
+    expect { scope.function_dirname([[]]) }.to( raise_error(Puppet::ParseError))
+  end
+  it "should complain about numeric argument" do
+    expect { scope.function_dirname([2112]) }.to( raise_error(Puppet::ParseError))
+  end
 end