Restore test using Validatable and support both scenarios

scambra · Apr 14, 2021 · c18b083 · c18b083
1 parent 8dc54b7
commit c18b083
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/lib/devise_invitable/models.rb b/lib/devise_invitable/models.rb
@@ -397,8 +397,10 @@ def after_invitation_accepted(*args, &blk)
           # lower + upper case, a digit and a symbol.
           # For more unusual rules, this method can be overridden.
           def random_password
+            length = respond_to?(:password_length) ? password_length : Devise.password_length
+
             prefix = 'aA1!'
-            prefix + Devise.friendly_token(Devise.password_length.last - prefix.length)
+            prefix + Devise.friendly_token(length.last - prefix.length)
           end
       end
     end

diff --git a/test/models/invitable_test.rb b/test/models/invitable_test.rb
@@ -1,6 +1,10 @@
 require 'test_helper'
 require 'model_tests_helper'
 
+class Validatable < User
+  devise :validatable, password_length: 10..20
+end
+
 class InvitableTest < ActiveSupport::TestCase
 
   def setup
@@ -766,4 +770,10 @@ def assert_callbacks_status(callback, user, fired)
     assert_empty user.errors
     assert_equal Devise.password_length.last, user.password.length
   end
+
+  test 'should set initial passsword using :validatable with custom password_length' do
+    user = Validatable.invite!(email: 'valid@email.com')
+    assert_empty user.errors
+    assert_equal Validatable.password_length.last, user.password.length
+  end
 end