/**
* @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());
}
