def find_asset(self, dbo):
"""Test find_asset, CRUD method
:param dbObject dbo: dbo
"""
self.reset(dbo)
# 1. find find_asset from db
with orm.db_session:
temp_asset, _ = Asset.find_asset_by_id(self.asset.id)
# 2. test value
self.assert_value(temp_asset)
temp_asset, err = Asset.find_asset_by_id(-1)
# 3. assert on error
self.assertEqual(err, "Asset Not Found !")
self.assertEqual(temp_asset, None)
# 4. find_all asset from db
temp_assets = Asset.find_all_assets()
# 5. test value
self.assertEqual(len(temp_assets), 1)
self.assert_value(temp_assets[0])
def seed_tasks():
# get shot and asset (create 2 task)
_shot = Shot.find_all_shots()[0]
_asset = Asset.find_all_assets()[0]
_task_shot = Task.create_task(name="shot_task_test", shot=_shot)
# create a needed task
_neededTask = Task.create_task("needed_task", asset=_asset)
_neededTask2 = Task.create_task("needed_task2", asset=_asset)
_task_asset = Task.create_task("asset_task_test",
need=[_neededTask, _neededTask2], asset=_asset)
return [_task_shot, _task_asset]