1 files changed, 122 insertions, 0 deletions
diff --git a/vendor/plugins/rake_tasks/tasks/spec.rake b/vendor/plugins/rake_tasks/tasks/spec.rake
new file mode 100644
index 000000000..9f07b0100
--- /dev/null
+++ b/vendor/plugins/rake_tasks/tasks/spec.rake
@@ -0,0 +1,122 @@
+namespace :spec do
+  unless defined?(RAILS_ROOT)
+    root_path = File.join(File.dirname(__FILE__), '..')
+    unless RUBY_PLATFORM =~ /mswin32/
+      require 'pathname'
+      root_path = Pathname.new(root_path).cleanpath(true).to_s
+    end
+    RAILS_ROOT = root_path
+  end
+
+  require "#{RAILS_ROOT}/config/environment"
+
+  PROJ_DIR = "#{RAILS_ROOT}/app/"
+  
+  def view_ext
+    if ActionView::Base.const_defined?('DEFAULT_TEMPLATE_HANDLER_PREFERENCE') &&
+       ActionView::Base::DEFAULT_TEMPLATE_HANDLER_PREFERENCE.include?(:erb) then
+      return ".html.erb"
+    else
+      return ".rhtml"
+    end
+  end
+  
+  def find_untested_ruby_files
+    files = {}
+    `find #{PROJ_DIR} -name '*.rb'`.split(/\n/).each do |file|
+      spec_file = file.sub('/app/', '/spec/').sub('.rb', '_spec.rb')
+      type = (File.dirname(file).split("/").last == "models" ? "model" : "controller")
+      File.exists?(spec_file) ? next : files[spec_file] = type
+    end
+    files
+  end
+
+  def find_untested_view_files
+    files = {}
+    `find #{PROJ_DIR} -name '*#{view_ext}'`.split(/\n/).each do |file|
+      spec_file = file.sub('/app/', '/spec/').sub(view_ext, "#{view_ext}_spec.rb")
+      type = File.dirname(file).split("/").last
+      File.exists?(spec_file) ? next : files[spec_file] = type
+    end
+    files
+  end
+
+  desc "Check files in the app directory for corresponding test files in the spec directory."
+  task :check do
+    files = find_untested_ruby_files.merge(find_untested_view_files)
+    unless files.empty?
+      puts "Missing test files:"
+      files.each {|file, type|  puts "  #{file}"}
+      puts
+      puts "Run the following command(s) to create the missing files:"
+      puts "  rake spec:sync"
+    end
+  end
+
+  desc "Check for missing test files in the spec directory and create them if they don't exist."
+  task :sync do
+    # Check if an application_controller file exists... hopefully it does not.
+    has_application_controller = File.exists?("#{PROJ_DIR}/controllers/application_controller.rb")
+    
+    files = find_untested_ruby_files
+    unless files.empty?
+      files.each do |file, type|
+        # Get rid of the _spec and file extension
+        name = File.basename(file).sub("_spec.rb", "").sub(/(_controller|_helper)/, "")
+
+        has_controller = has_helper = false
+        if type == "controller"
+          has_controller = File.exists?("#{PROJ_DIR}/controllers/#{name}_controller.rb")
+          has_helper = File.exists?("#{PROJ_DIR}/helpers/#{name}_helper.rb")
+        end
+
+        options = "--skip"
+        options += " --skip-migration" if type == "model"
+        puts `script/generate rspec_#{type} #{options} #{name} | grep create`
+        
+        unless has_controller
+          FileUtils.rm "#{PROJ_DIR}/controllers/#{name}_controller.rb" if File.exists?("#{PROJ_DIR}/controllers/#{name}_controller.rb")
+          FileUtils.rm "#{PROJ_DIR}/../spec/controllers/#{name}_controller_spec.rb" if File.exists?("#{PROJ_DIR}/../spec/controllers/#{name}_controller_spec.rb")
+        end
+
+        unless has_helper
+          FileUtils.rm "#{PROJ_DIR}/helpers/#{name}_helper.rb" if File.exists?("#{PROJ_DIR}/helpers/#{name}_helper.rb")
+          FileUtils.rm "#{PROJ_DIR}/../spec/helpers/#{name}_helper_spec.rb" if File.exists?("#{PROJ_DIR}/../spec/helpers/#{name}_helper_spec.rb")
+        end        
+      end
+    end
+
+    files = find_untested_view_files
+    unless files.empty?
+      action_list = {}
+      files.each do |file, controller|
+        action = File.basename(file)[0..-9].sub(view_ext, "")
+
+        if action_list[controller].blank?
+          action_list[controller] = action
+        else
+          action_list[controller] = "#{action_list[controller]} #{action}"
+        end
+      end
+      
+      action_list.each do |controller, actions|
+        next if actions.blank?
+
+        has_controller = File.exists?("#{PROJ_DIR}/controllers/#{controller}_controller.rb")
+        has_helper = File.exists?("#{PROJ_DIR}/helpers/#{controller}_helper.rb")
+
+        puts `script/generate rspec_controller --skip #{controller} #{actions} | grep create`
+        
+        unless has_controller
+          FileUtils.rm "#{PROJ_DIR}/controllers/#{controller}_controller.rb"
+          FileUtils.rm "#{PROJ_DIR}/../spec/controllers/#{controller}_controller_spec.rb"
+        end
+
+        unless has_helper
+          FileUtils.rm "#{PROJ_DIR}/helpers/#{controller}_helper.rb"
+          FileUtils.rm "#{PROJ_DIR}/../spec/helpers/#{controller}_helper_spec.rb"
+        end        
+      end
+    end
+  end
+end