df47f9df95
* refactor: remove old profiling function * refactor: move mocks into separate directory * refactor: move recommendations code into a new dir * docs: docstring consistency and branding updates * docs: add ADR for core versus experimental code
89 lines
2.5 KiB
Python
89 lines
2.5 KiB
Python
"""
|
|
Various utilities used for creating test data
|
|
"""
|
|
|
|
import datetime
|
|
from random import choice, getrandbits, randint
|
|
from time import time
|
|
from uuid import uuid4
|
|
|
|
from common.djangoapps.course_modes.models import CourseMode
|
|
from common.djangoapps.course_modes.tests.factories import CourseModeFactory
|
|
from common.djangoapps.student.tests.factories import CourseEnrollmentFactory
|
|
from openedx.core.djangoapps.content.course_overviews.tests.factories import (
|
|
CourseOverviewFactory,
|
|
)
|
|
from xmodule.modulestore.tests.factories import CourseFactory
|
|
|
|
|
|
def random_string():
|
|
"""Test util for generating a random string"""
|
|
return str(uuid4())
|
|
|
|
|
|
def random_bool():
|
|
"""Test util for generating a random boolean"""
|
|
return bool(getrandbits(1))
|
|
|
|
|
|
def random_date(allow_null=False):
|
|
"""Test util for generating a random date, optionally blank"""
|
|
|
|
# If null allowed, return null half the time
|
|
if allow_null and random_bool():
|
|
return None
|
|
|
|
d = randint(1, int(time()))
|
|
return datetime.datetime.fromtimestamp(d, tz=datetime.timezone.utc)
|
|
|
|
|
|
def random_url(allow_null=False):
|
|
"""Test util for generating a random URL, optionally blank"""
|
|
|
|
# If null allowed, return null half the time
|
|
if allow_null and random_bool():
|
|
return None
|
|
|
|
random_uuid = uuid4()
|
|
return choice([f"{random_uuid}.example.com", f"example.com/{random_uuid}"])
|
|
|
|
|
|
def random_grade():
|
|
"""Return a random grade (0-100) with 2 decimal places of padding"""
|
|
return randint(0, 10000) / 100
|
|
|
|
|
|
def decimal_to_grade_format(decimal):
|
|
"""Util for matching serialized grade format, pads a decimal to 2 places"""
|
|
return "{:.2f}".format(decimal)
|
|
|
|
|
|
def datetime_to_django_format(datetime_obj):
|
|
"""Util for matching serialized Django datetime format for comparison"""
|
|
if datetime_obj:
|
|
return datetime_obj.strftime("%Y-%m-%dT%H:%M:%SZ")
|
|
|
|
|
|
def create_test_enrollment(user, course_mode=CourseMode.AUDIT):
|
|
"""Create a test user, course, course overview, and enrollment. Return the enrollment."""
|
|
course = CourseFactory(self_paced=True)
|
|
|
|
CourseModeFactory(
|
|
course_id=course.id,
|
|
mode_slug=course_mode,
|
|
)
|
|
|
|
course_overview = CourseOverviewFactory(id=course.id)
|
|
|
|
# extra info for exercising serializers
|
|
course_overview.certificate_available_date = random_date()
|
|
|
|
test_enrollment = CourseEnrollmentFactory(
|
|
course_id=course.id, mode=course_mode, user_id=user.id
|
|
)
|
|
|
|
test_enrollment.course_overview.marketing_url = random_url()
|
|
test_enrollment.course_overview.end = random_date()
|
|
|
|
return test_enrollment
|