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