require rails.

Author: ryanfox1985

lib/annotate/annotate_routes/header_generator.rb

@@ -1,3 +1,4 @@
+require 'activerecord'
 require_relative './helpers'
 
 module AnnotateRoutes
@@ -16,7 +17,7 @@ def generate(options = {})
       private
 
       def routes_map(options)
-        command = Rails.version.first.to_i > 5 ? `rails routes` : `rake routes`
+        command = ActiveRecord.version.to_s.first.to_i > 5 ? `rails routes` : `rake routes`
         result = command.chomp("\n").split(/\n/, -1)
 
         # In old versions of Rake, the first line of output was the cwd.  Not so

spec/spec_helper.rb

@@ -18,6 +18,7 @@
 require 'bundler'
 Bundler.setup
 
+require 'activerecord'
 require 'rake'
 require 'rspec'