/**
* Test relations between AccountUserRole and Websites
*/
public function testWebsiteRelations()
{
$accountUserRole = new AccountUserRole();
$website = new Website();
$this->assertInstanceOf('Doctrine\\Common\\Collections\\ArrayCollection', $accountUserRole->getWebsites());
$this->assertCount(0, $accountUserRole->getWebsites());
$this->assertInstanceOf('OroB2B\\Bundle\\CustomerBundle\\Entity\\AccountUserRole', $accountUserRole->addWebsite($website));
$this->assertCount(1, $accountUserRole->getWebsites());
$accountUserRole->addWebsite($website);
$this->assertCount(1, $accountUserRole->getWebsites());
$accountUserRole->removeWebsite($website);
$this->assertCount(0, $accountUserRole->getWebsites());
}
/**
* @param ObjectManager $manager
* @param string $roleLabel
* @param string $websiteName
*/
protected function loadRoleWithWebsite(ObjectManager $manager, $roleLabel, $websiteName)
{
$entity = new AccountUserRole();
$entity->setLabel($roleLabel);
/** @var Website $website */
$website = $this->getReference($websiteName);
$entity->addWebsite($website);
$this->setReference($entity->getLabel(), $entity);
$manager->persist($entity);
}
/**
* @param ObjectManager $manager
* @param AccountUserRole $role
*/
protected function setWebsiteDefaultRoles(ObjectManager $manager, AccountUserRole $role)
{
$websites = $manager->getRepository('OroB2BWebsiteBundle:Website')->findAll();
foreach ($websites as $website) {
$role->addWebsite($website);
}
}