Setup test task in Rails 5 style

This bug was causing all rake tasks to fail in Rails 5 when draper gem was present Fixes drapergem#681
audionerd · Oct 8, 2015 · dd24576 · dd24576
1 parent 57a5141
commit dd24576
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/draper/tasks/test.rake b/lib/draper/tasks/test.rake
@@ -1,6 +1,6 @@
 require 'rake/testtask'
 
-test_task = if Rails.version.to_f < 3.2
+test_task = if Rails.version.to_f < 3.2 || Rails.version.start_with?('5')
   require 'rails/test_unit/railtie'
   Rake::TestTask
 else