Merge pull request #144 from doc75/return_regexp

add capability to check return code of function using regexp
rodjek · Jan 25, 2014 · a3161d3 · a3161d3
2 parents dd635cd + ac08115
commit a3161d3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/lib/rspec-puppet/matchers/run.rb b/lib/rspec-puppet/matchers/run.rb
@@ -30,7 +30,12 @@ def matches?(func_obj)
         else
           unless @expected_return.nil?
             @actual_return = @func.call
-            @actual_return == @expected_return
+            case @expected_return
+            when Regexp
+              @actual_return =~ @expected_return
+            else
+              @actual_return == @expected_return
+            end
           else
             begin
               @func.call

diff --git a/spec/functions/regsubst_spec.rb b/spec/functions/regsubst_spec.rb
@@ -0,0 +1,9 @@
+require 'spec_helper'
+
+describe 'regsubst' do
+  # used to test the fact that expected result can be a regexp
+  it { should run.with_params('thisisatest', '^192', '254').and_return(/sat/) }
+  it { should run.with_params('thisisatest', 'sat', 'xyz').and_return(/ixyze/) }
+  it { should run.with_params('thisisatest', 'sat', 'xyz').and_return('thisixyzest') }
+  it { should run.with_params('thisisatest', 'sat', 'xyz').and_return(/^thisixyzest$/) }
+end