Add Chromium-only Blender WebEngine parity work
This commit is contained in:
mes123456
229
blender-5.2.0/tests/python/bl_blendfile_io.py
Normal file
229
blender-5.2.0/tests/python/bl_blendfile_io.py
Normal file
@@ -0,0 +1,229 @@
|
||||
# SPDX-FileCopyrightText: 2020-2023 Blender Authors
|
||||
#
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
|
||||
# ./blender.bin --background --python tests/python/bl_blendfile_io.py
|
||||
import bpy
|
||||
import os
|
||||
import sys
|
||||
|
||||
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
||||
from bl_blendfile_utils import TestHelper
|
||||
|
||||
|
||||
class TestBlendFileSaveLoadBasic(TestHelper):
|
||||
|
||||
def __init__(self, args):
|
||||
super().__init__(args)
|
||||
|
||||
def test_save_load(self):
|
||||
bpy.ops.wm.read_homefile(use_empty=True, use_factory_startup=True)
|
||||
|
||||
bpy.data.meshes.new("OrphanedMesh")
|
||||
|
||||
output_dir = self.args.output_dir
|
||||
self.ensure_path(output_dir)
|
||||
|
||||
# Take care to keep the name unique so multiple test jobs can run at once.
|
||||
output_path = os.path.join(output_dir, "blendfile_io.blend")
|
||||
|
||||
orig_data = self.blender_data_to_tuple(bpy.data, "orig_data 1")
|
||||
|
||||
bpy.ops.wm.save_as_mainfile(filepath=output_path, check_existing=False, compress=False)
|
||||
bpy.ops.wm.open_mainfile(filepath=output_path, load_ui=False)
|
||||
|
||||
read_data = self.blender_data_to_tuple(bpy.data, "read_data 1")
|
||||
|
||||
# We have orphaned data, which should be removed by file reading, so there should not be equality here.
|
||||
self.assertNotEqual(orig_data, read_data)
|
||||
|
||||
bpy.data.orphans_purge()
|
||||
|
||||
orig_data = self.blender_data_to_tuple(bpy.data, "orig_data 2")
|
||||
|
||||
bpy.ops.wm.save_as_mainfile(filepath=output_path, check_existing=False, compress=False)
|
||||
bpy.ops.wm.open_mainfile(filepath=output_path, load_ui=False)
|
||||
|
||||
read_data = self.blender_data_to_tuple(bpy.data, "read_data 2")
|
||||
|
||||
self.assertEqual(orig_data, read_data)
|
||||
|
||||
|
||||
class TestBlendFileSavePartial(TestHelper):
|
||||
OBJECT_MESH_NAME = "ObjectMesh"
|
||||
OBJECT_MATERIAL_NAME = "ObjectMaterial"
|
||||
OBJECT_NAME = "Object"
|
||||
UNUSED_MESH_NAME = "UnusedMesh"
|
||||
|
||||
def __init__(self, args):
|
||||
super().__init__(args)
|
||||
|
||||
def test_save_load(self):
|
||||
bpy.ops.wm.read_homefile(use_empty=True, use_factory_startup=True)
|
||||
|
||||
ob_mesh = bpy.data.meshes.new(self.OBJECT_MESH_NAME)
|
||||
ob_material = bpy.data.materials.new(self.OBJECT_MATERIAL_NAME)
|
||||
ob_mesh.materials.append(ob_material)
|
||||
ob = bpy.data.objects.new(self.OBJECT_NAME, object_data=ob_mesh)
|
||||
bpy.context.collection.objects.link(ob)
|
||||
|
||||
unused_mesh = bpy.data.meshes.new(self.UNUSED_MESH_NAME)
|
||||
unused_mesh.materials.append(ob_material)
|
||||
|
||||
self.assertEqual(ob_mesh.users, 1)
|
||||
self.assertEqual(ob_material.users, 2)
|
||||
self.assertEqual(ob.users, 1)
|
||||
self.assertEqual(unused_mesh.users, 0)
|
||||
|
||||
output_dir = self.args.output_dir
|
||||
self.ensure_path(output_dir)
|
||||
|
||||
# Take care to keep the name unique so multiple test jobs can run at once.
|
||||
output_path = os.path.join(output_dir, "blendfile_io_partial.blend")
|
||||
|
||||
bpy.data.libraries.write(filepath=output_path, datablocks={ob, unused_mesh}, fake_user=False)
|
||||
bpy.ops.wm.open_mainfile(filepath=output_path, load_ui=False)
|
||||
|
||||
self.assertIn(self.OBJECT_MESH_NAME, bpy.data.meshes)
|
||||
self.assertIn(self.OBJECT_MATERIAL_NAME, bpy.data.materials)
|
||||
self.assertIn(self.OBJECT_NAME, bpy.data.objects)
|
||||
self.assertIn(self.UNUSED_MESH_NAME, bpy.data.meshes)
|
||||
|
||||
self.assertEqual(bpy.data.meshes[self.OBJECT_MESH_NAME].users, 1)
|
||||
self.assertEqual(bpy.data.materials[self.OBJECT_MATERIAL_NAME].users, 2)
|
||||
self.assertEqual(bpy.data.objects[self.OBJECT_NAME].users, 0)
|
||||
self.assertEqual(bpy.data.meshes[self.UNUSED_MESH_NAME].users, 0)
|
||||
|
||||
|
||||
# NOTE: Technically this should rather be in `bl_id_management.py` test, but that file uses `unittest` module,
|
||||
# which makes mixing it with tests system used here and passing extra parameters complicated.
|
||||
# Since the main effect of 'RUNTIME' ID tag is on file save, it can as well be here for now.
|
||||
class TestIdRuntimeTag(TestHelper):
|
||||
|
||||
def __init__(self, args):
|
||||
super().__init__(args)
|
||||
|
||||
def unique_blendfile_name(self, base_name):
|
||||
return base_name + self.__class__.__name__ + ".blend"
|
||||
|
||||
def test_basics(self):
|
||||
output_dir = self.args.output_dir
|
||||
self.ensure_path(output_dir)
|
||||
bpy.ops.wm.read_homefile(use_empty=False, use_factory_startup=True)
|
||||
|
||||
obj = bpy.data.objects['Cube']
|
||||
self.assertFalse(obj.is_runtime_data)
|
||||
self.assertTrue(bpy.context.view_layer.depsgraph.ids['Cube'].is_runtime_data)
|
||||
|
||||
output_work_path = os.path.join(output_dir, self.unique_blendfile_name("blendfile"))
|
||||
bpy.ops.wm.save_as_mainfile(filepath=output_work_path, check_existing=False, compress=False)
|
||||
|
||||
bpy.ops.wm.open_mainfile(filepath=output_work_path, load_ui=False)
|
||||
obj = bpy.data.objects['Cube']
|
||||
self.assertFalse(obj.is_runtime_data)
|
||||
|
||||
obj.is_runtime_data = True
|
||||
self.assertTrue(obj.is_runtime_data)
|
||||
|
||||
bpy.ops.wm.save_as_mainfile(filepath=output_work_path, check_existing=False, compress=False)
|
||||
bpy.ops.wm.open_mainfile(filepath=output_work_path, load_ui=False)
|
||||
|
||||
self.assertNotIn('Cube', bpy.data.objects)
|
||||
mesh = bpy.data.meshes['Cube']
|
||||
self.assertFalse(mesh.is_runtime_data)
|
||||
self.assertEqual(mesh.users, 0)
|
||||
|
||||
def test_linking(self):
|
||||
output_dir = self.args.output_dir
|
||||
self.ensure_path(output_dir)
|
||||
bpy.ops.wm.read_homefile(use_empty=False, use_factory_startup=True)
|
||||
|
||||
material = bpy.data.materials.new("LibMaterial")
|
||||
# Use a dummy mesh as user of the material, such that the material is saved
|
||||
# without having to use fake user on it.
|
||||
mesh = bpy.data.meshes.new("LibMesh")
|
||||
mesh.materials.append(material)
|
||||
mesh.use_fake_user = True
|
||||
|
||||
output_lib_path = os.path.join(output_dir, self.unique_blendfile_name("blendlib_runtimetag_basic"))
|
||||
bpy.ops.wm.save_as_mainfile(filepath=output_lib_path, check_existing=False, compress=False)
|
||||
|
||||
bpy.ops.wm.read_homefile(use_empty=False, use_factory_startup=True)
|
||||
|
||||
obj = bpy.data.objects['Cube']
|
||||
self.assertFalse(obj.is_runtime_data)
|
||||
obj.is_runtime_data = True
|
||||
|
||||
link_dir = os.path.join(output_lib_path, "Material")
|
||||
bpy.ops.wm.link(directory=link_dir, filename="LibMaterial")
|
||||
|
||||
linked_material = bpy.data.materials['LibMaterial']
|
||||
self.assertFalse(linked_material.is_library_indirect)
|
||||
|
||||
link_dir = os.path.join(output_lib_path, "Mesh")
|
||||
bpy.ops.wm.link(directory=link_dir, filename="LibMesh", instance_object_data=False)
|
||||
|
||||
linked_mesh = bpy.data.meshes['LibMesh']
|
||||
self.assertFalse(linked_mesh.is_library_indirect)
|
||||
self.assertFalse(linked_mesh.use_fake_user)
|
||||
|
||||
obj.data = linked_mesh
|
||||
obj.material_slots[0].link = 'OBJECT'
|
||||
obj.material_slots[0].material = linked_material
|
||||
|
||||
output_work_path = os.path.join(output_dir, self.unique_blendfile_name("blendfile"))
|
||||
bpy.ops.wm.save_as_mainfile(filepath=output_work_path, check_existing=False, compress=False)
|
||||
|
||||
# Only usage of this linked material is a runtime ID (object),
|
||||
# so writing .blend file will have properly reset its tag to indirectly linked data.
|
||||
self.assertTrue(linked_material.is_library_indirect)
|
||||
|
||||
# Only usage of this linked mesh is a runtime ID (object),
|
||||
# so writing .blend file will have properly reset its tag to indirectly linked data.
|
||||
self.assertTrue(linked_mesh.is_library_indirect)
|
||||
|
||||
bpy.ops.wm.open_mainfile(filepath=output_work_path, load_ui=False)
|
||||
|
||||
self.assertNotIn('Cube', bpy.data.objects)
|
||||
self.assertNotIn('LibMaterial', bpy.data.materials)
|
||||
self.assertNotIn('LibMesh', bpy.data.meshes)
|
||||
|
||||
|
||||
TESTS = (
|
||||
TestBlendFileSaveLoadBasic,
|
||||
TestBlendFileSavePartial,
|
||||
|
||||
TestIdRuntimeTag,
|
||||
)
|
||||
|
||||
|
||||
def argparse_create():
|
||||
import argparse
|
||||
|
||||
# When --help or no args are given, print this help
|
||||
description = "Test basic IO of blend file."
|
||||
parser = argparse.ArgumentParser(description=description)
|
||||
parser.add_argument(
|
||||
"--output-dir",
|
||||
dest="output_dir",
|
||||
help="Where to output temp saved blendfiles",
|
||||
required=True,
|
||||
)
|
||||
|
||||
return parser
|
||||
|
||||
|
||||
def main():
|
||||
args = argparse_create().parse_args()
|
||||
|
||||
# Don't write thumbnails into the home directory.
|
||||
bpy.context.preferences.filepaths.file_preview_type = 'NONE'
|
||||
|
||||
for Test in TESTS:
|
||||
Test(args).run_all_tests()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
import sys
|
||||
sys.argv = [__file__] + (sys.argv[sys.argv.index("--") + 1:] if "--" in sys.argv else [])
|
||||
main()
|
||||
Reference in New Issue
Block a user