fixed tests

mitodl · Aug 21, 2015 · 13ca41c · 13ca41c
1 parent 0dd83a9
commit 13ca41c
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/learningresources/models.py b/learningresources/models.py
@@ -251,4 +251,3 @@ def get_preview_url(resource, org=None, course_number=None, run=None):
                 key=key,
             )
         )
-
diff --git a/rest/serializers.py b/rest/serializers.py
@@ -30,7 +30,8 @@
     LearningResource,
     StaticAsset,
     LearningResourceType,
-    STATIC_ASSET_BASEPATH
+    STATIC_ASSET_BASEPATH,
+    get_preview_url as resource_preview_url,
 )
 
 
@@ -244,7 +245,7 @@ def validate_terms(self, terms):
     @staticmethod
     def get_preview_url(obj):
         """Construct preview URL for LearningResource."""
-        return obj.get_preview_url()
+        return resource_preview_url(obj)
 
 
 class StaticAssetSerializer(ModelSerializer):

diff --git a/rest/tests/test_learning_resource.py b/rest/tests/test_learning_resource.py
@@ -25,6 +25,7 @@
     LearningResource,
     LearningResourceType,
     Repository,
+    get_preview_url,
 )
 from importer.tasks import import_file
 from taxonomy.models import Vocabulary
@@ -263,7 +264,7 @@ def test_preview_url(self):
         )
         self.assertEqual(
             expected_jump_to_id_url,
-            learning_resource.get_preview_url()
+            get_preview_url(learning_resource)
         )
 
         resource_dict = self.get_learning_resource(
@@ -275,7 +276,7 @@ def test_preview_url(self):
         self.assertEqual(
             "https://www.sandbox.edx.org/courses/"
             "test-org/infinity/Febtober/courseware",
-            learning_resource.get_preview_url()
+            get_preview_url(learning_resource)
         )
 
     def test_learning_resource_exports_invalid_methods(self):