From 806cf43908e7a9710bd77dcee256a19138bc16ff Mon Sep 17 00:00:00 2001
From: Agrendalath <piotr@surowiec.it>
Date: Sun, 26 Mar 2023 20:15:22 +0200
Subject: [PATCH] feat!: use `replace_urls` service instead of `runtime`
 property

BREAKING CHANGE: This breaks Maple compatibility.
---
 Changelog.md                         |  7 +++++++
 drag_and_drop_v2/__init__.py         |  2 +-
 drag_and_drop_v2/drag_and_drop_v2.py |  4 ++--
 tests/utils.py                       | 11 ++++++-----
 4 files changed, 16 insertions(+), 8 deletions(-)

diff --git a/Changelog.md b/Changelog.md
index 8cef1f5a5..c5b6133bf 100644
--- a/Changelog.md
+++ b/Changelog.md
@@ -1,6 +1,13 @@
 Drag and Drop XBlock changelog
 ==============================
 
+Version 3.2.0 (2023-04-20)
+---------------------------
+
+* Use the `replace_urls` service instead of a runtime property.
+
+  BREAKING CHANGE: This version is no longer compatible with Maple. 
+
 Version 3.1.3 (2023-04-15)
 --------------------------
 
diff --git a/drag_and_drop_v2/__init__.py b/drag_and_drop_v2/__init__.py
index 139c6c8f9..39dfa10aa 100644
--- a/drag_and_drop_v2/__init__.py
+++ b/drag_and_drop_v2/__init__.py
@@ -1,4 +1,4 @@
 """ Drag and Drop v2 XBlock """
 from .drag_and_drop_v2 import DragAndDropBlock
 
-__version__ = "3.1.3"
+__version__ = "3.2.0"
diff --git a/drag_and_drop_v2/drag_and_drop_v2.py b/drag_and_drop_v2/drag_and_drop_v2.py
index ef92630d3..4cf933921 100644
--- a/drag_and_drop_v2/drag_and_drop_v2.py
+++ b/drag_and_drop_v2/drag_and_drop_v2.py
@@ -1056,8 +1056,8 @@ def _expand_static_url(self, url):
         This method is unfortunately a bit hackish since XBlock does not provide a low-level API
         for this.
         """
-        if hasattr(self.runtime, 'replace_urls'):
-            url = self.runtime.replace_urls(u'"{}"'.format(url))[1:-1]
+        if replace_urls_service := self.runtime.service(self, 'replace_urls'):
+            url = replace_urls_service.replace_urls(f'"{url}"')[1:-1]
         elif hasattr(self.runtime, 'course_id'):
             # edX Studio uses a different runtime for 'studio_view' than 'student_view',
             # and the 'studio_view' runtime doesn't provide the replace_urls API.
diff --git a/tests/utils.py b/tests/utils.py
index 7f4945d09..55e844711 100644
--- a/tests/utils.py
+++ b/tests/utils.py
@@ -31,6 +31,12 @@ def make_block():
     field_data = KvsFieldData(key_store)
     runtime = WorkbenchRuntime()
     runtime.course_id = "dummy_course_id"
+    # noinspection PyProtectedMember
+    runtime._services['replace_urls'] = Mock(  # pylint: disable=protected-access
+        replace_urls=lambda html, static_replace_only=False: re.sub(
+            r'"/static/([^"]*)"', r'"/course/test-course/assets/\1"', html
+        )
+    )
     def_id = runtime.id_generator.create_definition(block_type)
     usage_id = runtime.id_generator.create_usage(def_id)
     scope_ids = ScopeIds('user', block_type, def_id, usage_id)
@@ -60,11 +66,6 @@ def patch_workbench(self):
             'workbench.runtime.WorkbenchRuntime.local_resource_url',
             lambda _, _block, path: '/expanded/url/to/drag_and_drop_v2/' + path
         )
-        self.apply_patch(
-            'workbench.runtime.WorkbenchRuntime.replace_urls',
-            lambda _, html: re.sub(r'"/static/([^"]*)"', r'"/course/test-course/assets/\1"', html),
-            create=True,
-        )
         self.apply_patch(
             'drag_and_drop_v2.drag_and_drop_v2.get_grading_ignore_decoys_waffle_flag',
             lambda: Mock(is_enabled=lambda _: False),