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]