public function testGetEntityOwner()
{
$owner = new User();
$entity = new Entity();
$entity->setOwner($owner);
$this->assertSame($owner, $this->extension->getEntityOwner($entity));
}
/**
* @dataProvider preSetData
*/
public function testPrePersistUser($token, $securityConfig, $expect)
{
$entity = new Entity();
$this->securityContext->expects($this->once())->method('getToken')->will($this->returnValue($token));
$args = new LifecycleEventArgs($entity, $this->getMock('Doctrine\\Common\\Persistence\\ObjectManager'));
$this->configProvider->expects($this->once())->method('hasConfig')->will($this->returnValue(true));
$this->configProvider->expects($this->once())->method('getConfig')->will($this->returnValue($securityConfig));
$this->listener->prePersist($args);
if (isset($expect['owner'])) {
$this->assertEquals($expect['owner'], $entity->getOwner());
} else {
$this->assertNull($entity->getOwner());
}
if (isset($expect['organization'])) {
$this->assertEquals($expect['organization'], $entity->getOrganization());
} else {
$this->assertNull($entity->getOrganization());
}
}
