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