88class TestMailActivityDoneMethods (TransactionCase ):
99 def setUp (self ):
1010 super (TestMailActivityDoneMethods , self ).setUp ()
11-
1211 self .employee = self .env ["res.users" ].create (
1312 {
1413 "company_id" : self .env .ref ("base.main_company" ).id ,
1514 "name" : "Test User" ,
16- "login" : "testuser" ,
17- "groups_id" : [(6 , 0 , [self .env .ref ("base.group_user" ).id ])],
15+ "login" : "testuser@testuser.local" ,
16+ "email" : "testuser@testuser.local" ,
17+ "tz" : "Europe/Brussels" ,
18+ "groups_id" : [
19+ (
20+ 6 ,
21+ 0 ,
22+ [
23+ self .env .ref ("base.group_user" ).id ,
24+ self .env .ref ("base.group_partner_manager" ).id ,
25+ ],
26+ )
27+ ],
1828 }
1929 )
20- activity_type = self .env ["mail.activity.type" ].search (
21- [("name" , "=" , "Meeting" )], limit = 1
30+ self .partner = (
31+ self .env ["res.partner" ]
32+ .with_user (self .employee )
33+ .create ({"name" : "test partner" })
2234 )
23- self .act1 = self .env ["mail.activity" ].create (
24- {
25- "activity_type_id" : activity_type .id ,
26- "res_id" : self .env .ref ("base.res_partner_1" ).id ,
27- "res_model" : "res.partner" ,
28- "res_model_id" : self .env ["ir.model" ]._get ("res.partner" ).id ,
29- "user_id" : self .employee .id ,
30- "date_deadline" : date .today (),
31- }
35+ activity_type = self .env ["mail.activity.type" ].create (
36+ {"name" : "test activity type" }
37+ )
38+ self .act1 = (
39+ self .env ["mail.activity" ]
40+ .with_user (self .employee )
41+ .create (
42+ {
43+ "activity_type_id" : activity_type .id ,
44+ "res_id" : self .partner .id ,
45+ "res_model" : "res.partner" ,
46+ "res_model_id" : self .env ["ir.model" ]._get ("res.partner" ).id ,
47+ "user_id" : self .employee .id ,
48+ "date_deadline" : date .today (),
49+ }
50+ )
3251 )
52+ self .act2 = (
53+ self .env ["mail.activity" ]
54+ .with_user (self .employee )
55+ .create (
56+ {
57+ "activity_type_id" : activity_type .id ,
58+ "res_id" : self .partner .id ,
59+ "res_model" : "res.partner" ,
60+ "res_model_id" : self .env ["ir.model" ]._get ("res.partner" ).id ,
61+ "user_id" : self .employee .id ,
62+ "date_deadline" : date .today (),
63+ }
64+ )
65+ )
66+ self .activities = self .act1 + self .act2
67+
68+ def _set_activities_done (self ):
69+ self .activities ._action_done ()
70+ self .activities .flush_recordset ()
3371
3472 def test_mail_activity_done (self ):
3573 self .act1 ._action_done ()
@@ -39,20 +77,41 @@ def test_mail_activity_done(self):
3977 def test_systray_get_activities (self ):
4078 act_count = self .employee .with_user (self .employee ).systray_get_activities ()
4179 self .assertEqual (
42- len (act_count ), 1 , "Number of activities should be equal to one"
80+ act_count [0 ]["total_count" ],
81+ 2 ,
82+ "Number of activities should be equal to two" ,
83+ )
84+ self .act1 ._action_done ()
85+ act_count = self .employee .with_user (self .employee ).systray_get_activities ()
86+ self .assertEqual (
87+ act_count [0 ]["total_count" ],
88+ 1 ,
89+ "Number of activities should be equal to one" ,
4390 )
4491
4592 def test_read_progress_bar (self ):
46- res_partner = self .env ["res.partner" ].browse (self .act1 .res_model_id )
4793 params = {
4894 "domain" : [],
4995 "group_by" : "id" ,
5096 "progress_bar" : {"field" : "activity_state" },
5197 }
52- result = res_partner ._read_progress_bar (** params )
98+ result = self . partner ._read_progress_bar (** params )
5399 self .assertEqual (result [0 ]["__count" ], 1 )
54-
55- self .act1 ._action_done ()
56- self .assertEqual (self .act1 .state , "done" )
57- result = res_partner ._read_progress_bar (** params )
100+ self ._set_activities_done ()
101+ result = self .partner ._read_progress_bar (** params )
58102 self .assertEqual (len (result ), 0 )
103+
104+ def test_activity_state_search (self ):
105+ today_activities = (
106+ self .env ["res.partner" ]
107+ .with_user (self .employee )
108+ .search ([("activity_state" , "=" , "today" )])
109+ )
110+ self .assertEqual (len (today_activities ), 1 )
111+ self ._set_activities_done ()
112+ today_activities = (
113+ self .env ["res.partner" ]
114+ .with_user (self .employee )
115+ .search ([("activity_state" , "=" , "today" )])
116+ )
117+ self .assertEqual (len (today_activities ), 0 )
0 commit comments