diff --git a/controller/organization.js b/controller/organization.js index b89c921..7e3793f 100644 --- a/controller/organization.js +++ b/controller/organization.js @@ -5,10 +5,20 @@ import { getOrganizations, } from "#services/organization"; import { logger } from "#util"; +import { isEntityIdValid } from "#middleware/entityIdValidation"; +import Accreditation from "#models/accreditation"; +import Parent from "#models/organization"; async function addOrganization(req, res) { const { parent, startDate, name, accreditation } = req.body; + const isAccreditationValid = await isEntityIdValid(accreditation, Accreditation); + const isParentValid = await isEntityIdValid(parent, Parent); try { + if (!isAccreditationValid || !isParentValid) { + res.status(400).json({ + error: "Invalid Id", + }); + } const organization = await addNewOrganization( parent, startDate,