Fixes

mitodl · Dec 6, 2016 · 96a5015 · 96a5015
1 parent bba7fff
commit 96a5015
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/cms/serializers_test.py b/cms/serializers_test.py
@@ -52,7 +52,7 @@ def test_program_page_serializer(self):
         """
         Test program page serializer
         """
-        program = ProgramFactory.create(title="Supply Chain Management")
+        program = ProgramFactory.create(title="Supply Chain Management", course=None)
         course = CourseFactory.create(program=program, title="Learning How to Supply")
         page = ProgramPageFactory.create(program=program, title=program.title)
         faculty = FacultyFactory.create(

diff --git a/courses/factories.py b/courses/factories.py
@@ -7,7 +7,8 @@
 from factory import fuzzy
 from factory.django import DjangoModelFactory
 
-from .models import Program, Course, CourseRun
+from courses.models import Program, Course, CourseRun
+from ecommerce.models import CoursePrice
 
 FAKE = faker.Factory.create()
 
@@ -25,6 +26,7 @@ class ProgramFactory(DjangoModelFactory):
     def ecommerce(self, create, extracted, **kwargs):  # pylint: disable=unused-argument
         if not extracted:
             return self
+        from ecommerce.factories import CoursePriceFactory
         price = 0
         course_runs = CourseRun.objects.filter(course__program=self)
         for course_run in course_runs:
@@ -35,6 +37,7 @@ def ecommerce(self, create, extracted, **kwargs):  # pylint: disable=unused-argu
                 )
             price = course_price.price
         if self.financial_aid_availability:
+            from financialaid.factories import TierProgramFactory
             finaid_values = [
                 (0, 1000),
                 int(price / 10),

diff --git a/courses/models_test.py b/courses/models_test.py
@@ -45,7 +45,7 @@ class CourseModelTests(ESTestCase):
     @classmethod
     def setUpTestData(cls):
         super(CourseModelTests, cls).setUpTestData()
-        cls.course = CourseFactory.create(title="Title")
+        cls.course = CourseFactory.create(title="Title", course_run=None)
 
     def setUp(self):
         super(CourseModelTests, self).setUp()

diff --git a/dashboard/api_edx_cache_test.py b/dashboard/api_edx_cache_test.py
@@ -43,7 +43,7 @@ def setUpTestData(cls):
         cls.p1_course_run_keys = ['p1_course_run']
         cls.p2_course_run_keys = ['p2_course_run_1', 'p2_course_run_2']
         cls.p1_course_run = CourseRunFactory.create(edx_course_key=cls.p1_course_run_keys[0])
-        p2 = ProgramFactory.create(full=True)
+        p2 = ProgramFactory.create()
         first_course = p2.course_set.first()
         extra_course = CourseFactory.create(program=p2)
         cls.p2_course_run_1 = CourseRunFactory.create(course=first_course, edx_course_key=cls.p2_course_run_keys[0])