Add function to test for uniq values in hash

Author: Travis Fields

lib/puppet/parser/functions/sqlserver_validate_hash_uniq_values.rb

@@ -0,0 +1,21 @@
+# === Defined Parser Function: sqlserver_validate_hash_uniq_values
+#
+# [args*] A hash, that contains string or string[] for values
+#
+# @raise [Puppet::ParserError] When duplicates are found
+#
+module Puppet::Parser::Functions
+  newfunction(:sqlserver_validate_hash_uniq_values) do |arguments|
+
+    raise(Puppet::ParseError, 'Expect a Hash as an argument') unless arguments[0].is_a?(Hash)
+
+    value = arguments[0].each_value.collect { |v| v }.flatten
+
+    total_count = value.count
+    uniq_count = value.uniq.count
+    msg = arguments[1] ? arguments[1] : "Duplicate values passed to hash #{value}"
+    if uniq_count != total_count
+      raise(Puppet::ParseError, msg)
+    end
+  end
+end

spec/functions/sqlserver_validate_hash_uniq_values_spec.rb

@@ -0,0 +1,28 @@
+#! /usr/bin/env ruby -S rspec
+require 'spec_helper'
+
+describe "the sqlserver_validate_hash_uniq_values" do
+  let(:scope) { PuppetlabsSpec::PuppetInternals.scope }
+
+  it "should exist" do
+    expect(Puppet::Parser::Functions.function("sqlserver_validate_hash_uniq_values")).to eq("function_sqlserver_validate_hash_uniq_values")
+  end
+
+  it "should accept mixed value types of string and string[]" do
+    expect {
+      scope.function_sqlserver_validate_hash_uniq_values([{'test' => 'this', 'and' => ['test', 'this']}])
+    }.to raise_error
+  end
+
+  it "should pass validation" do
+    expect {
+      scope.function_sqlserver_validate_hash_uniq_values([{'test' => 'this', 'and' => ['test', 'another']}])
+    }.to_not raise_error
+  end
+
+  it "should require a hash" do
+    expect {
+      scope.function_sqlserver_validate_hash_uniq_values(["MyString"])
+    }.to raise_error
+  end
+end