/** * @param User $entity * @param array $businessUnits */ public function assignBusinessUnits($entity, array $businessUnits) { if ($businessUnits) { $businessUnits = $this->getBusinessUnitRepo()->getBusinessUnits($businessUnits); } else { $businessUnits = new ArrayCollection(); } $entity->setBusinessUnits($businessUnits); }
public function testBusinessUnit() { $user = new User(); $businessUnit = new BusinessUnit(); $user->setBusinessUnits(new ArrayCollection(array($businessUnit))); $this->assertContains($businessUnit, $user->getBusinessUnits()); $user->removeBusinessUnit($businessUnit); $this->assertNotContains($businessUnit, $user->getBusinessUnits()); $user->addBusinessUnit($businessUnit); $this->assertContains($businessUnit, $user->getBusinessUnits()); }