e955c94ed3
Listing the "Blender Foundation" as copyright holder implied the Blender Foundation holds copyright to files which may include work from many developers. While keeping copyright on headers makes sense for isolated libraries, Blender's own code may be refactored or moved between files in a way that makes the per file copyright holders less meaningful. Copyright references to the "Blender Foundation" have been replaced with "Blender Authors", with the exception of `./extern/` since these this contains libraries which are more isolated, any changed to license headers there can be handled on a case-by-case basis. Some directories in `./intern/` have also been excluded: - `./intern/cycles/` it's own `AUTHORS` file is planned. - `./intern/opensubdiv/`. An "AUTHORS" file has been added, using the chromium projects authors file as a template. Design task: #110784 Ref !110783.
187 lines
6.5 KiB
Python
187 lines
6.5 KiB
Python
# SPDX-FileCopyrightText: 2019-2023 Blender Authors
|
|
#
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
# ./blender.bin --background -noaudio --python tests/python/bl_id_management.py -- --verbose
|
|
import bpy
|
|
import unittest
|
|
import random
|
|
|
|
|
|
class TestHelper:
|
|
|
|
@property
|
|
def data_container(self):
|
|
return getattr(bpy.data, self.data_container_id)
|
|
|
|
def clear_container(self):
|
|
bpy.data.batch_remove(self.data_container)
|
|
|
|
def add_to_container(self, name=""):
|
|
return self.data_container.new(name)
|
|
|
|
def remove_from_container(self, data=None, name=None, index=None):
|
|
data_container = self.data_container
|
|
if not data:
|
|
if name:
|
|
data = data_container[name]
|
|
elif index:
|
|
data = data_container[index]
|
|
self.assertTrue(data is not None)
|
|
data_container.remove(data)
|
|
|
|
def add_items_with_randomized_names(self, number_items=1, name_prefix=None, name_suffix=""):
|
|
if name_prefix is None:
|
|
name_prefix = self.default_name
|
|
for i in range(number_items):
|
|
self.add_to_container(name=name_prefix + str(random.random())[2:5] + name_suffix)
|
|
|
|
def ensure_proper_order(self):
|
|
id_prev = None
|
|
for id in self.data_container:
|
|
if id_prev:
|
|
self.assertTrue(id_prev.name < id.name or id.library)
|
|
id_prev = id
|
|
|
|
|
|
class TestIdAddNameManagement(TestHelper, unittest.TestCase):
|
|
data_container_id = 'meshes'
|
|
default_name = "Mesh"
|
|
|
|
def test_add_remove_single(self):
|
|
self.clear_container()
|
|
self.assertEqual(len(self.data_container), 0)
|
|
data = self.add_to_container()
|
|
self.assertEqual(len(self.data_container), 1)
|
|
self.ensure_proper_order()
|
|
self.remove_from_container(data=data)
|
|
self.assertEqual(len(self.data_container), 0)
|
|
|
|
def test_add_head_tail(self):
|
|
self.clear_container()
|
|
self.add_items_with_randomized_names(10)
|
|
self.assertEqual(len(self.data_container), 10)
|
|
self.ensure_proper_order()
|
|
|
|
name_head = "AAA" + self.default_name
|
|
data = self.add_to_container(name=name_head)
|
|
self.assertEqual(len(self.data_container), 11)
|
|
self.assertEqual(self.data_container[0].name, name_head)
|
|
self.ensure_proper_order()
|
|
|
|
name_tail = "ZZZ" + self.default_name
|
|
data = self.add_to_container(name=name_tail)
|
|
self.assertEqual(len(self.data_container), 12)
|
|
self.assertEqual(self.data_container[-1].name, name_tail)
|
|
self.ensure_proper_order()
|
|
|
|
def test_add_long_names(self):
|
|
self.clear_container()
|
|
self.add_items_with_randomized_names(10)
|
|
self.assertEqual(len(self.data_container), 10)
|
|
self.ensure_proper_order()
|
|
|
|
for i in range(12000):
|
|
self.add_to_container(name="ABCDEFGHIJKLMNOPQRSTUVWXYZ" * 3)
|
|
self.assertEqual(len(self.data_container), 12010)
|
|
self.ensure_proper_order()
|
|
|
|
self.clear_container()
|
|
self.add_items_with_randomized_names(100, name_prefix="", name_suffix="ABCDEFGHIJKLMNOPQRSTUVWXYZ" * 3)
|
|
self.assertEqual(len(self.data_container), 100)
|
|
self.ensure_proper_order()
|
|
|
|
def test_add_invalid_number_suffixes(self):
|
|
self.clear_container()
|
|
|
|
name = "%s.%.3d" % (self.default_name, 1000000000)
|
|
data = self.add_to_container(name=name)
|
|
self.assertEqual(data.name, name)
|
|
|
|
data = self.add_to_container(name=name)
|
|
self.assertEqual(data.name, self.default_name + ".001")
|
|
|
|
data = self.add_to_container(name=name)
|
|
self.assertEqual(data.name, self.default_name + ".002")
|
|
|
|
data = self.add_to_container(name=self.default_name)
|
|
self.assertEqual(data.name, self.default_name)
|
|
|
|
data = self.add_to_container(name="%s.%.3d" % (self.default_name, 0))
|
|
self.assertEqual(data.name, self.default_name + ".000")
|
|
|
|
data = self.add_to_container(name="%s.%.3d" % (self.default_name, 0))
|
|
self.assertEqual(data.name, self.default_name + ".003")
|
|
|
|
self.assertEqual(len(self.data_container), 6)
|
|
self.ensure_proper_order()
|
|
|
|
def test_add_use_smallest_free_number(self):
|
|
self.clear_container()
|
|
|
|
data = self.add_to_container(name="%s.%.3d" % (self.default_name, 2))
|
|
self.assertEqual(data.name, self.default_name + ".002")
|
|
|
|
data = self.add_to_container(name="%s.%.3d" % (self.default_name, 2))
|
|
self.assertEqual(data.name, self.default_name + ".001")
|
|
|
|
data = self.add_to_container(name="%s.%.3d" % (self.default_name, 2))
|
|
self.assertEqual(data.name, self.default_name + ".003")
|
|
|
|
for i in range(5, 1111):
|
|
name = "%s.%.3d" % (self.default_name, i)
|
|
data = self.add_to_container(name=name)
|
|
self.assertEqual(data.name, name)
|
|
|
|
for i in range(1112, 1200):
|
|
name = "%s.%.3d" % (self.default_name, i)
|
|
data = self.add_to_container(name=name)
|
|
self.assertEqual(data.name, name)
|
|
|
|
# Only slot available below 1024: 004.
|
|
data = self.add_to_container(name="%s.%.3d" % (self.default_name, 2))
|
|
self.assertEqual(data.name, self.default_name + ".004")
|
|
|
|
# Slot available at 1111 is not 'found' and we get first highest free number, 1200.
|
|
data = self.add_to_container(name="%s.%.3d" % (self.default_name, 2))
|
|
self.assertEqual(data.name, self.default_name + ".1200")
|
|
|
|
self.assertEqual(len(self.data_container), 1199)
|
|
self.ensure_proper_order()
|
|
|
|
|
|
class TestIdRename(TestHelper, unittest.TestCase):
|
|
data_container_id = 'meshes'
|
|
default_name = "Mesh"
|
|
|
|
def test_rename(self):
|
|
self.clear_container()
|
|
self.add_items_with_randomized_names(100)
|
|
self.ensure_proper_order()
|
|
|
|
data = self.data_container[0]
|
|
data.name = "ZZZ" + data.name
|
|
self.assertEqual(self.data_container[-1], data)
|
|
self.ensure_proper_order()
|
|
data.name = "AAA" + data.name
|
|
self.assertEqual(self.data_container[0], data)
|
|
self.ensure_proper_order()
|
|
|
|
name = "%s.%.3d" % (self.default_name, 1000000000)
|
|
data.name = name
|
|
self.assertEqual(data.name, name)
|
|
for dt in self.data_container:
|
|
if dt is not data:
|
|
data = dt
|
|
break
|
|
data.name = name
|
|
# This can fail currently, see #71244.
|
|
# ~ self.assertEqual(data.name, self.default_name + ".001")
|
|
self.ensure_proper_order()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
sys.argv = [__file__] + (sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else [])
|
|
unittest.main()
|