def get(self, scene_id): """ Retrieve all shots that comes from given scene. """ scene = shots_service.get_scene(scene_id) user_service.check_project_access(scene["project_id"]) return scenes_service.get_shots_by_scene(scene_id)
def test_remove_shot_from_scene(self): scenes_service.add_shot_to_scene(self.scene, self.shot_01) scenes_service.add_shot_to_scene(self.scene, self.shot_02) scenes_service.remove_shot_from_scene(self.scene, self.shot_01) shots = scenes_service.get_shots_by_scene(self.scene["id"]) self.assertEqual(len(shots), 1) self.assertEqual(shots[0]["id"], self.shot_02["id"])