lib/sniff/rake_tasks.rb in sniff-0.6.8 vs lib/sniff/rake_tasks.rb in sniff-0.7.0

@@ -1,12 +1,14 @@
 require 'rake'
 require 'rake/clean'
-require 'rake/rdoctask'
+require 'rdoc/task'
 require 'rake/tasklib'
 
 module Sniff
   class RakeTasks
+    include Rake::DSL
+
     def self.define_tasks(&blk)
       new(&blk).define_tasks
     end
 
     attr_accessor :earth_domains, :cucumber, :rspec, :rcov, :rocco, :bueller
@@ -145,10 +147,10 @@
       unless test_tasks.empty?
         task :test => test_tasks
         task :default => :test
       end
 
-      Rake::RDocTask.new do |rdoc|
+      RDoc::Task.new do |rdoc|
         version = File.exist?('VERSION') ? File.read('VERSION') : ""
 
         rdoc.rdoc_dir = 'rdoc'
         rdoc.title = "#{gemname} #{version}"
         rdoc.rdoc_files.include('README*')