Add treat test for TCK compliance

This commit is contained in:
Christian Beikov 2022-02-08 11:19:55 +01:00
parent 0a5b62421e
commit cab03a238d
1 changed files with 132 additions and 0 deletions

View File

@ -0,0 +1,132 @@
/*
* 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.query.hql.treat;
import java.util.List;
import org.hibernate.testing.orm.junit.EntityManagerFactoryScope;
import org.hibernate.testing.orm.junit.Jpa;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import jakarta.persistence.DiscriminatorColumn;
import jakarta.persistence.DiscriminatorType;
import jakarta.persistence.DiscriminatorValue;
import jakarta.persistence.Entity;
import jakarta.persistence.Id;
import jakarta.persistence.Inheritance;
import jakarta.persistence.InheritanceType;
import jakarta.persistence.ManyToOne;
import jakarta.persistence.PrimaryKeyJoinColumn;
import jakarta.persistence.SecondaryTable;
import jakarta.persistence.SecondaryTables;
import jakarta.persistence.Table;
import static org.junit.jupiter.api.Assertions.assertEquals;
@Jpa(
annotatedClasses = {
HqlTreatTest.Product.class,
HqlTreatTest.SoftwareProduct.class,
HqlTreatTest.LineItem.class
}
)
public class HqlTreatTest {
@BeforeEach
public void setUp(EntityManagerFactoryScope scope) {
scope.inTransaction(
entityManager -> {
Product product1 = new Product( 1, "Monitor" );
Product product2 = new SoftwareProduct( 2, "Linux" );
LineItem lineItem1 = new LineItem( 1, 3, product1 );
LineItem lineItem2 = new LineItem( 2, 5, product2 );
entityManager.persist( product1 );
entityManager.persist( product2 );
entityManager.persist( lineItem1 );
entityManager.persist( lineItem2 );
}
);
}
@Test
public void treatJoinClassTest(EntityManagerFactoryScope scope) {
scope.inTransaction(
entityManager -> {
List<String> names = entityManager.createQuery(
"SELECT s.name FROM LineItem l JOIN TREAT(l.product AS SoftwareProduct) s" )
.getResultList();
assertEquals( 1, names.size() );
assertEquals( "Linux", names.get( 0 ) );
}
);
}
@Entity(name = "LineItem")
@Table(name = "LINEITEM_TABLE")
public static class LineItem {
@Id
private Integer id;
private int quantity;
@ManyToOne
private Product product;
public LineItem() {
}
public LineItem(Integer id, int quantity, Product product) {
this.id = id;
this.quantity = quantity;
this.product = product;
}
}
@Entity(name = "Product")
@Table(name = "PRODUCT_TABLE")
@SecondaryTables({
@SecondaryTable(name = "PRODUCT_DETAILS", pkJoinColumns = @PrimaryKeyJoinColumn(name = "ID"))
})
@Inheritance(strategy = InheritanceType.SINGLE_TABLE)
@DiscriminatorColumn(name = "PRODUCT_TYPE", discriminatorType = DiscriminatorType.STRING)
@DiscriminatorValue("Product")
public static class Product {
@Id
private Integer id;
private String name;
public Product() {
}
public Product(Integer id, String name) {
this.id = id;
this.name = name;
}
}
@Entity(name = "SoftwareProduct")
@DiscriminatorValue("SW")
public static class SoftwareProduct extends Product {
public SoftwareProduct() {
}
public SoftwareProduct(Integer id, String name) {
super( id, name );
}
}
}