From 0591d92f53e91f9aec88e561634da9e437f793ad Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rafael=20Mendon=C3=A7a=20Fran=C3=A7a?=
 <rafaelmfranca@gmail.com>
Date: Sat, 29 Apr 2017 10:12:28 -0700
Subject: [PATCH] Move the version check to the lib folder

Closes #4514.
Fixes #4513.
---
 lib/devise.rb                 | 4 ++++
 test/rails_app/config/boot.rb | 4 ----
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/lib/devise.rb b/lib/devise.rb
index 2262c9b8..1b7e1df1 100755
--- a/lib/devise.rb
+++ b/lib/devise.rb
@@ -290,6 +290,10 @@ module Devise
   mattr_accessor :token_generator
   @@token_generator = nil
 
+  def self.rails51? # :nodoc:
+    Rails.gem_version >= Gem::Version.new("5.1.x")
+  end
+
   # Default way to set up Devise. Run rails generate devise_install to create
   # a fresh initializer with all configuration values.
   def self.setup
diff --git a/test/rails_app/config/boot.rb b/test/rails_app/config/boot.rb
index 5831994a..a7ab3335 100644
--- a/test/rails_app/config/boot.rb
+++ b/test/rails_app/config/boot.rb
@@ -4,10 +4,6 @@ end
 
 module Devise
   # Detection for minor differences between Rails 4 and 5, and 5.1 in tests.
-  def self.rails51?
-    Rails.version.start_with? '5.1'
-  end
-
   def self.rails5?
     Rails.version.start_with? '5'
   end