##// END OF EJS Templates
Merged r9420 from trunk....
Jean-Philippe Lang -
r9302:d6d064f8750f
parent child
Show More
@@ -1,51 +1,51
1 1 # Redmine - project management software
2 2 # Copyright (C) 2006-2011 Jean-Philippe Lang
3 3 #
4 4 # This program is free software; you can redistribute it and/or
5 5 # modify it under the terms of the GNU General Public License
6 6 # as published by the Free Software Foundation; either version 2
7 7 # of the License, or (at your option) any later version.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU General Public License
15 15 # along with this program; if not, write to the Free Software
16 16 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17 17
18 18 class Token < ActiveRecord::Base
19 19 belongs_to :user
20 20 validates_uniqueness_of :value
21 21
22 22 before_create :delete_previous_tokens, :generate_new_token
23 23
24 24 @@validity_time = 1.day
25 25
26 26 def generate_new_token
27 27 self.value = Token.generate_token_value
28 28 end
29 29
30 30 # Return true if token has expired
31 31 def expired?
32 32 return Time.now > self.created_on + @@validity_time
33 33 end
34 34
35 35 # Delete all expired tokens
36 36 def self.destroy_expired
37 Token.delete_all ["action <> 'feeds' AND created_on < ?", Time.now - @@validity_time]
37 Token.delete_all ["action NOT IN (?) AND created_on < ?", ['feeds', 'api'], Time.now - @@validity_time]
38 38 end
39 39
40 40 private
41 41 def self.generate_token_value
42 42 Redmine::Utils.random_hex(20)
43 43 end
44 44
45 45 # Removes obsolete tokens (same user and action)
46 46 def delete_previous_tokens
47 47 if user
48 48 Token.delete_all(['user_id = ? AND action = ?', user.id, action])
49 49 end
50 50 end
51 51 end
@@ -1,38 +1,61
1 1 # Redmine - project management software
2 2 # Copyright (C) 2006-2011 Jean-Philippe Lang
3 3 #
4 4 # This program is free software; you can redistribute it and/or
5 5 # modify it under the terms of the GNU General Public License
6 6 # as published by the Free Software Foundation; either version 2
7 7 # of the License, or (at your option) any later version.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU General Public License
15 15 # along with this program; if not, write to the Free Software
16 16 # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17 17
18 18 require File.expand_path('../../test_helper', __FILE__)
19 19
20 20 class TokenTest < ActiveSupport::TestCase
21 21 fixtures :tokens
22 22
23 23 def test_create
24 24 token = Token.new
25 25 token.save
26 26 assert_equal 40, token.value.length
27 27 assert !token.expired?
28 28 end
29 29
30 30 def test_create_should_remove_existing_tokens
31 31 user = User.find(1)
32 32 t1 = Token.create(:user => user, :action => 'autologin')
33 33 t2 = Token.create(:user => user, :action => 'autologin')
34 34 assert_not_equal t1.value, t2.value
35 35 assert !Token.exists?(t1.id)
36 36 assert Token.exists?(t2.id)
37 37 end
38
39 def test_destroy_expired_should_not_destroy_feeds_and_api_tokens
40 Token.delete_all
41
42 Token.create!(:user_id => 1, :action => 'api', :created_on => 7.days.ago)
43 Token.create!(:user_id => 1, :action => 'feeds', :created_on => 7.days.ago)
44
45 assert_no_difference 'Token.count' do
46 assert_equal 0, Token.destroy_expired
47 end
48 end
49
50 def test_destroy_expired_should_destroy_expired_tokens
51 Token.delete_all
52
53 Token.create!(:user_id => 1, :action => 'autologin', :created_on => 7.days.ago)
54 Token.create!(:user_id => 2, :action => 'autologin', :created_on => 3.days.ago)
55 Token.create!(:user_id => 3, :action => 'autologin', :created_on => 1.hour.ago)
56
57 assert_difference 'Token.count', -2 do
58 assert_equal 2, Token.destroy_expired
59 end
60 end
38 61 end
General Comments 0
You need to be logged in to leave comments. Login now