-
Notifications
You must be signed in to change notification settings - Fork 3.5k
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
288 additions
and
0 deletions.
There are no files selected for viewing
288 changes: 288 additions & 0 deletions
288
hibernate-core/src/test/java/org/hibernate/orm/test/fetch/FetchModeSubselectTest.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,288 @@ | ||
/* | ||
* 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.fetch; | ||
|
||
import java.util.HashSet; | ||
import java.util.List; | ||
import java.util.Set; | ||
|
||
import org.hibernate.annotations.Fetch; | ||
import org.hibernate.annotations.FetchMode; | ||
|
||
import org.hibernate.testing.TestForIssue; | ||
import org.hibernate.testing.orm.junit.DomainModel; | ||
import org.hibernate.testing.orm.junit.SessionFactory; | ||
import org.hibernate.testing.orm.junit.SessionFactoryScope; | ||
import org.junit.jupiter.api.AfterEach; | ||
import org.junit.jupiter.api.BeforeEach; | ||
import org.junit.jupiter.api.Test; | ||
|
||
import jakarta.persistence.CascadeType; | ||
import jakarta.persistence.Column; | ||
import jakarta.persistence.Entity; | ||
import jakarta.persistence.FetchType; | ||
import jakarta.persistence.Id; | ||
import jakarta.persistence.JoinColumn; | ||
import jakarta.persistence.JoinTable; | ||
import jakarta.persistence.ManyToMany; | ||
import jakarta.persistence.ManyToOne; | ||
import jakarta.persistence.OneToMany; | ||
import jakarta.persistence.Table; | ||
|
||
import static org.assertj.core.api.AssertionsForClassTypes.assertThat; | ||
|
||
@DomainModel( | ||
annotatedClasses = { | ||
FetchModeSubselectTest.User.class, | ||
FetchModeSubselectTest.Role.class, | ||
FetchModeSubselectTest.Agency.class | ||
} | ||
) | ||
@SessionFactory | ||
@TestForIssue(jiraKey = "HHH-15453") | ||
public class FetchModeSubselectTest { | ||
|
||
@BeforeEach | ||
public void setUp(SessionFactoryScope scope) { | ||
scope.inTransaction( | ||
session -> { | ||
User ann = new User( 1, "Ann" ); | ||
User bob = new User( 2, "Bob" ); | ||
|
||
Role manager = new Role( 1, "manager" ); | ||
Role developer = new Role( 2, "developer" ); | ||
|
||
ann.addRole( manager ); | ||
|
||
Agency agency = new Agency( 100, "Test Agency" ); | ||
|
||
agency.addUser( ann ); | ||
agency.addUser( bob ); | ||
|
||
session.persist( ann ); | ||
session.persist( bob ); | ||
session.persist( manager ); | ||
session.persist( developer ); | ||
session.persist( agency ); | ||
} | ||
); | ||
} | ||
|
||
@AfterEach | ||
public void tearDown(SessionFactoryScope scope) { | ||
scope.inTransaction( | ||
session -> { | ||
session.createMutationQuery( "delete from User" ).executeUpdate(); | ||
session.createMutationQuery( "delete from Role" ).executeUpdate(); | ||
session.createMutationQuery( "delete from Agency" ).executeUpdate(); | ||
} | ||
); | ||
} | ||
|
||
@Test | ||
public void testSelectUserWithRole(SessionFactoryScope scope) { | ||
scope.inSession( | ||
session -> { | ||
List<User> userList = session.createQuery( "FROM User WHERE id = 1", User.class ).list(); | ||
assertThat( userList.size() ).isEqualTo( 1 ); | ||
assertThat( userList.get( 0 ).getRoles().size() ).isEqualTo( 1 ); | ||
} | ||
); | ||
} | ||
|
||
@Test | ||
public void testGetUserWithRole(SessionFactoryScope scope) { | ||
scope.inSession( | ||
session -> { | ||
User user = session.get( User.class, 1 ); | ||
assertThat( user ).isNotNull(); | ||
assertThat( user.getRoles().size() ).isEqualTo( 1 ); | ||
} | ||
); | ||
} | ||
|
||
@Test | ||
public void testSelectUserWithoutRole(SessionFactoryScope scope) { | ||
scope.inSession( | ||
session -> { | ||
List<User> userList = session.createQuery( "FROM User WHERE id = 2", User.class ).list(); | ||
assertThat( userList.size() ).isEqualTo( 1 ); | ||
assertThat( userList.get( 0 ).getRoles().size() ).isEqualTo( 0 ); | ||
} | ||
); | ||
} | ||
|
||
@Test | ||
public void testGetUserWithoutRole(SessionFactoryScope scope) { | ||
scope.inSession( | ||
session -> { | ||
User user = session.get( User.class, 2 ); | ||
assertThat( user ).isNotNull(); | ||
assertThat( user.getRoles().size() ).isEqualTo( 0 ); | ||
} | ||
); | ||
} | ||
|
||
@Entity(name = "User") | ||
@Table(name = "USER_TABLE") | ||
public static class User { | ||
|
||
private Integer id; | ||
private String name; | ||
private Agency agency; | ||
private Set<Role> roles; | ||
|
||
public User() { | ||
} | ||
|
||
public User(Integer id, String name) { | ||
this.id = id; | ||
this.name = name; | ||
} | ||
|
||
@Id | ||
@Column(name = "USER_ID") | ||
public Integer getId() { | ||
return id; | ||
} | ||
|
||
public void setId(Integer userId) { | ||
this.id = userId; | ||
} | ||
|
||
@Column(name = "USER_NAME") | ||
public String getName() { | ||
return name; | ||
} | ||
|
||
public void setName(String userName) { | ||
this.name = userName; | ||
} | ||
|
||
@ManyToOne | ||
@JoinColumn(name = "AGENCY_ID") | ||
public Agency getAgency() { | ||
return agency; | ||
} | ||
|
||
public void setAgency(Agency agency) { | ||
this.agency = agency; | ||
} | ||
|
||
@ManyToMany(cascade = CascadeType.PERSIST, fetch = FetchType.EAGER) | ||
@JoinTable( | ||
name = "USER_ROLE", | ||
joinColumns = @JoinColumn(name = "USER_ID"), | ||
inverseJoinColumns = @JoinColumn(name = "ROLE_ID")) | ||
@Fetch(FetchMode.SUBSELECT) | ||
public Set<Role> getRoles() { | ||
return roles; | ||
} | ||
|
||
public void setRoles(Set<Role> roles) { | ||
this.roles = roles; | ||
} | ||
|
||
public void addRole(Role role) { | ||
if ( roles == null ) { | ||
roles = new HashSet<>(); | ||
} | ||
roles.add( role ); | ||
} | ||
|
||
} | ||
|
||
@Entity(name = "Role") | ||
@Table(name = "ROLE_TABLE") | ||
public static class Role { | ||
|
||
private Integer id; | ||
private String name; | ||
|
||
public Role() { | ||
} | ||
|
||
public Role(Integer id, String name) { | ||
this.id = id; | ||
this.name = name; | ||
} | ||
|
||
@Id | ||
@Column(name = "ROLE_ID") | ||
public Integer getId() { | ||
return id; | ||
} | ||
|
||
public void setId(Integer roleId) { | ||
this.id = roleId; | ||
} | ||
|
||
@Column(name = "ROLE_NAME") | ||
public String getName() { | ||
return name; | ||
} | ||
|
||
public void setName(String roleName) { | ||
this.name = roleName; | ||
} | ||
|
||
} | ||
|
||
@Entity(name = "Agency") | ||
@Table(name = "AGENCY_TABLE") | ||
public static class Agency { | ||
|
||
private Integer id; | ||
private String name; | ||
private Set<User> users; | ||
|
||
public Agency() { | ||
} | ||
|
||
public Agency(Integer id, String name) { | ||
this.id = id; | ||
this.name = name; | ||
} | ||
|
||
@Id | ||
@Column(name = "AGENCY_ID") | ||
public Integer getId() { | ||
return id; | ||
} | ||
|
||
public void setId(Integer roleId) { | ||
this.id = roleId; | ||
} | ||
|
||
@Column(name = "AGENCY_NAME") | ||
public String getName() { | ||
return name; | ||
} | ||
|
||
public void setName(String roleName) { | ||
this.name = roleName; | ||
} | ||
|
||
@OneToMany(cascade = CascadeType.ALL, fetch = FetchType.EAGER, mappedBy = "agency") | ||
@Fetch(FetchMode.SELECT) | ||
public Set<User> getUsers() { | ||
return users; | ||
} | ||
|
||
public void setUsers(Set<User> users) { | ||
this.users = users; | ||
} | ||
|
||
public void addUser(User user) { | ||
if ( users == null ) { | ||
users = new HashSet<>(); | ||
} | ||
users.add( user ); | ||
user.setAgency( this ); | ||
} | ||
} | ||
} |