Userlist test

dribdat/public/api.py

@@ -272,15 +272,23 @@ def project_info_json(project_id):
 # ------ USERS ----------
 
 
+def get_users_for_event(event=None):
+    if event is None: return []
+    userlist = []
+    for u in GetEventUsers(event):
+         # with_challenges=True, limit=-1, event=None
+        udata = u.data
+        udata['teams'] = u.joined_projects(True, -1, event)
+        userlist.append(udata)
+    return userlist
+
+
 @blueprint.route('/event/<int:event_id>/participants.csv')
 @admin_required
 def event_participants_csv(event_id):
     """Download a CSV of event participants."""
     event = Event.query.filter_by(id=event_id).first_or_404()
-    userlist = []
-    for u in GetEventUsers(event):
-        u['teams'] = u.joined_projects(True, -1, event)
-        userlist.append(u)
+    userlist = get_users_for_event(event)
     headers = {
         'Content-Disposition': 'attachment; '
         + 'filename=user_list_%d.csv' % event.id

tests/test_api.py

@@ -60,6 +60,12 @@ def test_get_event_data(self, testapp):
         ProjectActivity(project, 'star', user2)
         assert project in user1.joined_projects()
         assert project not in user1.joined_projects(False) # no challenges
+
+        # Test event API
+        userlist = get_users_for_event(event)
+        assert user1.username in [ u['username'] for u in userlist ]
+        assert user2.username in [ u['username'] for u in userlist ]
+
 
     def test_get_platform_data(self):
         """More global data types."""