Skip to content

Commit

Permalink
HHH-12937 : test cases
Browse files Browse the repository at this point in the history
  • Loading branch information
gbadner committed Oct 16, 2018
1 parent f2b8244 commit da50076
Show file tree
Hide file tree
Showing 7 changed files with 1,417 additions and 1 deletion.
@@ -0,0 +1,309 @@
/*
* 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.test.where.annotations;

import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import javax.persistence.CollectionTable;
import javax.persistence.Column;
import javax.persistence.ElementCollection;
import javax.persistence.Entity;
import javax.persistence.Id;
import javax.persistence.JoinColumn;
import javax.persistence.Table;

import org.hibernate.Hibernate;
import org.hibernate.Session;
import org.hibernate.annotations.Immutable;
import org.hibernate.annotations.Where;

import org.hibernate.testing.TestForIssue;
import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

/**
* @author Gail Badner
*/
public class LazyElementCollectionBasicNonUniqueIdWhereTest extends BaseCoreFunctionalTestCase {

protected Class<?>[] getAnnotatedClasses() {
return new Class[] { Material.class, Building.class };
}

@Before
public void setup() {
Session session = openSession();
session.beginTransaction();
{
session.createSQLQuery( "DROP TABLE MAIN_TABLE" ).executeUpdate();
session.createSQLQuery( "DROP TABLE COLLECTION_TABLE" ).executeUpdate();
session.createSQLQuery( "DROP TABLE MATERIAL_RATINGS" ).executeUpdate();

session.createSQLQuery(
"create table MAIN_TABLE( " +
"ID integer not null, NAME varchar(255) not null, CODE varchar(10) not null, " +
"primary key (ID, CODE) )"
).executeUpdate();

session.createSQLQuery( "insert into MAIN_TABLE(ID, NAME, CODE) VALUES( 1, 'plastic', 'MATERIAL' )" )
.executeUpdate();
session.createSQLQuery( "insert into MAIN_TABLE(ID, NAME, CODE) VALUES( 1, 'house', 'BUILDING' )" )
.executeUpdate();
session.createSQLQuery( "insert into MAIN_TABLE(ID, NAME, CODE) VALUES( 1, 'high', 'RATING' )" )
.executeUpdate();
session.createSQLQuery( "insert into MAIN_TABLE(ID, NAME, CODE) VALUES( 2, 'medium', 'RATING' )" )
.executeUpdate();
session.createSQLQuery( "insert into MAIN_TABLE(ID, NAME, CODE) VALUES( 3, 'low', 'RATING' )" )
.executeUpdate();
session.createSQLQuery( "insert into MAIN_TABLE(ID, NAME, CODE) VALUES( 1, 'small', 'SIZE' )" )
.executeUpdate();
session.createSQLQuery( "insert into MAIN_TABLE(ID, NAME, CODE) VALUES( 2, 'medium', 'SIZE' )" )
.executeUpdate();

session.createSQLQuery(
"create table COLLECTION_TABLE( " +
"MAIN_ID integer not null, MAIN_CODE varchar(10) not null, " +
"VALUE varchar(10) not null, VALUE_CODE varchar(10) not null, " +
"primary key (MAIN_ID, MAIN_CODE, VALUE, VALUE_CODE))"
).executeUpdate();

session.createSQLQuery(
"insert into COLLECTION_TABLE(MAIN_ID, MAIN_CODE, VALUE, VALUE_CODE) " +
"VALUES( 1, 'MATERIAL', 'high', 'RATING' )"
).executeUpdate();
session.createSQLQuery(
"insert into COLLECTION_TABLE(MAIN_ID, MAIN_CODE, VALUE, VALUE_CODE) " +
"VALUES( 1, 'MATERIAL', 'medium', 'RATING' )"
).executeUpdate();
session.createSQLQuery(
"insert into COLLECTION_TABLE(MAIN_ID, MAIN_CODE, VALUE, VALUE_CODE) " +
"VALUES( 1, 'MATERIAL', 'low', 'RATING' )"
).executeUpdate();

session.createSQLQuery(
"insert into COLLECTION_TABLE(MAIN_ID, MAIN_CODE, VALUE, VALUE_CODE) " +
"VALUES( 1, 'MATERIAL', 'medium', 'SIZE' )"
).executeUpdate();

session.createSQLQuery(
"insert into COLLECTION_TABLE(MAIN_ID, MAIN_CODE, VALUE, VALUE_CODE) " +
"VALUES( 1, 'BUILDING', 'high', 'RATING' )"
).executeUpdate();

session.createSQLQuery(
"insert into COLLECTION_TABLE(MAIN_ID, MAIN_CODE, VALUE, VALUE_CODE) " +
"VALUES( 1, 'BUILDING', 'small', 'SIZE' )"
).executeUpdate();


session.createSQLQuery(
"create table MATERIAL_RATINGS( " +
"MATERIAL_ID integer not null, RATING varchar(10) not null," +
" primary key (MATERIAL_ID, RATING))"
).executeUpdate();

session.createSQLQuery(
"insert into MATERIAL_RATINGS(MATERIAL_ID, RATING) VALUES( 1, 'high' )"
).executeUpdate();

}
session.getTransaction().commit();
session.close();
}

@After
public void cleanup() {
Session session = openSession();
session.beginTransaction();
{
session.createSQLQuery( "delete from MATERIAL_RATINGS" ).executeUpdate();
session.createSQLQuery( "delete from COLLECTION_TABLE" ).executeUpdate();
session.createSQLQuery( "delete from MAIN_TABLE" ).executeUpdate();
}
session.getTransaction().commit();
session.close();
}

@Test
@TestForIssue( jiraKey = "HHH-12937")
public void testInitializeFromUniqueAssociationTable() {
Session session = openSession();
session.beginTransaction();
{
Material material = session.get( Material.class, 1 );
assertEquals( "plastic", material.getName() );

// Material#ratings is mapped with lazy="true"
assertFalse( Hibernate.isInitialized( material.getRatings() ) );
assertEquals( 1, material.getRatings().size() );
assertTrue( Hibernate.isInitialized( material.getRatings() ) );

assertEquals( "high", material.getRatings().iterator().next() );
}
session.getTransaction().commit();
session.close();
}

@Test
@TestForIssue( jiraKey = "HHH-12937")
public void testInitializeFromNonUniqueAssociationTable() {
Session session = openSession();
session.beginTransaction();
{
Material material = session.get( Material.class, 1 );
assertEquals( "plastic", material.getName() );

// Material#sizesFromCombined is mapped with lazy="true"
assertFalse( Hibernate.isInitialized( material.getSizesFromCombined() ) );
assertEquals( 1, material.getSizesFromCombined().size() );
assertTrue( Hibernate.isInitialized( material.getSizesFromCombined() ) );

assertEquals( "medium", material.getSizesFromCombined().iterator().next() );

Building building = session.get( Building.class, 1 );

// building.ratingsFromCombined is mapped with lazy="true"
assertFalse( Hibernate.isInitialized( building.getRatingsFromCombined() ) );
assertEquals( 1, building.getRatingsFromCombined().size() );
assertTrue( Hibernate.isInitialized( building.getRatingsFromCombined() ) );
assertEquals( "high", building.getRatingsFromCombined().iterator().next() );

// Building#sizesFromCombined is mapped with lazy="true"
assertFalse( Hibernate.isInitialized( building.getSizesFromCombined() ) );
assertEquals( 1, building.getSizesFromCombined().size() );
assertTrue( Hibernate.isInitialized( building.getSizesFromCombined() ) );
assertEquals( "small", building.getSizesFromCombined().iterator().next() );
}
session.getTransaction().commit();
session.close();
}

@Entity( name = "Material" )
@Table( name = "MAIN_TABLE" )
@Where( clause = "CODE = 'MATERIAL'" )
public static class Material {
private int id;

private String name;
private Set<String> sizesFromCombined = new HashSet<>();
private List<String> mediumOrHighRatingsFromCombined = new ArrayList<>();
private Set<String> ratings = new HashSet<>();

@Id
@Column( name = "ID" )
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}

@Column( name = "NAME")
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}

@ElementCollection
@CollectionTable(
name = "COLLECTION_TABLE",
joinColumns = { @JoinColumn( name = "MAIN_ID" ) }
)
@Column( name="VALUE")
@Where( clause = "MAIN_CODE='MATERIAL' AND VALUE_CODE='SIZE'")
@Immutable
public Set<String> getSizesFromCombined() {
return sizesFromCombined;
}
public void setSizesFromCombined(Set<String> sizesFromCombined) {
this.sizesFromCombined = sizesFromCombined;
}

@ElementCollection
@CollectionTable(
name = "MATERIAL_RATINGS",
joinColumns = { @JoinColumn( name = "MATERIAL_ID" ) }
)
@Column( name="RATING")
@Immutable
public Set<String> getRatings() {
return ratings;
}
public void setRatings(Set<String> ratings) {
this.ratings = ratings;
}
}

@Entity( name = "Building" )
@Table( name = "MAIN_TABLE" )
@Where( clause = "CODE = 'BUILDING'" )
public static class Building {
private int id;
private String name;
private Set<String> sizesFromCombined = new HashSet<>();
private Set<String> ratingsFromCombined = new HashSet<>();
private List<String> mediumOrHighRatings = new ArrayList<>();

@Id
@Column( name = "ID" )
public int getId() {
return id;
}
public void setId(int id) {
this.id = id;
}

@Column( name = "NAME")
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}

@ElementCollection
@CollectionTable(
name = "COLLECTION_TABLE",
joinColumns = { @JoinColumn( name = "MAIN_ID" ) }
)
@Column( name="VALUE")
@Where( clause = "MAIN_CODE='BUILDING' AND VALUE_CODE='SIZE'")
@Immutable
public Set<String> getSizesFromCombined() {
return sizesFromCombined;
}
public void setSizesFromCombined(Set<String> sizesFromCombined) {
this.sizesFromCombined = sizesFromCombined;
}

@ElementCollection
@CollectionTable(
name = "COLLECTION_TABLE",
joinColumns = { @JoinColumn( name = "MAIN_ID" ) }
)
@Column( name="VALUE")
@Where( clause = "MAIN_CODE='BUILDING' AND VALUE_CODE='RATING'" )
@Immutable
public Set<String> getRatingsFromCombined() {
return ratingsFromCombined;
}
public void setRatingsFromCombined(Set<String> ratingsFromCombined) {
this.ratingsFromCombined = ratingsFromCombined;
}

}
}

0 comments on commit da50076

Please sign in to comment.