Make test v3 compatible

django-cms · Nov 26, 2024 · a6e09c4 · a6e09c4
1 parent c96ff20
commit a6e09c4
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/test_html.py b/tests/test_html.py
@@ -7,7 +7,7 @@
 
 from djangocms_text import html, settings
 from djangocms_text.html import dynamic_href, dynamic_src, render_dynamic_attributes
-from tests.fixtures import TestFixture
+from tests.fixtures import DJANGO_CMS4, TestFixture
 
 
 class HtmlSanitizerAdditionalProtocolsTests:
@@ -176,9 +176,12 @@ class DjangoCMSPictureIntegrationTestCase(CMSTestCase):
     def setUp(self):
         super().setUp()
         self.home = self.create_homepage("home", "page.html", "en")
-        self.placeholder = (
-            self.home.pagecontent_set(manager="admin_manager").first().get_placeholders().get(slot="content")
-        )
+        if DJANGO_CMS4:
+            self.placeholder = (
+                self.home.pagecontent_set(manager="admin_manager").first().get_placeholders().get(slot="content")
+            )
+        else:
+            self.placeholder = self.home.get_placeholders("en").get(slot="content")
 
     def test_extract_images(self):
         with patch("tests.test_html.save_image") as mock_save_image: