add method to save generated rspec

Author: h0tw1r3

README.md

@@ -45,7 +45,7 @@ There is no one single workflow that works, but this is what we use. In your spe
 it { dump_catalog }
 ```
 
-This will dump the full current catalog as rspec-code. Here is an example:
+This will dump the full current catalog as rspec-code to standard output. Here is an example:
 
 ```
 it {
@@ -76,6 +76,13 @@ it {
 }
 ```
 
+Alternatively, instead of dumping to standard output, use the following to save the generated rspec:
+
+```
+it { rspec_catalog('generated_rspec_catalog.rb') }
+```
+
+
 Now copy and paste the parts that you want to check in your manifest into your rspec-code and voila.
 ## Development
 

lib/puppet-catalog_rspec/rspec.rb

@@ -1,3 +1,17 @@
 def dump_catalog
   catalogue.to_rspec
 end
+
+def rspec_catalog(output = undef)
+  o_stdout = $stdout
+  if output
+    $stdout = File.open(output, 'w+')
+  else
+    $stdout = StringIO.new
+  end
+  catalogue.to_rspec
+  yield
+  $stdout.string
+ensure
+  $stdout = o_stdout
+end