a86576e69b
Run isort -rc lms/djangoapps/program_enrollments Run pylint lms/djangoapps/program_enrollment and fix messages. Stop pylint from complaining about DictFactoryBase instances
23 lines
573 B
Python
23 lines
573 B
Python
"""
|
|
Sharable utilities for testing program enrollments
|
|
"""
|
|
|
|
from factory import LazyAttributeSequence, SubFactory
|
|
from factory.django import DjangoModelFactory
|
|
from social_django.models import UserSocialAuth
|
|
|
|
from student.tests.factories import UserFactory
|
|
|
|
|
|
class UserSocialAuthFactory(DjangoModelFactory):
|
|
"""
|
|
Factory for UserSocialAuth records.
|
|
"""
|
|
class Meta(object):
|
|
model = UserSocialAuth
|
|
user = SubFactory(UserFactory)
|
|
uid = LazyAttributeSequence(lambda o, n: '%s:%d' % (o.slug, n))
|
|
|
|
class Params(object):
|
|
slug = 'gatech'
|