4040from openedx .core .djangoapps .enrollments .errors import CourseEnrollmentError
4141from openedx .core .djangoapps .enrollments .views import EnrollmentUserThrottle
4242from openedx .core .djangoapps .notifications .config .waffle import ENABLE_NOTIFICATIONS
43- from openedx .core .djangoapps .notifications .models import CourseNotificationPreference
4443from openedx .core .djangoapps .oauth_dispatch .jwt import create_jwt_for_user
4544from openedx .core .djangoapps .user_api .models import RetirementState , UserOrgTag , UserRetirementStatus
4645from openedx .core .djangolib .testing .utils import skip_unless_lms
@@ -56,20 +55,20 @@ class EnrollmentTestMixin:
5655 API_KEY = "i am a key"
5756
5857 def assert_enrollment_status (
59- self ,
60- course_id = None ,
61- username = None ,
62- expected_status = status .HTTP_200_OK ,
63- email_opt_in = None ,
64- as_server = False ,
65- mode = CourseMode .DEFAULT_MODE_SLUG ,
66- is_active = None ,
67- enrollment_attributes = None ,
68- min_mongo_calls = 0 ,
69- max_mongo_calls = 0 ,
70- linked_enterprise_customer = None ,
71- cohort = None ,
72- force_enrollment = False ,
58+ self ,
59+ course_id = None ,
60+ username = None ,
61+ expected_status = status .HTTP_200_OK ,
62+ email_opt_in = None ,
63+ as_server = False ,
64+ mode = CourseMode .DEFAULT_MODE_SLUG ,
65+ is_active = None ,
66+ enrollment_attributes = None ,
67+ min_mongo_calls = 0 ,
68+ max_mongo_calls = 0 ,
69+ linked_enterprise_customer = None ,
70+ cohort = None ,
71+ force_enrollment = False ,
7372 ):
7473 """
7574 Enroll in the course and verify the response's status code. If the expected status is 200, also validates
@@ -199,10 +198,6 @@ def setUp(self):
199198 password = self .PASSWORD ,
200199 )
201200 self .client .login (username = self .USERNAME , password = self .PASSWORD )
202- CourseNotificationPreference .objects .create (
203- user = self .user ,
204- course_id = self .course .id ,
205- )
206201
207202 @ddt .data (
208203 # Default (no course modes in the database)
@@ -982,11 +977,11 @@ def test_downgrade_enrollment_with_mode(self):
982977 assert course_mode == CourseMode .DEFAULT_MODE_SLUG
983978
984979 @ddt .data (
985- ((CourseMode .DEFAULT_MODE_SLUG , ), CourseMode .DEFAULT_MODE_SLUG ),
980+ ((CourseMode .DEFAULT_MODE_SLUG ,), CourseMode .DEFAULT_MODE_SLUG ),
986981 ((CourseMode .DEFAULT_MODE_SLUG , CourseMode .VERIFIED ), CourseMode .DEFAULT_MODE_SLUG ),
987982 ((CourseMode .DEFAULT_MODE_SLUG , CourseMode .VERIFIED ), CourseMode .VERIFIED ),
988- ((CourseMode .PROFESSIONAL , ), CourseMode .PROFESSIONAL ),
989- ((CourseMode .NO_ID_PROFESSIONAL_MODE , ), CourseMode .NO_ID_PROFESSIONAL_MODE ),
983+ ((CourseMode .PROFESSIONAL ,), CourseMode .PROFESSIONAL ),
984+ ((CourseMode .NO_ID_PROFESSIONAL_MODE ,), CourseMode .NO_ID_PROFESSIONAL_MODE ),
990985 ((CourseMode .VERIFIED , CourseMode .CREDIT_MODE ), CourseMode .VERIFIED ),
991986 ((CourseMode .VERIFIED , CourseMode .CREDIT_MODE ), CourseMode .CREDIT_MODE ),
992987 )
@@ -1274,7 +1269,7 @@ def test_enterprise_course_enrollment_with_ec_uuid(self, mock_enterprise_custome
12741269 username = 'enterprise_worker' ,
12751270 linked_enterprise_customer = 'this-is-a-real-uuid' ,
12761271 )
1277- assert httpretty .last_request ().path == '/consent/api/v1/data_sharing_consent' # pylint: disable=no-member
1272+ assert httpretty .last_request ().path == '/consent/api/v1/data_sharing_consent' # pylint: disable=no-member
12781273 assert httpretty .last_request ().method == httpretty .POST
12791274
12801275 def test_enrollment_attributes_always_written (self ):
@@ -1907,15 +1902,15 @@ def test_query_string_parameters_invalid_errors(self, query_params, error_fields
19071902
19081903 @ddt .data (
19091904 # Non-existent user
1910- ({'username' : 'nobody' }, ),
1911- ({'username' : 'nobody' , 'course_id' : 'e/d/X' }, ),
1905+ ({'username' : 'nobody' },),
1906+ ({'username' : 'nobody' , 'course_id' : 'e/d/X' },),
19121907
19131908 # Non-existent course
1914- ({'course_id' : 'a/b/c' }, ),
1915- ({'course_id' : 'a/b/c' , 'username' : 'student1' }, ),
1909+ ({'course_id' : 'a/b/c' },),
1910+ ({'course_id' : 'a/b/c' , 'username' : 'student1' },),
19161911
19171912 # Non-existent course and user
1918- ({'course_id' : 'a/b/c' , 'username' : 'dummy' }, )
1913+ ({'course_id' : 'a/b/c' , 'username' : 'dummy' },)
19191914 )
19201915 @ddt .unpack
19211916 def test_non_existent_course_user (self , query_params ):
0 commit comments