tasks.rb in puppet-check-1.1.0

- old
+ new
@@ -1,14 +1,52 @@
-require 'rake'
+require 'rspec/core/rake_task'
 require 'rake/tasklib'
+require 'rake/task'
 require_relative '../puppet-check'
 
 # the rake interface for PuppetCheck
 class PuppetCheck::Tasks < ::Rake::TaskLib
   def initialize
-    desc 'Execute Puppet-Check file checks'
-    task 'puppetcheck:file' do
-      exit PuppetCheck.new.run(Dir.glob('*'))
+    desc 'Execute all Puppet-Check checks'
+    task :puppetcheck do
+      %w(puppetcheck:file puppetcheck:spec puppetcheck:beaker).each { |task| Rake::Task[task.to_sym].invoke }
+    end
+
+    namespace :puppetcheck do
+      desc 'Execute Puppet-Check file checks'
+      task :file do
+        exit_code = PuppetCheck.new.run(Dir.glob('*'))
+        # changes nothing if this task is run separately; aborts 'puppetcheck' task if there are errors here
+        exit exit_code if exit_code != 0
+      end
+
+      desc 'Execute RSpec and RSpec-Puppet tests'
+      RSpec::Core::RakeTask.new(:spec) do |task|
+        rspec_puppet_setup
+        # generate tasks for all recognized directories inside of spec directories
+        task.pattern = '**/{classes, defines, facter, functions, hosts, puppet, unit, types}/**/*_spec.rb'
+      end
+
+      desc 'Execute Beaker acceptance tests'
+      RSpec::Core::RakeTask.new(:beaker) do |task|
+        task.pattern = '**/acceptance'
+      end
+    end
+  end
+
+  # prepare the directories for rspec-puppet testing
+  def rspec_puppet_setup
+    # leave method immediately if there is no rspec-puppet installed
+    begin
+      require 'rspec-puppet/setup'
+    rescue LoadError
+      return
+    end
+
+    # executes rspec::puppet::setup in every module directory to ensure module spec directories are configured correctly
+    Dir.glob('**/spec').each do |specdir|
+      Dir.chdir(specdir + '/..')
+      RSpec::Puppet::Setup.run
     end
   end
 end
 
 PuppetCheck::Tasks.new