##// END OF EJS Templates
Fixed: watchers selection lost when issue creation fails (#5406). #watched_by? was fixed in order to work with #watcher_user_ids= used in controllers on unsaved objects....
Jean-Philippe Lang -
r3591:a00d1eabd7c4
parent child
Show More
@@ -1,105 +1,111
1 1 # Redmine - project management software
2 2 # Copyright (C) 2006-2009 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.dirname(__FILE__) + '/../test_helper'
19 19
20 20 class WatcherTest < ActiveSupport::TestCase
21 21 fixtures :projects, :users, :members, :member_roles, :roles, :enabled_modules,
22 22 :issues,
23 23 :boards, :messages,
24 24 :wikis, :wiki_pages,
25 25 :watchers
26 26
27 27 def setup
28 28 @user = User.find(1)
29 29 @issue = Issue.find(1)
30 30 end
31 31
32 32 def test_watch
33 33 assert @issue.add_watcher(@user)
34 34 @issue.reload
35 35 assert @issue.watchers.detect { |w| w.user == @user }
36 36 end
37 37
38 38 def test_cant_watch_twice
39 39 assert @issue.add_watcher(@user)
40 40 assert !@issue.add_watcher(@user)
41 41 end
42 42
43 43 def test_watched_by
44 44 assert @issue.add_watcher(@user)
45 45 @issue.reload
46 46 assert @issue.watched_by?(@user)
47 47 assert Issue.watched_by(@user).include?(@issue)
48 48 end
49 49
50 def test_watcher_user_ids
51 issue = Issue.new
52 issue.watcher_user_ids = ['1', '3']
53 assert issue.watched_by?(User.find(1))
54 end
55
50 56 def test_recipients
51 57 @issue.watchers.delete_all
52 58 @issue.reload
53 59
54 60 assert @issue.watcher_recipients.empty?
55 61 assert @issue.add_watcher(@user)
56 62
57 63 @user.mail_notification = true
58 64 @user.save
59 65 @issue.reload
60 66 assert @issue.watcher_recipients.include?(@user.mail)
61 67
62 68 @user.mail_notification = false
63 69 @user.save
64 70 @issue.reload
65 71 assert @issue.watcher_recipients.include?(@user.mail)
66 72 end
67 73
68 74 def test_unwatch
69 75 assert @issue.add_watcher(@user)
70 76 @issue.reload
71 77 assert_equal 1, @issue.remove_watcher(@user)
72 78 end
73 79
74 80 def test_prune
75 81 Watcher.delete_all("user_id = 9")
76 82 user = User.find(9)
77 83
78 84 # public
79 85 Watcher.create!(:watchable => Issue.find(1), :user => user)
80 86 Watcher.create!(:watchable => Issue.find(2), :user => user)
81 87 Watcher.create!(:watchable => Message.find(1), :user => user)
82 88 Watcher.create!(:watchable => Wiki.find(1), :user => user)
83 89 Watcher.create!(:watchable => WikiPage.find(2), :user => user)
84 90
85 91 # private project (id: 2)
86 92 Member.create!(:project => Project.find(2), :principal => user, :role_ids => [1])
87 93 Watcher.create!(:watchable => Issue.find(4), :user => user)
88 94 Watcher.create!(:watchable => Message.find(7), :user => user)
89 95 Watcher.create!(:watchable => Wiki.find(2), :user => user)
90 96 Watcher.create!(:watchable => WikiPage.find(3), :user => user)
91 97
92 98 assert_no_difference 'Watcher.count' do
93 99 Watcher.prune(:user => User.find(9))
94 100 end
95 101
96 102 Member.delete_all
97 103
98 104 assert_difference 'Watcher.count', -4 do
99 105 Watcher.prune(:user => User.find(9))
100 106 end
101 107
102 108 assert Issue.find(1).watched_by?(user)
103 109 assert !Issue.find(4).watched_by?(user)
104 110 end
105 111 end
@@ -1,71 +1,71
1 1 # ActsAsWatchable
2 2 module Redmine
3 3 module Acts
4 4 module Watchable
5 5 def self.included(base)
6 6 base.extend ClassMethods
7 7 end
8 8
9 9 module ClassMethods
10 10 def acts_as_watchable(options = {})
11 11 return if self.included_modules.include?(Redmine::Acts::Watchable::InstanceMethods)
12 12 send :include, Redmine::Acts::Watchable::InstanceMethods
13 13
14 14 class_eval do
15 15 has_many :watchers, :as => :watchable, :dependent => :delete_all
16 16 has_many :watcher_users, :through => :watchers, :source => :user
17 17
18 18 named_scope :watched_by, lambda { |user_id|
19 19 { :include => :watchers,
20 20 :conditions => ["#{Watcher.table_name}.user_id = ?", user_id] }
21 21 }
22 22 attr_protected :watcher_ids, :watcher_user_ids
23 23 end
24 24 end
25 25 end
26 26
27 27 module InstanceMethods
28 28 def self.included(base)
29 29 base.extend ClassMethods
30 30 end
31 31
32 32 # Returns an array of users that are proposed as watchers
33 33 def addable_watcher_users
34 34 self.project.users.sort - self.watcher_users
35 35 end
36 36
37 37 # Adds user as a watcher
38 38 def add_watcher(user)
39 39 self.watchers << Watcher.new(:user => user)
40 40 end
41 41
42 42 # Removes user from the watchers list
43 43 def remove_watcher(user)
44 44 return nil unless user && user.is_a?(User)
45 45 Watcher.delete_all "watchable_type = '#{self.class}' AND watchable_id = #{self.id} AND user_id = #{user.id}"
46 46 end
47 47
48 48 # Adds/removes watcher
49 49 def set_watcher(user, watching=true)
50 50 watching ? add_watcher(user) : remove_watcher(user)
51 51 end
52 52
53 # Returns true if object is watched by user
53 # Returns true if object is watched by +user+
54 54 def watched_by?(user)
55 !!(user && self.watchers.detect {|w| w.user_id == user.id })
55 !!(user && self.watcher_user_ids.detect {|uid| uid == user.id })
56 56 end
57 57
58 58 # Returns an array of watchers' email addresses
59 59 def watcher_recipients
60 60 notified = watchers.collect(&:user).select(&:active?)
61 61 if respond_to?(:visible?)
62 62 notified.reject! {|user| !visible?(user)}
63 63 end
64 64 notified.collect(&:mail).compact
65 65 end
66 66
67 67 module ClassMethods; end
68 68 end
69 69 end
70 70 end
71 71 end
General Comments 0
You need to be logged in to leave comments. Login now