def find_shot(self, dbo): """Test find_shot, CRUD method :param dbObject dbo: dbo """ self.reset(dbo) # 1. find shot from db with orm.db_session: temp_shot, _ = Shot.find_shot_by_id(self.shot.id) # 2. test value self.assert_value(temp_shot) temp_shot, err = Shot.find_shot_by_id(-1) # 3. assert on error self.assertEqual(err, "Shot Not Found !") self.assertEqual(temp_shot, None) # 4. find_all shot from db temp_shots = Shot.find_all_shots() # 5. test value self.assertEqual(len(temp_shots), 1) self.assert_value(temp_shots[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]