spec/spec_helper.rb in activity_notification-0.0.9 vs spec/spec_helper.rb in activity_notification-0.0.10

@@ -3,18 +3,24 @@
 require 'bundler/setup'
 Bundler.setup
 
 require 'simplecov'
 require 'coveralls'
+require 'rails'
 Coveralls.wear!
 SimpleCov.formatter = SimpleCov::Formatter::MultiFormatter.new [
   SimpleCov::Formatter::HTMLFormatter,
   Coveralls::SimpleCov::Formatter
 ]
 SimpleCov.start('rails') do
   add_filter '/spec/'
   add_filter '/lib/generators/templates/'
   add_filter '/lib/activity_notification/version.rb'
+  if Rails::VERSION::MAJOR == 5
+    nocov_token 'skip-rails5'
+  elsif Rails::VERSION::MAJOR == 4
+    nocov_token 'skip-rails4'
+  end
 end
 
 # Testing with Devise
 require 'devise'
 # Dummy application
\ No newline at end of file