public function testGettersAndSetters()
{
$entity1 = new \stdClass();
$entity2 = new \stdClass();
$this->entityRegistry->addEntity('Test\\Entity', 'test1', $entity1);
$this->entityRegistry->addEntity('Test\\Entity', 'test2', $entity2);
// entity1
$this->assertTrue($this->entityRegistry->hasEntity('Test\\Entity', 'test1'));
$this->assertSame($entity1, $this->entityRegistry->getEntity('Test\\Entity', 'test1'));
// entity2
$this->assertTrue($this->entityRegistry->hasEntity('Test\\Entity', 'test2'));
$this->assertSame($entity2, $this->entityRegistry->getEntity('Test\\Entity', 'test2'));
// unknown entity
$this->assertFalse($this->entityRegistry->hasEntity('Test\\Entity', 'test3'));
$this->setExpectedException('Oro\\Bundle\\ImportExportBundle\\Exception\\LogicException', 'The entity "Test\\Entity" with key "test3" does not exist.');
$this->entityRegistry->getEntity('Test\\Entity', 'test3');
}
public function testGetEntityData()
{
$entityKey = 'test1';
$entity = new \stdClass();
$repository = $this->getMock('Oro\\Bundle\\ImportExportBundle\\TemplateFixture\\TemplateEntityRepositoryInterface');
$repository->expects($this->once())->method('getEntityClass')->will($this->returnValue(self::ENTITY_CLASS));
$this->templateManager->addEntityRepository($repository);
$this->templateRepository->expects($this->once())->method('createEntity')->with($entityKey)->will($this->returnValue($entity));
$repository->expects($this->once())->method('fillEntityData')->with($entityKey, $this->identicalTo($entity));
// test that new entity is created
$data = $this->callProtectedMethod($this->templateRepository, 'getEntityData', [$entityKey]);
$data = iterator_to_array($data);
$this->assertSame($entity, current($data));
$this->assertSame($entity, $this->entityRegistry->getEntity(self::ENTITY_CLASS, $entityKey));
// test that existing entity is returned
$data = $this->callProtectedMethod($this->templateRepository, 'getEntityData', [$entityKey]);
$data = iterator_to_array($data);
$this->assertSame($entity, current($data));
$this->assertSame($entity, $this->entityRegistry->getEntity(self::ENTITY_CLASS, $entityKey));
}