def store_mau_metrics(site, overwrite=False):
"""
Save "snapshot" of MAU metrics
"""
today = datetime.utcnow()
# get site data
student_modules = get_student_modules_for_site(site)
site_mau = get_mau_from_student_modules(student_modules=student_modules,
year=today.year,
month=today.month)
# store site data
site_mau_obj, _created = SiteMauMetrics.save_metrics(
site=site,
date_for=today.date(),
data=dict(mau=site_mau.count()),
overwrite=overwrite)
course_mau_objects = []
for course_key in get_course_keys_for_site(site):
course_student_modules = student_modules.filter(course_id=course_key)
course_mau = get_mau_from_student_modules(
student_modules=course_student_modules,
year=today.year,
month=today.month)
course_mau_obj, _created = CourseMauMetrics.save_metrics(
site=site,
course_id=str(course_key),
date_for=today.date(),
data=dict(mau=course_mau.count()),
overwrite=overwrite)
course_mau_objects.append(course_mau_obj)
return dict(smo=site_mau_obj, cmos=course_mau_objects)
def test_list_method_all(self, monkeypatch, lm_test_data):
"""Partial test coverage to check we get all site users
Checks returned user ids against all user ids for the site
Checks top level keys
Does NOT check values in the `enrollments` key. This should be done as
follow up work
"""
us = lm_test_data['us']
them = lm_test_data['them']
our_courses = us['courses']
caller = make_caller(us['org'])
assert us['site'].domain != them['site'].domain
assert len(our_courses) > 1
response = self.make_request(request_path=self.base_request_path,
monkeypatch=monkeypatch,
site=us['site'],
caller=caller,
action='list')
assert response.status_code == status.HTTP_200_OK
assert is_response_paginated(response.data)
results = response.data['results']
# Check users
result_ids = [obj['id'] for obj in results]
# Get all enrolled users
course_keys = get_course_keys_for_site(site=us['site'])
users = users_enrolled_in_courses(course_keys)
user_ids = [user.id for user in users]
assert set(result_ids) == set(user_ids)
# Spot check the first record
top_keys = [
'id', 'username', 'email', 'fullname', 'is_active', 'date_joined',
'enrollments'
]
assert set(results[0].keys()) == set(top_keys)