@@ -49,7 +49,7 @@ def test_nodes_for_course_single(self):
49
49
50
50
self .create_mode ('verified' , 'Verified Certificate' )
51
51
modes = CourseMode .modes_for_course (self .course_id )
52
- mode = Mode (u'verified' , u'Verified Certificate' , 0 , '' , 'usd' )
52
+ mode = Mode (u'verified' , u'Verified Certificate' , 0 , '' , 'usd' , None )
53
53
self .assertEqual ([mode ], modes )
54
54
55
55
modes_dict = CourseMode .modes_for_course_dict (self .course_id )
@@ -61,8 +61,8 @@ def test_modes_for_course_multiple(self):
61
61
"""
62
62
Finding the modes when there's multiple modes
63
63
"""
64
- mode1 = Mode (u'honor' , u'Honor Code Certificate' , 0 , '' , 'usd' )
65
- mode2 = Mode (u'verified' , u'Verified Certificate' , 0 , '' , 'usd' )
64
+ mode1 = Mode (u'honor' , u'Honor Code Certificate' , 0 , '' , 'usd' , None )
65
+ mode2 = Mode (u'verified' , u'Verified Certificate' , 0 , '' , 'usd' , None )
66
66
set_modes = [mode1 , mode2 ]
67
67
for mode in set_modes :
68
68
self .create_mode (mode .slug , mode .name , mode .min_price , mode .suggested_prices )
@@ -81,9 +81,9 @@ def test_min_course_price_for_currency(self):
81
81
self .assertEqual (0 , CourseMode .min_course_price_for_currency (self .course_id , 'usd' ))
82
82
83
83
# create some modes
84
- mode1 = Mode (u'honor' , u'Honor Code Certificate' , 10 , '' , 'usd' )
85
- mode2 = Mode (u'verified' , u'Verified Certificate' , 20 , '' , 'usd' )
86
- mode3 = Mode (u'honor' , u'Honor Code Certificate' , 80 , '' , 'cny' )
84
+ mode1 = Mode (u'honor' , u'Honor Code Certificate' , 10 , '' , 'usd' , None )
85
+ mode2 = Mode (u'verified' , u'Verified Certificate' , 20 , '' , 'usd' , None )
86
+ mode3 = Mode (u'honor' , u'Honor Code Certificate' , 80 , '' , 'cny' , None )
87
87
set_modes = [mode1 , mode2 , mode3 ]
88
88
for mode in set_modes :
89
89
self .create_mode (mode .slug , mode .name , mode .min_price , mode .suggested_prices , mode .currency )
@@ -98,14 +98,15 @@ def test_modes_for_course_expired(self):
98
98
modes = CourseMode .modes_for_course (self .course_id )
99
99
self .assertEqual ([CourseMode .DEFAULT_MODE ], modes )
100
100
101
- mode1 = Mode (u'honor' , u'Honor Code Certificate' , 0 , '' , 'usd' )
101
+ mode1 = Mode (u'honor' , u'Honor Code Certificate' , 0 , '' , 'usd' , None )
102
102
self .create_mode (mode1 .slug , mode1 .name , mode1 .min_price , mode1 .suggested_prices )
103
103
modes = CourseMode .modes_for_course (self .course_id )
104
104
self .assertEqual ([mode1 ], modes )
105
105
106
- expired_mode .expiration_date = datetime .now (pytz .UTC ) + timedelta (days = 1 )
106
+ expiration_date = datetime .now (pytz .UTC ) + timedelta (days = 1 )
107
+ expired_mode .expiration_date = expiration_date
107
108
expired_mode .save ()
108
- expired_mode_value = Mode (u'verified' , u'Verified Certificate' , 0 , '' , 'usd' )
109
+ expired_mode_value = Mode (u'verified' , u'Verified Certificate' , 0 , '' , 'usd' , expiration_date . date () )
109
110
modes = CourseMode .modes_for_course (self .course_id )
110
111
self .assertEqual ([expired_mode_value , mode1 ], modes )
111
112
0 commit comments