From 1e84ad99418656066d89a5c51bf94f7665bd24e5 Mon Sep 17 00:00:00 2001 From: Calen Pennington Date: Fri, 10 Jul 2015 13:19:00 -0400 Subject: [PATCH] Make CcxFactory fill in a coach if one isn't supplied --- lms/djangoapps/ccx/tests/factories.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/lms/djangoapps/ccx/tests/factories.py b/lms/djangoapps/ccx/tests/factories.py index 36c976970f..507df6b44a 100644 --- a/lms/djangoapps/ccx/tests/factories.py +++ b/lms/djangoapps/ccx/tests/factories.py @@ -1,7 +1,9 @@ """ Dummy factories for tests """ +from factory import SubFactory from factory.django import DjangoModelFactory +from student.tests.factories import UserFactory from ccx.models import CustomCourseForEdX # pylint: disable=import-error from ccx.models import CcxMembership # pylint: disable=import-error from ccx.models import CcxFutureMembership # pylint: disable=import-error @@ -11,6 +13,7 @@ class CcxFactory(DjangoModelFactory): # pylint: disable=missing-docstring FACTORY_FOR = CustomCourseForEdX display_name = "Test CCX" id = None # pylint: disable=redefined-builtin, invalid-name + coach = SubFactory(UserFactory) class CcxMembershipFactory(DjangoModelFactory): # pylint: disable=missing-docstring