def test_is_asset_ready(self):
self.generate_fixture_department()
self.generate_fixture_task_type()
self.generate_fixture_task_status()
self.generate_fixture_person()
self.generate_fixture_assigner()
self.task_type_compositing = tasks_service.get_or_create_task_type(
self.department_animation.serialize(),
"compositing",
color="#FFFFFF",
short_name="compo",
for_entity="Shot",
)
self.task_type_layout_id = str(self.task_type_layout.id)
self.task_type_animation_id = str(self.task_type_animation.id)
self.task_type_compositing_id = self.task_type_compositing["id"]
projects_service.create_project_task_type_link(
self.project_id, self.task_type_layout_id, 1
)
projects_service.create_project_task_type_link(
self.project_id, self.task_type_animation_id, 2
)
projects_service.create_project_task_type_link(
self.project_id, self.task_type_compositing_id, 3
)
self.task_layout = self.generate_fixture_shot_task(
task_type_id=self.task_type_layout_id
)
self.task_animation = self.generate_fixture_shot_task(
task_type_id=self.task_type_animation_id
)
self.task_compositing = self.generate_fixture_shot_task(
task_type_id=self.task_type_compositing_id
)
priority_map = breakdown_service._get_task_type_priority_map(
self.project_id
)
self.assertEqual(priority_map[self.task_type_layout_id], 1)
self.assertEqual(priority_map[self.task_type_animation_id], 2)
self.assertEqual(priority_map[self.task_type_compositing_id], 3)
asset = {"ready_for": str(self.task_type_animation.id)}
self.assertTrue(
breakdown_service._is_asset_ready(
asset, self.task_layout, priority_map
)
)
self.assertTrue(
breakdown_service._is_asset_ready(
asset, self.task_animation, priority_map
)
)
self.assertFalse(
breakdown_service._is_asset_ready(
asset, self.task_compositing, priority_map
)
)
self.shot_id = str(self.shot.id)
self.sequence_id = str(self.sequence.id)
self.asset_id = str(self.asset.id)
self.asset_character_id = str(self.asset_character.id)
new_casting = [
{"asset_id": self.asset_id, "nb_occurences": 1},
{"asset_id": self.asset_character_id, "nb_occurences": 3},
]
breakdown_service.update_casting(self.shot_id, new_casting)
asset = assets_service.get_asset_raw(self.asset_id)
asset.update({"ready_for": self.task_type_animation_id})
char = assets_service.get_asset_raw(self.asset_character_id)
char.update({"ready_for": self.task_type_compositing_id})
breakdown_service.refresh_casting_stats(asset.serialize())
self.task_layout = tasks_service.get_task(self.task_layout.id)
self.task_animation = tasks_service.get_task(self.task_animation.id)
self.task_compositing = tasks_service.get_task(
self.task_compositing.id
)
self.assertEqual(self.task_layout["nb_assets_ready"], 2)
self.assertEqual(self.task_animation["nb_assets_ready"], 2)
self.assertEqual(self.task_compositing["nb_assets_ready"], 1)
def get_tasks_for_asset(asset_id):
"""
Get all tasks for given asset.
"""
asset = assets_service.get_asset_raw(asset_id)
return get_task_dicts_for_entity(asset.id)
def get_tasks_for_asset(asset_id):
asset = assets_service.get_asset_raw(asset_id)
return get_task_dicts_for_entity(asset.id)
