@@ -47,6 +47,16 @@ def _create_learner(
4747 attempt_ratio_order = 0 ,
4848 videos_viewed = 0 ,
4949 enrollment_date = '2015-01-28' ,
50+ user_id = None ,
51+ language = None ,
52+ location = None ,
53+ year_of_birth = None ,
54+ level_of_education = None ,
55+ gender = None ,
56+ mailing_address = None ,
57+ city = None ,
58+ country = None ,
59+ goals = None ,
5060 ):
5161 """Create a single learner roster entry in the elasticsearch index."""
5262 body = {
@@ -61,6 +71,16 @@ def _create_learner(
6171 'attempt_ratio_order' : attempt_ratio_order ,
6272 'videos_viewed' : videos_viewed ,
6373 'enrollment_date' : enrollment_date ,
74+ "user_id" : user_id ,
75+ "language" : language ,
76+ "location" : location ,
77+ "year_of_birth" : year_of_birth ,
78+ "level_of_education" : level_of_education ,
79+ "gender" : gender ,
80+ "mailing_address" : mailing_address ,
81+ "city" : city ,
82+ "country" : country ,
83+ "goals" : goals ,
6484 }
6585
6686 # leave null fields from being stored in the index. Otherwise, they will have
@@ -112,13 +132,20 @@ class LearnerTests(VerifyCourseIdMixin, LearnerAPITestMixin, TestCaseWithAuthent
112132 @ddt .data (
113133 ('ed_xavier' , 'Edward Xavier' , 'edX/DemoX/Demo_Course' , 'honor' , ['has_potential' ], 'Team edX' ,
114134 43 , 3 , 6 , 0 , 8.4 , 2 , '2015-04-24' , '2015-08-05' ),
135+ ('ed_xavier' , 'Edward Xavier' , 'edX/DemoX/Demo_Course' , 'honor' , ['has_potential' ], 'Team edX' ,
136+ 43 , 3 , 6 , 0 , 8.4 , 2 , '2015-04-24' , '2015-08-05' ,
137+ 10 , 'French' , 'Berlin' , 1976 , 'Bachelors' , 'Male' , '123 Sesame St' , 'Springfield' , 'France' ,
138+ 'I\' ve always wanted to play the piano.' ),
115139 ('ed_xavier' , 'Edward Xavier' , 'edX/DemoX/Demo_Course' , 'verified' ),
116140 )
117141 @ddt .unpack
118142 def test_get_user (self , username , name , course_id , enrollment_mode , segments = None , cohort = None ,
119143 problems_attempted = None , problems_completed = None , videos_viewed = None ,
120144 discussion_contributions = None , problem_attempts_per_completed = None ,
121- attempt_ratio_order = None , enrollment_date = None , last_updated = None ):
145+ attempt_ratio_order = None , enrollment_date = None , last_updated = None ,
146+ user_id = None , language = None , location = None ,
147+ year_of_birth = None , level_of_education = None , gender = None ,
148+ mailing_address = None , city = None , country = None , goals = None ):
122149
123150 self .create_learners ([{
124151 "username" : username ,
@@ -134,6 +161,16 @@ def test_get_user(self, username, name, course_id, enrollment_mode, segments=Non
134161 "problem_attempts_per_completed" : problem_attempts_per_completed ,
135162 "attempt_ratio_order" : attempt_ratio_order ,
136163 "enrollment_date" : enrollment_date ,
164+ "user_id" : user_id ,
165+ "language" : language ,
166+ "location" : location ,
167+ "year_of_birth" : year_of_birth ,
168+ "level_of_education" : level_of_education ,
169+ "gender" : gender ,
170+ "mailing_address" : mailing_address ,
171+ "city" : city ,
172+ "country" : country ,
173+ "goals" : goals ,
137174 }])
138175 self .create_update_index (last_updated )
139176
@@ -157,6 +194,16 @@ def test_get_user(self, username, name, course_id, enrollment_mode, segments=Non
157194 },
158195 "enrollment_date" : enrollment_date ,
159196 "last_updated" : last_updated ,
197+ "user_id" : user_id ,
198+ "language" : language ,
199+ "location" : location ,
200+ "year_of_birth" : year_of_birth ,
201+ "level_of_education" : level_of_education ,
202+ "gender" : gender ,
203+ "mailing_address" : mailing_address ,
204+ "city" : city ,
205+ "country" : country ,
206+ "goals" : goals ,
160207 }
161208 self .assertDictEqual (expected , response .data )
162209
0 commit comments