HHH-10934 - Initial unit test
This commit is contained in:
MilovanderZee
Chris Cranford
parent
6307a47c13
commit
736da3247c
|
|
@ -0,0 +1,357 @@
|
|||
package org.hibernate.test.tool.schema.internal;
|
||||
|
||||
import java.lang.reflect.InvocationTargetException;
|
||||
import java.lang.reflect.Method;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.Set;
|
||||
|
||||
import org.hibernate.boot.Metadata;
|
||||
import org.hibernate.boot.model.TruthValue;
|
||||
import org.hibernate.boot.model.naming.Identifier;
|
||||
import org.hibernate.boot.model.relational.Namespace;
|
||||
import org.hibernate.boot.model.relational.Namespace.Name;
|
||||
import org.hibernate.boot.model.relational.QualifiedTableName;
|
||||
import org.hibernate.dialect.Dialect;
|
||||
import org.hibernate.engine.jdbc.env.spi.IdentifierHelper;
|
||||
import org.hibernate.engine.jdbc.internal.Formatter;
|
||||
import org.hibernate.mapping.Column;
|
||||
import org.hibernate.mapping.ForeignKey;
|
||||
import org.hibernate.mapping.Table;
|
||||
import org.hibernate.tool.schema.extract.internal.ColumnInformationImpl;
|
||||
import org.hibernate.tool.schema.extract.internal.ForeignKeyInformationImpl;
|
||||
import org.hibernate.tool.schema.extract.internal.TableInformationImpl;
|
||||
import org.hibernate.tool.schema.extract.spi.ColumnInformation;
|
||||
import org.hibernate.tool.schema.extract.spi.DatabaseInformation;
|
||||
import org.hibernate.tool.schema.extract.spi.ForeignKeyInformation;
|
||||
import org.hibernate.tool.schema.extract.spi.ForeignKeyInformation.ColumnReferenceMapping;
|
||||
import org.hibernate.tool.schema.extract.spi.InformationExtractor;
|
||||
import org.hibernate.tool.schema.extract.spi.NameSpaceTablesInformation;
|
||||
import org.hibernate.tool.schema.extract.spi.TableInformation;
|
||||
import org.hibernate.tool.schema.internal.AbstractSchemaMigrator;
|
||||
import org.hibernate.tool.schema.internal.exec.GenerationTarget;
|
||||
import org.hibernate.tool.schema.spi.ExecutionOptions;
|
||||
import org.junit.Assert;
|
||||
import org.junit.Test;
|
||||
import org.mockito.Mockito;
|
||||
|
||||
/**
|
||||
* @author Milo van der Zee
|
||||
*/
|
||||
public class CheckForExistingForeignKeyTest {
|
||||
|
||||
private class SchemaMigrator extends AbstractSchemaMigrator {
|
||||
|
||||
/**
|
||||
* Needed constructor.
|
||||
*/
|
||||
public SchemaMigrator() {
|
||||
super( null, null );
|
||||
}
|
||||
|
||||
/**
|
||||
* Needed implementation. Not used in test.
|
||||
*/
|
||||
@Override
|
||||
protected NameSpaceTablesInformation performTablesMigration(Metadata metadata, DatabaseInformation existingDatabase, ExecutionOptions options,
|
||||
Dialect dialect,
|
||||
Formatter formatter, Set<String> exportIdentifiers, boolean tryToCreateCatalogs, boolean tryToCreateSchemas,
|
||||
Set<Identifier> exportedCatalogs, Namespace namespace, GenerationTarget[] targets) {
|
||||
return null;
|
||||
}
|
||||
}
|
||||
|
||||
private class ColumnReferenceMappingImpl implements ColumnReferenceMapping {
|
||||
|
||||
private ColumnInformation referencingColumnMetadata;
|
||||
private ColumnInformation referencedColumnMetadata;
|
||||
|
||||
public ColumnReferenceMappingImpl(ColumnInformation referencingColumnMetadata, ColumnInformation referencedColumnMetadata) {
|
||||
this.referencingColumnMetadata = referencingColumnMetadata;
|
||||
this.referencedColumnMetadata = referencedColumnMetadata;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ColumnInformation getReferencingColumnMetadata() {
|
||||
return referencingColumnMetadata;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ColumnInformation getReferencedColumnMetadata() {
|
||||
return referencedColumnMetadata;
|
||||
}
|
||||
}
|
||||
|
||||
private class IdentifierHelperImpl implements IdentifierHelper {
|
||||
|
||||
@Override
|
||||
public Identifier normalizeQuoting(Identifier identifier) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Identifier toIdentifier(String text) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Identifier toIdentifier(String text, boolean quoted) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Identifier applyGlobalQuoting(String text) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public boolean isReservedWord(String word) {
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toMetaDataCatalogName(Identifier catalogIdentifier) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toMetaDataSchemaName(Identifier schemaIdentifier) {
|
||||
return null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String toMetaDataObjectName(Identifier identifier) {
|
||||
return identifier.getText();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* If the key has no name it should never be found. Result is that those keys are always recreated. But keys always
|
||||
* have a name so this is no problem.
|
||||
*
|
||||
* @throws NoSuchMethodException - error
|
||||
* @throws SecurityException - error
|
||||
* @throws IllegalAccessException - error
|
||||
* @throws IllegalArgumentException - error
|
||||
* @throws InvocationTargetException - error
|
||||
*/
|
||||
@Test
|
||||
public void testForeignKeyWithoutName()
|
||||
throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
|
||||
// Get the private method
|
||||
Method method = AbstractSchemaMigrator.class.getDeclaredMethod( "checkForExistingForeignKey", ForeignKey.class, TableInformation.class );
|
||||
method.setAccessible( true );
|
||||
|
||||
// foreignKey name with same name should match
|
||||
ForeignKey foreignKey = new ForeignKey();
|
||||
TableInformation tableInformation = new TableInformationImpl( null, null, null, false, null );
|
||||
boolean found = (boolean) method.invoke( new SchemaMigrator(), foreignKey, tableInformation );
|
||||
Assert.assertFalse( "Key should not be found", found );
|
||||
}
|
||||
|
||||
/**
|
||||
* Test key not found if tableinformation is missing.
|
||||
*
|
||||
* @throws NoSuchMethodException - error
|
||||
* @throws SecurityException - error
|
||||
* @throws IllegalAccessException - error
|
||||
* @throws IllegalArgumentException - error
|
||||
* @throws InvocationTargetException - error
|
||||
*/
|
||||
@Test
|
||||
public void testMissingTableInformation()
|
||||
throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
|
||||
// Get the private method
|
||||
Method method = AbstractSchemaMigrator.class.getDeclaredMethod( "checkForExistingForeignKey", ForeignKey.class, TableInformation.class );
|
||||
method.setAccessible( true );
|
||||
|
||||
// foreignKey name with same name should match
|
||||
ForeignKey foreignKey = new ForeignKey();
|
||||
foreignKey.setName( "objectId2id" );
|
||||
boolean found = (boolean) method.invoke( new SchemaMigrator(), foreignKey, null );
|
||||
Assert.assertFalse( "Key should not be found", found );
|
||||
}
|
||||
|
||||
/**
|
||||
* Check detection of existing foreign keys with the same name exists.
|
||||
*
|
||||
* @throws SecurityException - error
|
||||
* @throws NoSuchMethodException - error
|
||||
* @throws InvocationTargetException - error
|
||||
* @throws IllegalArgumentException - error
|
||||
* @throws IllegalAccessException - error
|
||||
* @throws NoSuchFieldException - error
|
||||
*/
|
||||
@Test
|
||||
public void testKeyWithSameNameExists()
|
||||
throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException, InvocationTargetException,
|
||||
NoSuchFieldException {
|
||||
// Get the private method
|
||||
Method method = AbstractSchemaMigrator.class.getDeclaredMethod( "checkForExistingForeignKey", ForeignKey.class, TableInformation.class );
|
||||
method.setAccessible( true );
|
||||
|
||||
ForeignKey foreignKey = new ForeignKey();
|
||||
foreignKey.setName( "objectId2id" );
|
||||
foreignKey.addColumn( new Column( "id" ) );
|
||||
foreignKey.setReferencedTable( new Table( "table2" ) );
|
||||
|
||||
InformationExtractor informationExtractor = Mockito.mock( InformationExtractor.class );
|
||||
IdentifierHelper identifierHelper = new IdentifierHelperImpl();
|
||||
List<ForeignKeyInformation> fks = new ArrayList<>();
|
||||
fks.add( new ForeignKeyInformationImpl( new Identifier( "objectId2id", false ), new ArrayList<>() ) );
|
||||
Mockito.when( informationExtractor.getForeignKeys( Mockito.any() ) ).thenReturn( fks );
|
||||
Name schemaName = new Name( new Identifier( "-", false ), new Identifier( "-", false ) );
|
||||
QualifiedTableName tableName = new QualifiedTableName( schemaName, new Identifier( "-", false ) );
|
||||
TableInformation tableInformation = new TableInformationImpl( informationExtractor, identifierHelper, tableName, false, null );
|
||||
|
||||
// foreignKey name with same name should match
|
||||
boolean found = (boolean) method.invoke( new SchemaMigrator(), foreignKey, tableInformation );
|
||||
Assert.assertTrue( "Key should be found", found );
|
||||
}
|
||||
|
||||
/**
|
||||
* Check detection of existing foreign keys with the same name exists.
|
||||
*
|
||||
* @throws SecurityException - error
|
||||
* @throws NoSuchMethodException - error
|
||||
* @throws InvocationTargetException - error
|
||||
* @throws IllegalArgumentException - error
|
||||
* @throws IllegalAccessException - error
|
||||
* @throws NoSuchFieldException - error
|
||||
*/
|
||||
@Test
|
||||
public void testKeyWithSameNameNotExists()
|
||||
throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException, InvocationTargetException,
|
||||
NoSuchFieldException {
|
||||
// Get the private method
|
||||
Method method = AbstractSchemaMigrator.class.getDeclaredMethod( "checkForExistingForeignKey", ForeignKey.class, TableInformation.class );
|
||||
method.setAccessible( true );
|
||||
|
||||
ForeignKey foreignKey = new ForeignKey();
|
||||
foreignKey.setName( "objectId2id_1" );
|
||||
foreignKey.addColumn( new Column( "id" ) );
|
||||
foreignKey.setReferencedTable( new Table( "table2" ) );
|
||||
|
||||
InformationExtractor informationExtractor = Mockito.mock( InformationExtractor.class );
|
||||
IdentifierHelper identifierHelper = new IdentifierHelperImpl();
|
||||
List<ForeignKeyInformation> fks = new ArrayList<>();
|
||||
fks.add( new ForeignKeyInformationImpl( new Identifier( "objectId2id_2", false ), new ArrayList<>() ) );
|
||||
Mockito.when( informationExtractor.getForeignKeys( Mockito.any() ) ).thenReturn( fks );
|
||||
Name schemaName = new Name( new Identifier( "-", false ), new Identifier( "-", false ) );
|
||||
QualifiedTableName tableName = new QualifiedTableName( schemaName, new Identifier( "-", false ) );
|
||||
TableInformation tableInformation = new TableInformationImpl( informationExtractor, identifierHelper, tableName, false, null );
|
||||
|
||||
// foreignKey name with same name should match
|
||||
boolean found = (boolean) method.invoke( new SchemaMigrator(), foreignKey, tableInformation );
|
||||
Assert.assertFalse( "Key should not be found", found );
|
||||
}
|
||||
|
||||
/**
|
||||
* Check detection of existing foreign key with the same mappings for a simple mapping (table1.objectId =>
|
||||
* table2.id).
|
||||
*
|
||||
* @throws SecurityException - error
|
||||
* @throws NoSuchMethodException - error
|
||||
* @throws InvocationTargetException - error
|
||||
* @throws IllegalArgumentException - error
|
||||
* @throws IllegalAccessException - error
|
||||
* @throws NoSuchFieldException - error
|
||||
*/
|
||||
@Test
|
||||
public void testCheckForExistingForeignKeyOne2One() throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException,
|
||||
InvocationTargetException, NoSuchFieldException {
|
||||
// Get the private method
|
||||
Method method = AbstractSchemaMigrator.class.getDeclaredMethod( "checkForExistingForeignKey", ForeignKey.class, TableInformation.class );
|
||||
method.setAccessible( true );
|
||||
|
||||
ForeignKey foreignKey = new ForeignKey();
|
||||
foreignKey.setName( "objectId2id_1" ); // Make sure the match is not successful based on key name
|
||||
foreignKey.addColumn( new Column( "id" ) );
|
||||
foreignKey.setReferencedTable( new Table( "table2" ) );
|
||||
|
||||
Name schemaName = new Name( new Identifier( "-", false ), new Identifier( "-", false ) );
|
||||
InformationExtractor informationExtractor = Mockito.mock( InformationExtractor.class );
|
||||
IdentifierHelper identifierHelper = new IdentifierHelperImpl();
|
||||
List<ForeignKeyInformation> fks = new ArrayList<>();
|
||||
fks.add( getForeignKeyInformation( "table2", "id", "object2Id_2" ) );
|
||||
Mockito.when( informationExtractor.getForeignKeys( Mockito.any() ) ).thenReturn( fks );
|
||||
QualifiedTableName tableName = new QualifiedTableName( schemaName, new Identifier( "-", false ) );
|
||||
TableInformation tableInformation = new TableInformationImpl( informationExtractor, identifierHelper, tableName, false, null );
|
||||
AbstractSchemaMigrator schemaMigrator = new SchemaMigrator();
|
||||
|
||||
// Check single-column-key to single-column-key, existing (table1.objectId => table2.id)
|
||||
boolean found = (boolean) method.invoke( schemaMigrator, foreignKey, tableInformation );
|
||||
Assert.assertTrue( "Key should be found", found );
|
||||
}
|
||||
|
||||
/**
|
||||
* Check detection of not existing foreign key with the same mappings for a simple mapping (table1.objectId =>
|
||||
* table2.id).
|
||||
*
|
||||
* @throws SecurityException - error
|
||||
* @throws NoSuchMethodException - error
|
||||
* @throws InvocationTargetException - error
|
||||
* @throws IllegalArgumentException - error
|
||||
* @throws IllegalAccessException - error
|
||||
* @throws NoSuchFieldException - error
|
||||
*/
|
||||
@Test
|
||||
public void testCheckForNotExistingForeignKeyOne2One() throws NoSuchMethodException, SecurityException, IllegalAccessException, IllegalArgumentException,
|
||||
InvocationTargetException, NoSuchFieldException {
|
||||
// Get the private method
|
||||
Method method = AbstractSchemaMigrator.class.getDeclaredMethod( "checkForExistingForeignKey", ForeignKey.class, TableInformation.class );
|
||||
method.setAccessible( true );
|
||||
|
||||
ForeignKey foreignKey = new ForeignKey();
|
||||
foreignKey.setName( "objectId2id_1" ); // Make sure the match is not successful based on key name
|
||||
foreignKey.addColumn( new Column( "id" ) );
|
||||
foreignKey.setReferencedTable( new Table( "table2" ) );
|
||||
|
||||
Name schemaName = new Name( new Identifier( "-", false ), new Identifier( "-", false ) );
|
||||
InformationExtractor informationExtractor = Mockito.mock( InformationExtractor.class );
|
||||
IdentifierHelper identifierHelper = new IdentifierHelperImpl();
|
||||
List<ForeignKeyInformation> fks = new ArrayList<>();
|
||||
fks.add( getForeignKeyInformation( "table2", "blah", "blahKey_001" ) );
|
||||
fks.add( getForeignKeyInformation( "table3", "id", "blahKey_002" ) );
|
||||
fks.add( getForeignKeyInformation( "table3", "blah", "blahKey_003" ) );
|
||||
Mockito.when( informationExtractor.getForeignKeys( Mockito.any() ) ).thenReturn( fks );
|
||||
QualifiedTableName tableName = new QualifiedTableName( schemaName, new Identifier( "-", false ) );
|
||||
TableInformation tableInformation = new TableInformationImpl( informationExtractor, identifierHelper, tableName, false, null );
|
||||
AbstractSchemaMigrator schemaMigrator = new SchemaMigrator();
|
||||
|
||||
// Check single-column-key to single-column-key, existing (table1.objectId => table2.id)
|
||||
boolean found = (boolean) method.invoke( schemaMigrator, foreignKey, tableInformation );
|
||||
Assert.assertFalse( "Key should not be found", found );
|
||||
}
|
||||
|
||||
/**
|
||||
* @param referencedTableName - String
|
||||
* @param referencingColumnName - String
|
||||
* @param keyName - String
|
||||
* @return ForeignKeyInformation
|
||||
*/
|
||||
private ForeignKeyInformation getForeignKeyInformation(String referencedTableName, String referencingColumnName, String keyName) {
|
||||
List<ColumnReferenceMapping> columnMappingList = new ArrayList<>();
|
||||
ColumnInformation referencingColumnMetadata = getColumnInformation( "-", referencingColumnName );
|
||||
ColumnInformation referencedColumnMetadata = getColumnInformation( referencedTableName, "-" );
|
||||
ColumnReferenceMapping columnReferenceMapping = new ColumnReferenceMappingImpl( referencingColumnMetadata, referencedColumnMetadata );
|
||||
columnMappingList.add( columnReferenceMapping );
|
||||
ForeignKeyInformationImpl foreignKeyInformation = new ForeignKeyInformationImpl( new Identifier( keyName, false ), columnMappingList );
|
||||
return foreignKeyInformation;
|
||||
}
|
||||
|
||||
private ColumnInformation getColumnInformation(String tableName, String columnName) {
|
||||
Name schemaName = new Name( new Identifier( "-", false ), new Identifier( "-", false ) );
|
||||
TableInformation containingTableInformation = new TableInformationImpl( null, null,
|
||||
new QualifiedTableName( schemaName, new Identifier( tableName, false ) ), false, null );
|
||||
Identifier columnIdentifier = new Identifier( columnName, false );
|
||||
int typeCode = 0;
|
||||
String typeName = null;
|
||||
int columnSize = 0;
|
||||
int decimalDigits = 0;
|
||||
TruthValue nullable = null;
|
||||
ColumnInformationImpl columnInformation = new ColumnInformationImpl( containingTableInformation, columnIdentifier, typeCode, typeName, columnSize,
|
||||
decimalDigits, nullable );
|
||||
return columnInformation;
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue