@@ -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 |
|
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