Skip to content

Commit

Permalink
HHH-17065 Add CompositePrimaryKeyColumnOrderTest
Browse files Browse the repository at this point in the history
  • Loading branch information
bwaldvogel authored and beikov committed Oct 11, 2023
1 parent 8629e01 commit 3a84e40
Showing 1 changed file with 126 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,126 @@
/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later.
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.orm.test.schemaupdate.uniqueconstraint;

import java.io.Serializable;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.EnumSet;
import java.util.List;
import java.util.Map;
import java.util.Objects;

import org.hibernate.boot.Metadata;
import org.hibernate.cfg.AvailableSettings;
import org.hibernate.dialect.H2Dialect;
import org.hibernate.jpa.boot.spi.Bootstrap;
import org.hibernate.jpa.boot.spi.EntityManagerFactoryBuilder;
import org.hibernate.jpa.boot.spi.PersistenceUnitDescriptor;
import org.hibernate.tool.hbm2ddl.SchemaExport;
import org.hibernate.tool.schema.TargetType;

import org.hibernate.testing.orm.jpa.PersistenceUnitDescriptorAdapter;
import org.hibernate.testing.orm.junit.JiraKey;
import org.hibernate.testing.orm.junit.RequiresDialect;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.io.TempDir;

import jakarta.persistence.Entity;
import jakarta.persistence.EntityManagerFactory;
import jakarta.persistence.Id;
import jakarta.persistence.IdClass;

import static org.assertj.core.api.Assertions.assertThat;

@JiraKey("HHH-17065")
@RequiresDialect( H2Dialect.class )
class CompositePrimaryKeyColumnOrderTest {

@Test
public void testUniqueConstraintColumnOrder(@TempDir Path tempDir) throws Exception {
PersistenceUnitDescriptor puDescriptor = new PersistenceUnitDescriptorAdapter() {
@Override
public List<String> getManagedClassNames() {
return List.of( TestEntity.class.getName() );
}
};

Map<Object, Object> settings = Map.of( AvailableSettings.HBM2DDL_AUTO, "create" );
EntityManagerFactoryBuilder emfBuilder = Bootstrap.getEntityManagerFactoryBuilder( puDescriptor, settings );

Path ddlScript = tempDir.resolve( "ddl.sql" );

try (EntityManagerFactory entityManagerFactory = emfBuilder.build()) {
// we do not need the entityManagerFactory, but we need to build it in order to demonstrate the issue (HHH-17065)

Metadata metadata = emfBuilder.metadata();

new SchemaExport()
.setHaltOnError( true )
.setOutputFile( ddlScript.toString() )
.setFormat( true )
.create( EnumSet.of( TargetType.SCRIPT ), metadata );
}

String ddl = Files.readString( ddlScript );
assertThat( ddl ).contains( "primary key (b, a)" );
}

@Entity
@IdClass( CompositePrimaryKey.class )
static class TestEntity {

@Id
private String b;

@Id
private String a;

}

private static class CompositePrimaryKey implements Serializable {

private static final long serialVersionUID = 1L;

private String b;

private String a;

public String getB() {
return b;
}

public void setB(String b) {
this.b = b;
}

public String getA() {
return a;
}

public void setA(String a) {
this.a = a;
}

@Override
public boolean equals(Object o) {
if ( this == o ) {
return true;
}
if ( o == null || getClass() != o.getClass() ) {
return false;
}
CompositePrimaryKey that = (CompositePrimaryKey) o;
return Objects.equals( getB(), that.getB() ) && Objects.equals( getA(), that.getA() );
}

@Override
public int hashCode() {
return Objects.hash( getB(), getA() );
}
}
}

0 comments on commit 3a84e40

Please sign in to comment.