@@ -16,38 +16,133 @@ def setUp(self):
1616 with requests_mock .Mocker () as m :
1717 requires = {
1818 "current_user" : [
19- "add_favorite_course" ,
20- "add_favorite_group" ,
2119 "get_by_id" ,
2220 ]
2321 }
2422 register_uris (requires , m )
2523
2624 self .user = self .canvas .get_current_user ()
27- self .favorite_course = self .user .add_favorite_course (1 )
28- self .favorite_group = self .user .add_favorite_group (1 )
2925
3026 # __str__()
31- def test__str__ (self , m ):
32- string = str (self .favorite_course )
27+ def test_uncapitalized___str__ (self , m ):
28+
29+ register_uris (
30+ {
31+ "current_user" : [
32+ "add_favorite_course" ,
33+ "add_favorite_group" ,
34+ ]
35+ },
36+ m ,
37+ )
38+
39+ favorite_course = self .user .add_favorite_course (1 )
40+ favorite_group = self .user .add_favorite_group (1 )
41+
42+ string = str (favorite_course )
43+ self .assertIsInstance (string , str )
44+
45+ string = str (favorite_group )
46+ self .assertIsInstance (string , str )
47+
48+ def test_capitalized___str__ (self , m ):
49+
50+ register_uris (
51+ {
52+ "current_user" : [
53+ "add_favorite_course_cap_context_type" ,
54+ "add_favorite_group_cap_context_type" ,
55+ ]
56+ },
57+ m ,
58+ )
59+
60+ favorite_course = self .user .add_favorite_course (1 )
61+ favorite_group = self .user .add_favorite_group (1 )
62+
63+ string = str (favorite_course )
3364 self .assertIsInstance (string , str )
3465
35- string = str (self . favorite_group )
66+ string = str (favorite_group )
3667 self .assertIsInstance (string , str )
3768
3869 # remove()
39- def test_remove_favorite_course (self , m ):
40- register_uris ({"current_user" : ["remove_favorite_course" ]}, m )
70+ def test_remove_uncapitalized_favorite_course (self , m ):
71+
72+ register_uris (
73+ {
74+ "current_user" : [
75+ "add_favorite_course" ,
76+ "remove_favorite_course" ,
77+ ]
78+ },
79+ m ,
80+ )
81+
82+ favorite_course = self .user .add_favorite_course (1 )
83+
84+ evnt = favorite_course .remove ()
4185
42- evnt = self .favorite_course .remove ()
4386 self .assertIsInstance (evnt , Favorite )
4487 self .assertEqual (evnt .context_type , "course" )
4588 self .assertEqual (evnt .context_id , 1 )
4689
47- def test_remove_favorite_group (self , m ):
48- register_uris ({"current_user" : ["remove_favorite_group" ]}, m )
90+ def test_remove_uncapitalized_favorite_group (self , m ):
91+
92+ register_uris (
93+ {
94+ "current_user" : [
95+ "add_favorite_group" ,
96+ "remove_favorite_group" ,
97+ ]
98+ },
99+ m ,
100+ )
101+
102+ favorite_group = self .user .add_favorite_group (1 )
103+
104+ evnt = favorite_group .remove ()
49105
50- evnt = self .favorite_group .remove ()
51106 self .assertIsInstance (evnt , Favorite )
52107 self .assertEqual (evnt .context_type , "group" )
53108 self .assertEqual (evnt .context_id , 1 )
109+
110+ def test_remove_capitalized_favorite_course (self , m ):
111+
112+ register_uris (
113+ {
114+ "current_user" : [
115+ "add_favorite_course_cap_context_type" ,
116+ "remove_favorite_course_cap_context_type" ,
117+ ]
118+ },
119+ m ,
120+ )
121+
122+ favorite_course = self .user .add_favorite_course (1 )
123+
124+ evnt = favorite_course .remove ()
125+
126+ self .assertIsInstance (evnt , Favorite )
127+ self .assertEqual (evnt .context_type , "Course" )
128+ self .assertEqual (evnt .context_id , 1 )
129+
130+ def test_remove_capitalized_favorite_group (self , m ):
131+
132+ register_uris (
133+ {
134+ "current_user" : [
135+ "add_favorite_group_cap_context_type" ,
136+ "remove_favorite_group_cap_context_type" ,
137+ ]
138+ },
139+ m ,
140+ )
141+
142+ favorite_group = self .user .add_favorite_group (1 )
143+
144+ evnt = favorite_group .remove ()
145+
146+ self .assertIsInstance (evnt , Favorite )
147+ self .assertEqual (evnt .context_type , "Group" )
148+ self .assertEqual (evnt .context_id , 1 )
0 commit comments