Make sure that token `ensure_*` method always returns a token

This commit is contained in:
Grzegorz Bizon 2015-12-11 13:37:54 +00:00
parent 28ad40d974
commit 917effb737
2 changed files with 14 additions and 3 deletions

View File

@ -17,8 +17,12 @@ module TokenAuthenticatable
end
define_method("ensure_#{token_field}") do
write_attribute(token_field, generate_token_for(token_field)) if
read_attribute(token_field).blank?
current_token = read_attribute(token_field)
if current_token.blank?
write_attribute(token_field, generate_token_for(token_field))
else
current_token
end
end
define_method("reset_#{token_field}!") do

View File

@ -13,7 +13,7 @@ describe User, 'TokenAuthenticatable' do
let(:token_field) { :authentication_token }
it_behaves_like 'TokenAuthenticatable'
describe 'ensured authentication token' do
describe 'ensures authentication token' do
subject { create(:user).send(token_field) }
it { is_expected.to be_a String }
end
@ -29,6 +29,13 @@ describe ApplicationSetting, 'TokenAuthenticatable' do
context 'token is not generated yet' do
it { expect(token).to be nil }
describe 'ensured token' do
subject { described_class.new.send("ensure_#{token_field}") }
it { is_expected.to be_a String }
it { is_expected.to_not be_blank }
end
end
context 'token is generated' do