1 files changed, 57 insertions, 0 deletions
diff --git a/spec/unit/functions/match_spec.rb b/spec/unit/functions/match_spec.rb
new file mode 100644
index 000000000..f4e2e383b
--- /dev/null
+++ b/spec/unit/functions/match_spec.rb
@@ -0,0 +1,57 @@
+require 'spec_helper'
+require 'puppet/pops'
+require 'puppet/loaders'
+
+describe 'the match function' do
+
+  before(:all) do
+    loaders = Puppet::Pops::Loaders.new(Puppet::Node::Environment.create(:testing, []))
+    Puppet.push_context({:loaders => loaders}, "test-examples")
+  end
+
+  after(:all) do
+    Puppet::Pops::Loaders.clear
+    Puppet::pop_context()
+  end
+
+  let(:func) do
+    Puppet.lookup(:loaders).puppet_system_loader.load(:function, 'match')
+  end
+
+  let(:type_parser) { Puppet::Pops::Types::TypeParser.new }
+
+
+  it 'matches string and regular expression without captures' do
+    expect(func.call({}, 'abc123', /[a-z]+[1-9]+/)).to eql(['abc123'])
+  end
+
+  it 'matches string and regular expression with captures' do
+    expect(func.call({}, 'abc123', /([a-z]+)([1-9]+)/)).to eql(['abc123', 'abc', '123'])
+  end
+
+  it 'produces nil if match is not found' do
+    expect(func.call({}, 'abc123', /([x]+)([6]+)/)).to be_nil
+  end
+
+  [ 'Pattern[/([a-z]+)([1-9]+)/]',       # regexp
+    'Pattern["([a-z]+)([1-9]+)"]',       # string
+    'Regexp[/([a-z]+)([1-9]+)/]',        # regexp type
+    'Pattern[/x9/, /([a-z]+)([1-9]+)/]', # regexp, first found matches
+  ].each do |pattern|
+    it "matches string and type #{pattern} with captures" do
+      expect(func.call({}, 'abc123', type(pattern))).to eql(['abc123', 'abc', '123'])
+    end
+  end
+
+  it 'matches an array of strings and yields a map of the result' do
+    expect(func.call({}, ['abc123', '2a', 'xyz2'], /([a-z]+)[1-9]+/)).to eql([['abc123', 'abc'], nil, ['xyz2', 'xyz']])
+  end
+
+  it 'raises error if Regexp type without regexp is used' do
+    expect{func.call({}, 'abc123', type('Regexp'))}.to raise_error(ArgumentError, /Given Regexp Type has no regular expression/)
+  end
+
+  def type(s)
+    Puppet::Pops::Types::TypeParser.new.parse(s) 
+  end
+end