From f49a736ff4023231483c7e535ca2a7f2869d641d Mon Sep 17 00:00:00 2001
From: Matheus Santos <MatheusSantos>
Date: Thu, 7 Apr 2022 14:32:21 +1000
Subject: Text Editor: Get/Set region text API

Add the ability to get/set the selected text.

**Calling the new methods:**

- `bpy.data.texts["Text"].region_as_string()`
- `bpy.data.texts["Text"].region_from_string("Replacement")`
---
 tests/python/CMakeLists.txt   |  5 ++++
 tests/python/bl_pyapi_text.py | 65 +++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 70 insertions(+)
 create mode 100644 tests/python/bl_pyapi_text.py

(limited to 'tests/python')

diff --git a/tests/python/CMakeLists.txt b/tests/python/CMakeLists.txt
index fabac659555..a0b2e6207bf 100644
--- a/tests/python/CMakeLists.txt
+++ b/tests/python/CMakeLists.txt
@@ -116,6 +116,11 @@ add_blender_test(
   --python ${CMAKE_CURRENT_LIST_DIR}/bl_pyapi_prop_array.py
 )
 
+add_blender_test(
+  script_pyapi_text
+  --python ${CMAKE_CURRENT_LIST_DIR}/bl_pyapi_text.py
+)
+
 # ------------------------------------------------------------------------------
 # DATA MANAGEMENT TESTS
 
diff --git a/tests/python/bl_pyapi_text.py b/tests/python/bl_pyapi_text.py
new file mode 100644
index 00000000000..67e07e7d907
--- /dev/null
+++ b/tests/python/bl_pyapi_text.py
@@ -0,0 +1,65 @@
+# SPDX-License-Identifier: Apache-2.0
+
+# ./blender.bin --background -noaudio --python tests/python/bl_pyapi_text.py -- --verbose
+import bpy
+import unittest
+
+
+class TestText(unittest.TestCase):
+
+    def setUp(self):
+        self.text = bpy.data.texts.new("test_text")
+
+    def tearDown(self):
+        bpy.data.texts.remove(self.text)
+        del self.text
+
+    def test_text_new(self):
+        self.assertEqual(len(bpy.data.texts), 1)
+        self.assertEqual(self.text.name, "test_text")
+        self.assertEqual(self.text.as_string(), "\n")
+
+    def test_text_clear(self):
+        self.text.clear()
+        self.assertEqual(self.text.as_string(), "\n")
+
+    def test_text_fill(self):
+        tmp_text = (
+            "Line 1: Test line 1\n"
+            "Line 2: test line 2\n"
+            "Line 3: test line 3"
+        )
+        self.text.write(tmp_text)
+        self.assertEqual(self.text.as_string(), tmp_text + "\n")
+
+    def test_text_region_as_string(self):
+        tmp_text = (
+            "Line 1: Test line 1\n"
+            "Line 2: test line 2\n"
+            "Line 3: test line 3"
+        )
+        self.text.write(tmp_text)
+        # Get string in the middle of the text.
+        self.assertEqual(self.text.region_as_string(((1, 0), (1, -1))), "Line 2: test line 2")
+        # Big range test.
+        self.assertEqual(self.text.region_as_string(((-10000, -10000), (10000, 10000))), tmp_text)
+
+    def test_text_region_from_string(self):
+        tmp_text = (
+            "Line 1: Test line 1\n"
+            "Line 2: test line 2\n"
+            "Line 3: test line 3"
+        )
+        self.text.write(tmp_text)
+        # Set string in the middle of the text.
+        self.text.region_from_string("line 2", ((1, 0), (1, -1)))
+        self.assertEqual(self.text.as_string(), tmp_text.replace("Line 2: test line 2", "line 2") + "\n")
+        # Large range test.
+        self.text.region_from_string("New Text", ((-10000, -10000), (10000, 10000)))
+        self.assertEqual(self.text.as_string(), "New Text\n")
+
+
+if __name__ == "__main__":
+    import sys
+    sys.argv = [__file__] + (sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else [])
+    unittest.main()
-- 
cgit v1.2.3