##// END OF EJS Templates
Don't remove watchers on permission change....
Jean-Philippe Lang -
r11070:5fea79504c74
parent child
Show More
@@ -1,124 +1,113
1 # Redmine - project management software
1 # Redmine - project management software
2 # Copyright (C) 2006-2013 Jean-Philippe Lang
2 # Copyright (C) 2006-2013 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 Member < ActiveRecord::Base
18 class Member < ActiveRecord::Base
19 belongs_to :user
19 belongs_to :user
20 belongs_to :principal, :foreign_key => 'user_id'
20 belongs_to :principal, :foreign_key => 'user_id'
21 has_many :member_roles, :dependent => :destroy
21 has_many :member_roles, :dependent => :destroy
22 has_many :roles, :through => :member_roles
22 has_many :roles, :through => :member_roles
23 belongs_to :project
23 belongs_to :project
24
24
25 validates_presence_of :principal, :project
25 validates_presence_of :principal, :project
26 validates_uniqueness_of :user_id, :scope => :project_id
26 validates_uniqueness_of :user_id, :scope => :project_id
27 validate :validate_role
27 validate :validate_role
28
28
29 before_destroy :set_issue_category_nil
29 before_destroy :set_issue_category_nil
30 after_destroy :unwatch_from_permission_change
31
30
32 def role
31 def role
33 end
32 end
34
33
35 def role=
34 def role=
36 end
35 end
37
36
38 def name
37 def name
39 self.user.name
38 self.user.name
40 end
39 end
41
40
42 alias :base_role_ids= :role_ids=
41 alias :base_role_ids= :role_ids=
43 def role_ids=(arg)
42 def role_ids=(arg)
44 ids = (arg || []).collect(&:to_i) - [0]
43 ids = (arg || []).collect(&:to_i) - [0]
45 # Keep inherited roles
44 # Keep inherited roles
46 ids += member_roles.select {|mr| !mr.inherited_from.nil?}.collect(&:role_id)
45 ids += member_roles.select {|mr| !mr.inherited_from.nil?}.collect(&:role_id)
47
46
48 new_role_ids = ids - role_ids
47 new_role_ids = ids - role_ids
49 # Add new roles
48 # Add new roles
50 new_role_ids.each {|id| member_roles << MemberRole.new(:role_id => id) }
49 new_role_ids.each {|id| member_roles << MemberRole.new(:role_id => id) }
51 # Remove roles (Rails' #role_ids= will not trigger MemberRole#on_destroy)
50 # Remove roles (Rails' #role_ids= will not trigger MemberRole#on_destroy)
52 member_roles_to_destroy = member_roles.select {|mr| !ids.include?(mr.role_id)}
51 member_roles_to_destroy = member_roles.select {|mr| !ids.include?(mr.role_id)}
53 if member_roles_to_destroy.any?
52 if member_roles_to_destroy.any?
54 member_roles_to_destroy.each(&:destroy)
53 member_roles_to_destroy.each(&:destroy)
55 unwatch_from_permission_change
56 end
54 end
57 end
55 end
58
56
59 def <=>(member)
57 def <=>(member)
60 a, b = roles.sort.first, member.roles.sort.first
58 a, b = roles.sort.first, member.roles.sort.first
61 if a == b
59 if a == b
62 if principal
60 if principal
63 principal <=> member.principal
61 principal <=> member.principal
64 else
62 else
65 1
63 1
66 end
64 end
67 elsif a
65 elsif a
68 a <=> b
66 a <=> b
69 else
67 else
70 1
68 1
71 end
69 end
72 end
70 end
73
71
74 def deletable?
72 def deletable?
75 member_roles.detect {|mr| mr.inherited_from}.nil?
73 member_roles.detect {|mr| mr.inherited_from}.nil?
76 end
74 end
77
75
78 def include?(user)
76 def include?(user)
79 if principal.is_a?(Group)
77 if principal.is_a?(Group)
80 !user.nil? && user.groups.include?(principal)
78 !user.nil? && user.groups.include?(principal)
81 else
79 else
82 self.user == user
80 self.user == user
83 end
81 end
84 end
82 end
85
83
86 def set_issue_category_nil
84 def set_issue_category_nil
87 if user
85 if user
88 # remove category based auto assignments for this member
86 # remove category based auto assignments for this member
89 IssueCategory.update_all "assigned_to_id = NULL", ["project_id = ? AND assigned_to_id = ?", project.id, user.id]
87 IssueCategory.update_all "assigned_to_id = NULL", ["project_id = ? AND assigned_to_id = ?", project.id, user.id]
90 end
88 end
91 end
89 end
92
90
93 # Find or initilize a Member with an id, attributes, and for a Principal
91 # Find or initilize a Member with an id, attributes, and for a Principal
94 def self.edit_membership(id, new_attributes, principal=nil)
92 def self.edit_membership(id, new_attributes, principal=nil)
95 @membership = id.present? ? Member.find(id) : Member.new(:principal => principal)
93 @membership = id.present? ? Member.find(id) : Member.new(:principal => principal)
96 @membership.attributes = new_attributes
94 @membership.attributes = new_attributes
97 @membership
95 @membership
98 end
96 end
99
97
100 # Finds or initilizes a Member for the given project and principal
98 # Finds or initilizes a Member for the given project and principal
101 def self.find_or_new(project, principal)
99 def self.find_or_new(project, principal)
102 project_id = project.is_a?(Project) ? project.id : project
100 project_id = project.is_a?(Project) ? project.id : project
103 principal_id = principal.is_a?(Principal) ? principal.id : principal
101 principal_id = principal.is_a?(Principal) ? principal.id : principal
104
102
105 member = Member.find_by_project_id_and_user_id(project_id, principal_id)
103 member = Member.find_by_project_id_and_user_id(project_id, principal_id)
106 member ||= Member.new(:project_id => project_id, :user_id => principal_id)
104 member ||= Member.new(:project_id => project_id, :user_id => principal_id)
107 member
105 member
108 end
106 end
109
107
110 protected
108 protected
111
109
112 def validate_role
110 def validate_role
113 errors.add_on_empty :role if member_roles.empty? && roles.empty?
111 errors.add_on_empty :role if member_roles.empty? && roles.empty?
114 end
112 end
115
116 private
117
118 # Unwatch things that the user is no longer allowed to view inside project
119 def unwatch_from_permission_change
120 if user
121 Watcher.prune(:user => user, :project => project)
122 end
123 end
124 end
113 end
@@ -1,74 +1,71
1 # Redmine - project management software
1 # Redmine - project management software
2 # Copyright (C) 2006-2013 Jean-Philippe Lang
2 # Copyright (C) 2006-2013 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 MemberRole < ActiveRecord::Base
18 class MemberRole < ActiveRecord::Base
19 belongs_to :member
19 belongs_to :member
20 belongs_to :role
20 belongs_to :role
21
21
22 after_destroy :remove_member_if_empty
22 after_destroy :remove_member_if_empty
23
23
24 after_create :add_role_to_group_users, :add_role_to_subprojects
24 after_create :add_role_to_group_users, :add_role_to_subprojects
25 after_destroy :remove_inherited_roles
25 after_destroy :remove_inherited_roles
26
26
27 validates_presence_of :role
27 validates_presence_of :role
28 validate :validate_role_member
28 validate :validate_role_member
29
29
30 def validate_role_member
30 def validate_role_member
31 errors.add :role_id, :invalid if role && !role.member?
31 errors.add :role_id, :invalid if role && !role.member?
32 end
32 end
33
33
34 def inherited?
34 def inherited?
35 !inherited_from.nil?
35 !inherited_from.nil?
36 end
36 end
37
37
38 private
38 private
39
39
40 def remove_member_if_empty
40 def remove_member_if_empty
41 if member.roles.empty?
41 if member.roles.empty?
42 member.destroy
42 member.destroy
43 end
43 end
44 end
44 end
45
45
46 def add_role_to_group_users
46 def add_role_to_group_users
47 if member.principal.is_a?(Group) && !inherited?
47 if member.principal.is_a?(Group) && !inherited?
48 member.principal.users.each do |user|
48 member.principal.users.each do |user|
49 user_member = Member.find_or_new(member.project_id, user.id)
49 user_member = Member.find_or_new(member.project_id, user.id)
50 user_member.member_roles << MemberRole.new(:role => role, :inherited_from => id)
50 user_member.member_roles << MemberRole.new(:role => role, :inherited_from => id)
51 user_member.save!
51 user_member.save!
52 end
52 end
53 end
53 end
54 end
54 end
55
55
56 def add_role_to_subprojects
56 def add_role_to_subprojects
57 member.project.children.each do |subproject|
57 member.project.children.each do |subproject|
58 if subproject.inherit_members?
58 if subproject.inherit_members?
59 child_member = Member.find_or_new(subproject.id, member.user_id)
59 child_member = Member.find_or_new(subproject.id, member.user_id)
60 child_member.member_roles << MemberRole.new(:role => role, :inherited_from => id)
60 child_member.member_roles << MemberRole.new(:role => role, :inherited_from => id)
61 child_member.save!
61 child_member.save!
62 end
62 end
63 end
63 end
64 end
64 end
65
65
66 def remove_inherited_roles
66 def remove_inherited_roles
67 MemberRole.where(:inherited_from => id).all.group_by(&:member).each do |member, member_roles|
67 MemberRole.where(:inherited_from => id).all.group_by(&:member).each do |member, member_roles|
68 member_roles.each(&:destroy)
68 member_roles.each(&:destroy)
69 if member && member.user
70 Watcher.prune(:user => member.user, :project => member.project)
71 end
72 end
69 end
73 end
70 end
74 end
71 end
@@ -1,191 +1,125
1 # Redmine - project management software
1 # Redmine - project management software
2 # Copyright (C) 2006-2013 Jean-Philippe Lang
2 # Copyright (C) 2006-2013 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 MemberTest < ActiveSupport::TestCase
20 class MemberTest < ActiveSupport::TestCase
21 fixtures :projects, :trackers, :issue_statuses, :issues,
21 fixtures :projects, :trackers, :issue_statuses, :issues,
22 :enumerations, :users, :issue_categories,
22 :enumerations, :users, :issue_categories,
23 :projects_trackers,
23 :projects_trackers,
24 :roles,
24 :roles,
25 :member_roles,
25 :member_roles,
26 :members,
26 :members,
27 :enabled_modules,
27 :enabled_modules,
28 :workflows,
28 :workflows,
29 :groups_users,
29 :groups_users,
30 :watchers,
30 :watchers,
31 :journals, :journal_details,
31 :journals, :journal_details,
32 :messages,
32 :messages,
33 :wikis, :wiki_pages, :wiki_contents, :wiki_content_versions,
33 :wikis, :wiki_pages, :wiki_contents, :wiki_content_versions,
34 :boards
34 :boards
35
35
36 include Redmine::I18n
36 include Redmine::I18n
37
37
38 def setup
38 def setup
39 @jsmith = Member.find(1)
39 @jsmith = Member.find(1)
40 end
40 end
41
41
42 def test_create
42 def test_create
43 member = Member.new(:project_id => 1, :user_id => 4, :role_ids => [1, 2])
43 member = Member.new(:project_id => 1, :user_id => 4, :role_ids => [1, 2])
44 assert member.save
44 assert member.save
45 member.reload
45 member.reload
46
46
47 assert_equal 2, member.roles.size
47 assert_equal 2, member.roles.size
48 assert_equal Role.find(1), member.roles.sort.first
48 assert_equal Role.find(1), member.roles.sort.first
49 end
49 end
50
50
51 def test_update
51 def test_update
52 assert_equal "eCookbook", @jsmith.project.name
52 assert_equal "eCookbook", @jsmith.project.name
53 assert_equal "Manager", @jsmith.roles.first.name
53 assert_equal "Manager", @jsmith.roles.first.name
54 assert_equal "jsmith", @jsmith.user.login
54 assert_equal "jsmith", @jsmith.user.login
55
55
56 @jsmith.mail_notification = !@jsmith.mail_notification
56 @jsmith.mail_notification = !@jsmith.mail_notification
57 assert @jsmith.save
57 assert @jsmith.save
58 end
58 end
59
59
60 def test_update_roles
60 def test_update_roles
61 assert_equal 1, @jsmith.roles.size
61 assert_equal 1, @jsmith.roles.size
62 @jsmith.role_ids = [1, 2]
62 @jsmith.role_ids = [1, 2]
63 assert @jsmith.save
63 assert @jsmith.save
64 assert_equal 2, @jsmith.reload.roles.size
64 assert_equal 2, @jsmith.reload.roles.size
65 end
65 end
66
66
67 def test_validate
67 def test_validate
68 member = Member.new(:project_id => 1, :user_id => 2, :role_ids => [2])
68 member = Member.new(:project_id => 1, :user_id => 2, :role_ids => [2])
69 # same use can't have more than one membership for a project
69 # same use can't have more than one membership for a project
70 assert !member.save
70 assert !member.save
71
71
72 # must have one role at least
72 # must have one role at least
73 user = User.new(:firstname => "new1", :lastname => "user1", :mail => "test_validate@somenet.foo")
73 user = User.new(:firstname => "new1", :lastname => "user1", :mail => "test_validate@somenet.foo")
74 user.login = "test_validate"
74 user.login = "test_validate"
75 user.password, user.password_confirmation = "password", "password"
75 user.password, user.password_confirmation = "password", "password"
76 assert user.save
76 assert user.save
77
77
78 set_language_if_valid 'fr'
78 set_language_if_valid 'fr'
79 member = Member.new(:project_id => 1, :user_id => user.id, :role_ids => [])
79 member = Member.new(:project_id => 1, :user_id => user.id, :role_ids => [])
80 assert !member.save
80 assert !member.save
81 assert_include I18n.translate('activerecord.errors.messages.empty'), member.errors[:role]
81 assert_include I18n.translate('activerecord.errors.messages.empty'), member.errors[:role]
82 str = "R\xc3\xb4le doit \xc3\xaatre renseign\xc3\xa9(e)"
82 str = "R\xc3\xb4le doit \xc3\xaatre renseign\xc3\xa9(e)"
83 str.force_encoding('UTF-8') if str.respond_to?(:force_encoding)
83 str.force_encoding('UTF-8') if str.respond_to?(:force_encoding)
84 assert_equal str, [member.errors.full_messages].flatten.join
84 assert_equal str, [member.errors.full_messages].flatten.join
85 end
85 end
86
86
87 def test_validate_member_role
87 def test_validate_member_role
88 user = User.new(:firstname => "new1", :lastname => "user1", :mail => "test_validate@somenet.foo")
88 user = User.new(:firstname => "new1", :lastname => "user1", :mail => "test_validate@somenet.foo")
89 user.login = "test_validate_member_role"
89 user.login = "test_validate_member_role"
90 user.password, user.password_confirmation = "password", "password"
90 user.password, user.password_confirmation = "password", "password"
91 assert user.save
91 assert user.save
92 member = Member.new(:project_id => 1, :user_id => user.id, :role_ids => [5])
92 member = Member.new(:project_id => 1, :user_id => user.id, :role_ids => [5])
93 assert !member.save
93 assert !member.save
94 end
94 end
95
95
96 def test_destroy
96 def test_destroy
97 category1 = IssueCategory.find(1)
97 category1 = IssueCategory.find(1)
98 assert_equal @jsmith.user.id, category1.assigned_to_id
98 assert_equal @jsmith.user.id, category1.assigned_to_id
99 assert_difference 'Member.count', -1 do
99 assert_difference 'Member.count', -1 do
100 assert_difference 'MemberRole.count', -1 do
100 assert_difference 'MemberRole.count', -1 do
101 @jsmith.destroy
101 @jsmith.destroy
102 end
102 end
103 end
103 end
104 assert_raise(ActiveRecord::RecordNotFound) { Member.find(@jsmith.id) }
104 assert_raise(ActiveRecord::RecordNotFound) { Member.find(@jsmith.id) }
105 category1.reload
105 category1.reload
106 assert_nil category1.assigned_to_id
106 assert_nil category1.assigned_to_id
107 end
107 end
108
108
109 def test_sort_without_roles
109 def test_sort_without_roles
110 a = Member.new(:roles => [Role.first])
110 a = Member.new(:roles => [Role.first])
111 b = Member.new
111 b = Member.new
112
112
113 assert_equal -1, a <=> b
113 assert_equal -1, a <=> b
114 assert_equal 1, b <=> a
114 assert_equal 1, b <=> a
115 end
115 end
116
116
117 def test_sort_without_principal
117 def test_sort_without_principal
118 role = Role.first
118 role = Role.first
119 a = Member.new(:roles => [role], :principal => User.first)
119 a = Member.new(:roles => [role], :principal => User.first)
120 b = Member.new(:roles => [role])
120 b = Member.new(:roles => [role])
121
121
122 assert_equal -1, a <=> b
122 assert_equal -1, a <=> b
123 assert_equal 1, b <=> a
123 assert_equal 1, b <=> a
124 end
124 end
125
126 context "removing permissions" do
127 setup do
128 Watcher.delete_all("user_id = 9")
129 user = User.find(9)
130 # public
131 Watcher.create!(:watchable => Issue.find(1), :user => user)
132 # private
133 Watcher.create!(:watchable => Issue.find(4), :user => user)
134 Watcher.create!(:watchable => Message.find(7), :user => user)
135 Watcher.create!(:watchable => Wiki.find(2), :user => user)
136 Watcher.create!(:watchable => WikiPage.find(3), :user => user)
137 end
138
139 context "of user" do
140 setup do
141 @member = Member.create!(:project => Project.find(2), :principal => User.find(9), :role_ids => [1, 2])
142 end
143
144 context "by deleting membership" do
145 should "prune watchers" do
146 assert_difference 'Watcher.count', -4 do
147 @member.destroy
148 end
149 end
150 end
151
152 context "by updating roles" do
153 should "prune watchers" do
154 Role.find(2).remove_permission! :view_wiki_pages
155 member = Member.first(:order => 'id desc')
156 assert_difference 'Watcher.count', -2 do
157 member.role_ids = [2]
158 member.save
159 end
160 assert !Message.find(7).watched_by?(@user)
161 end
162 end
163 end
164
165 context "of group" do
166 setup do
167 group = Group.find(10)
168 @member = Member.create!(:project => Project.find(2), :principal => group, :role_ids => [1, 2])
169 group.users << User.find(9)
170 end
171
172 context "by deleting membership" do
173 should "prune watchers" do
174 assert_difference 'Watcher.count', -4 do
175 @member.destroy
176 end
177 end
178 end
179
180 context "by updating roles" do
181 should "prune watchers" do
182 Role.find(2).remove_permission! :view_wiki_pages
183 assert_difference 'Watcher.count', -2 do
184 @member.role_ids = [2]
185 @member.save
186 end
187 end
188 end
189 end
190 end
191 end
125 end
General Comments 0
You need to be logged in to leave comments. Login now