public function testGetEntityClassByTableName()
{
$metadata = $this->getMockBuilder('Doctrine\\ORM\\Mapping\\ClassMetadata')->disableOriginalConstructor()->getMock();
$metadata->expects($this->once())->method('getTableName')->will($this->returnValue('acme_test'));
$metadata->expects($this->once())->method('getName')->will($this->returnValue('Oro\\Bundle\\EntityBundle\\Tests\\Unit\\ORM\\Fixtures\\TestEntity'));
$metadataFactory = $this->getMockBuilder('Doctrine\\ORM\\Mapping\\ClassMetadataFactory')->disableOriginalConstructor()->getMock();
$metadataFactory->expects($this->once())->method('getAllMetadata')->will($this->returnValue([$metadata]));
$em = $this->getMockBuilder('Doctrine\\ORM\\EntityManager')->disableOriginalConstructor()->getMock();
$em->expects($this->once())->method('getMetadataFactory')->will($this->returnValue($metadataFactory));
$this->doctrine->expects($this->once())->method('getManagers')->will($this->returnValue(array('default' => $em)));
$this->doctrine->expects($this->once())->method('getManager')->with($this->equalTo('default'))->will($this->returnValue($em));
$this->assertEquals('Oro\\Bundle\\EntityBundle\\Tests\\Unit\\ORM\\Fixtures\\TestEntity', $this->helper->getEntityClassByTableName('acme_test'));
}
/**
* @param Schema $schema
* @param QueryBag $queries
* @param string $tableName
* @param string $columnName
* @param array $options
*/
protected function processColumn(Schema $schema, QueryBag $queries, $tableName, $columnName, $options)
{
$className = $this->entityMetadataHelper->getEntityClassByTableName($tableName);
$table = $schema->getTable($tableName);
if (!isset($options[ExtendOptionsManager::NEW_NAME_OPTION])) {
if (isset($options[ExtendOptionsManager::TYPE_OPTION])) {
$columnType = $options[ExtendOptionsManager::TYPE_OPTION];
if (!in_array($columnType, ['oneToMane', 'manyToMany', 'manyToOne', 'optionSet'])) {
$indexName = $this->nameGenerator->generateIndexNameForExtendFieldVisibleInGrid($className, $columnName);
$enabled = !isset($options['datagrid']['is_visible']) || $options['datagrid']['is_visible'];
if ($enabled && !$table->hasIndex($indexName)) {
$table->addIndex([$columnName], $indexName);
} elseif (!$enabled && $table->hasIndex($indexName)) {
$table->dropIndex($indexName);
}
}
}
} else {
// in case of renaming column name we should rename existing index
$newColumnName = $options[ExtendOptionsManager::NEW_NAME_OPTION];
$indexName = $this->nameGenerator->generateIndexNameForExtendFieldVisibleInGrid($className, $columnName);
if ($table->hasIndex($indexName)) {
$table->dropIndex($indexName);
$newIndexName = $this->nameGenerator->generateIndexNameForExtendFieldVisibleInGrid($className, $newColumnName);
$this->renameExtension->addIndex($schema, $queries, $tableName, [$newColumnName], $newIndexName);
}
}
}
/**
* Gets an entity class name by its table name
*
* @param string $tableName
* @param string $customEntityClassName The name of a custom entity
* @param bool $throwExceptionIfNotFound
*
* @return string|null
*
* @throws \RuntimeException if an entity class name was not found and $throwExceptionIfNotFound = TRUE
*/
protected function getEntityClassName($tableName, $customEntityClassName = null, $throwExceptionIfNotFound = true)
{
if (!isset($this->tableToEntityMap[$tableName])) {
$entityClassName = !empty($customEntityClassName) ? $customEntityClassName : $this->entityMetadataHelper->getEntityClassByTableName($tableName);
if ($throwExceptionIfNotFound && empty($entityClassName)) {
throw new \RuntimeException(sprintf('Cannot find entity for "%s" table.', $tableName));
}
$this->tableToEntityMap[$tableName] = $entityClassName;
}
$result = $this->tableToEntityMap[$tableName];
if ($throwExceptionIfNotFound && empty($result)) {
throw new \RuntimeException(sprintf('Cannot find entity for "%s" table.', $tableName));
}
return $result;
}
/**
* @param Schema $schema
* @param QueryBag $queries
* @param string $tableName
* @param string $columnName
* @param array $options
*/
protected function processColumn(Schema $schema, QueryBag $queries, $tableName, $columnName, $options)
{
$className = $this->entityMetadataHelper->getEntityClassByTableName($tableName);
if (null === $className) {
return;
}
$table = $schema->getTable($tableName);
if (!$table->hasColumn($columnName)) {
return;
}
if (!isset($options[ExtendOptionsManager::NEW_NAME_OPTION])) {
if (isset($options[ExtendOptionsManager::TYPE_OPTION])) {
$this->buildIndex($columnName, $options, $className, $table);
}
} else {
// in case of renaming column name we should rename existing index
$this->renameIndex($schema, $queries, $tableName, $columnName, $options, $className, $table);
}
}
protected function renameOneToManyExtendField(Schema $schema, QueryBag $queries, Table $table, $associationName, $targetEntityClassName, EntityMetadataHelper $entityMetadataHelper)
{
$entityClassName = $entityMetadataHelper->getEntityClassByTableName($table->getName());
$targetTableName = $entityMetadataHelper->getTableNameByEntityClass($targetEntityClassName);
if ($schema->hasTable($targetTableName)) {
$targetTable = $schema->getTable($targetTableName);
$oldTargetColumnName = sprintf('field_%s_%s_id', strtolower(ExtendHelper::getShortClassName($entityClassName)), $associationName);
if ($targetTable->hasColumn($oldTargetColumnName)) {
$newTargetColumnName = $this->nameGenerator->generateOneToManyRelationColumnName($entityClassName, $associationName);
$oldIndexName = $this->nameGenerator->generateIndexName($targetTableName, [$oldTargetColumnName], false, true);
if ($targetTable->hasIndex($oldIndexName)) {
$targetTable->dropIndex($oldIndexName);
}
$oldForeignKeyName = $this->nameGenerator->generateForeignKeyConstraintName($targetTableName, [$oldTargetColumnName], true);
if ($targetTable->hasForeignKey($oldForeignKeyName)) {
$targetTable->removeForeignKey($oldForeignKeyName);
}
$this->renameExtension->renameColumn($schema, $queries, $targetTable, $oldTargetColumnName, $newTargetColumnName);
$this->renameExtension->addIndex($schema, $queries, $targetTable->getName(), [$newTargetColumnName]);
$this->renameExtension->addForeignKeyConstraint($schema, $queries, $targetTable->getName(), $table->getName(), [$newTargetColumnName], $table->getPrimaryKeyColumns(), ['onDelete' => 'SET NULL']);
}
}
}
/**
* Gets an entity full class name by a table name
*
* @param string $tableName
*
* @return string|null
*/
public function getEntityClassByTableName($tableName)
{
return $this->entityMetadataHelper->getEntityClassByTableName($tableName);
}
