diff --git a/representative_image/representative_image.py b/representative_image/representative_image.py index 7d5d61c..3d91d36 100644 --- a/representative_image/representative_image.py +++ b/representative_image/representative_image.py @@ -25,7 +25,7 @@ def images_extraction(instance): representativeImage = imageTag['src'] # Set the attribute to content instance - instance.repImage = representativeImage + instance.featured_image = representativeImage def register(): signals.content_object_init.connect(images_extraction) diff --git a/representative_image/test_representative_image.py b/representative_image/test_representative_image.py index 60efa8b..4e65c5d 100644 --- a/representative_image/test_representative_image.py +++ b/representative_image/test_representative_image.py @@ -29,7 +29,7 @@ class TestRepresentativeImage(unittest.TestCase): } article = Article(**args) - self.assertEqual(article.repImage, TEST_CONTENT_IMAGE_URL) + self.assertEqual(article.featured_image, TEST_CONTENT_IMAGE_URL) def test_extract_image_from_summary(self): args = { @@ -40,7 +40,7 @@ class TestRepresentativeImage(unittest.TestCase): } article = Article(**args) - self.assertEqual(article.repImage, TEST_SUMMARY_IMAGE_URL) + self.assertEqual(article.featured_image, TEST_SUMMARY_IMAGE_URL) self.assertEqual(article.summary, TEST_SUMMARY_WITHOUTIMAGE) if __name__ == '__main__':