public function testGetOrganizations()
{
$user = new User();
$disabledOrganization = new Organization();
$organization = new Organization();
$organization->setEnabled(true);
$user->setOrganizations(new ArrayCollection(array($organization, $disabledOrganization)));
$token = $this->getMock('Symfony\\Component\\Security\\Core\\Authentication\\Token\\TokenInterface');
$this->securityContext->expects($this->once())->method('getToken')->will($this->returnValue($token));
$token->expects($this->once())->method('getUser')->will($this->returnValue($user));
$result = $this->twigExtension->getOrganizations();
$this->assertInternalType('array', $result);
$this->assertCount(1, $result);
$this->assertSame($organization, $result[0]);
}
public function testOrganizations()
{
$user = new User();
$disabledOrganization = new Organization();
$organization = new Organization();
$organization->setEnabled(true);
$user->setOrganizations(new ArrayCollection(array($organization)));
$this->assertContains($organization, $user->getOrganizations());
$user->removeOrganization($organization);
$this->assertNotContains($organization, $user->getOrganizations());
$user->addOrganization($organization);
$this->assertContains($organization, $user->getOrganizations());
$user->addOrganization($disabledOrganization);
$result = $user->getOrganizations(true);
$this->assertTrue($result->count() == 1);
$this->assertSame($result->first(), $organization);
}
