131 lines
4.8 KiB
Python
131 lines
4.8 KiB
Python
# ############################################################
|
|
# Importing - Same For All Render Layer Tests
|
|
# ############################################################
|
|
|
|
import unittest
|
|
import os
|
|
import sys
|
|
|
|
from render_layer_common import *
|
|
|
|
|
|
# ############################################################
|
|
# Testing
|
|
# ############################################################
|
|
|
|
class UnitTesting(RenderLayerTesting):
|
|
def test_operator_context(self):
|
|
"""
|
|
See if render layer context is properly set/get with operators overrides
|
|
when we set render_layer in context, the collection should change as well
|
|
"""
|
|
import bpy
|
|
import os
|
|
|
|
class SampleOperator(bpy.types.Operator):
|
|
bl_idname = "testing.sample"
|
|
bl_label = "Sample Operator"
|
|
|
|
render_layer = bpy.props.StringProperty(
|
|
default="Not Set",
|
|
options={'SKIP_SAVE'},
|
|
)
|
|
|
|
scene_collection = bpy.props.StringProperty(
|
|
default="",
|
|
options={'SKIP_SAVE'},
|
|
)
|
|
|
|
use_verbose = bpy.props.BoolProperty(
|
|
default=False,
|
|
options={'SKIP_SAVE'},
|
|
)
|
|
|
|
def execute(self, context):
|
|
render_layer = context.render_layer
|
|
ret = {'FINISHED'}
|
|
|
|
# this is simply playing safe
|
|
if render_layer.name != self.render_layer:
|
|
if self.use_verbose:
|
|
print('ERROR: Render Layer mismatch: "{0}" != "{1}"'.format(
|
|
render_layer.name, self.render_layer))
|
|
ret = {'CANCELLED'}
|
|
|
|
scene_collection_name = None
|
|
if self.scene_collection:
|
|
scene_collection_name = self.scene_collection
|
|
else:
|
|
scene_collection_name = render_layer.collections.active.name
|
|
|
|
# while this is the real test
|
|
if context.scene_collection.name != scene_collection_name:
|
|
if self.use_verbose:
|
|
print('ERROR: Scene Collection mismatch: "{0}" != "{1}"'.format(
|
|
context.scene_collection.name, scene_collection_name))
|
|
ret = {'CANCELLED'}
|
|
return ret
|
|
|
|
bpy.utils.register_class(SampleOperator)
|
|
|
|
# open sample file
|
|
ROOT = self.get_root()
|
|
filepath_layers = os.path.join(ROOT, 'layers.blend')
|
|
bpy.ops.wm.open_mainfile('EXEC_DEFAULT', filepath=filepath_layers)
|
|
self.rename_collections()
|
|
|
|
# change the file
|
|
three_b = bpy.data.objects.get('T.3b')
|
|
three_c = bpy.data.objects.get('T.3c')
|
|
scene = bpy.context.scene
|
|
subzero = scene.master_collection.collections['1'].collections.new('sub-zero')
|
|
scorpion = subzero.collections.new('scorpion')
|
|
subzero.objects.link(three_b)
|
|
scorpion.objects.link(three_c)
|
|
layer = scene.render_layers.new('Fresh new Layer')
|
|
layer.collections.unlink(layer.collections.active)
|
|
layer.collections.link(subzero)
|
|
layer.collections.active_index = 3
|
|
self.assertEqual(layer.collections.active.name, 'scorpion')
|
|
|
|
# Change active scene layer (do it for workspace too just to don't get mangled in workspace bugs)
|
|
scene = bpy.context.scene
|
|
scene.render_layers.active_index = len(scene.render_layers) - 2
|
|
self.assertEqual(scene.render_layers.active.name, "Viewport")
|
|
bpy.context.workspace.render_layer = bpy.context.scene.render_layers['Viewport']
|
|
|
|
# old layer
|
|
self.assertEqual(bpy.ops.testing.sample(render_layer='Viewport', use_verbose=True), {'FINISHED'})
|
|
|
|
# expected to fail
|
|
self.assertTrue(bpy.ops.testing.sample(render_layer=layer.name), {'CANCELLED'})
|
|
|
|
# set render layer and scene collection
|
|
override = bpy.context.copy()
|
|
override["render_layer"] = layer
|
|
override["scene_collection"] = subzero
|
|
self.assertEqual(bpy.ops.testing.sample(
|
|
override,
|
|
render_layer=layer.name,
|
|
scene_collection=subzero.name, # 'sub-zero'
|
|
use_verbose=True), {'FINISHED'})
|
|
|
|
# set only render layer
|
|
override = bpy.context.copy()
|
|
override["render_layer"] = layer
|
|
|
|
self.assertEqual(bpy.ops.testing.sample(
|
|
override,
|
|
render_layer=layer.name,
|
|
scene_collection=layer.collections.active.name, # 'scorpion'
|
|
use_verbose=True), {'FINISHED'})
|
|
|
|
|
|
# ############################################################
|
|
# Main - Same For All Render Layer Tests
|
|
# ############################################################
|
|
|
|
if __name__ == '__main__':
|
|
UnitTesting._extra_arguments = setup_extra_arguments(__file__)
|
|
unittest.main()
|