From 5586a74acdd4f0026186c8dc08b5d70deaed96c4 Mon Sep 17 00:00:00 2001
From: AbdallahSawan <abdallahsawan.e@gmail.com>
Date: Tue, 16 Feb 2021 15:47:38 +0200
Subject: [PATCH] Usage of the Hibernate @LazyCollection Annotation Article by
 Abdallah Sawan

---
 .../{ => lazyCollection}/model/Branch.java     |  0
 .../{ => lazyCollection}/model/Employee.java   | 13 ++++++-------
 .../LazyCollectionIntegrationTest.java         | 18 +++++++++---------
 3 files changed, 15 insertions(+), 16 deletions(-)
 rename persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/{ => lazyCollection}/model/Branch.java (100%)
 rename persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/{ => lazyCollection}/model/Employee.java (83%)

diff --git a/persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/model/Branch.java b/persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/lazyCollection/model/Branch.java
similarity index 100%
rename from persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/model/Branch.java
rename to persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/lazyCollection/model/Branch.java
diff --git a/persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/model/Employee.java b/persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/lazyCollection/model/Employee.java
similarity index 83%
rename from persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/model/Employee.java
rename to persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/lazyCollection/model/Employee.java
index 29c89bb94a..05949334c9 100644
--- a/persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/model/Employee.java
+++ b/persistence-modules/hibernate-annotations/src/main/java/com/baeldung/hibernate/lazyCollection/model/Employee.java
@@ -11,15 +11,14 @@ public class Employee {
 
     private String name;
 
-    private Long rank;
+    private String address;
 
     public Employee() {
 
     }
 
-    public Employee(String name, Long rank, Branch mainBranch, Branch subBranch, Branch additionalBranch) {
+    public Employee(String name, Branch mainBranch, Branch subBranch, Branch additionalBranch) {
         this.name = name;
-        this.rank = rank;
         this.mainBranch = mainBranch;
         this.subBranch = subBranch;
         this.additionalBranch = additionalBranch;
@@ -50,12 +49,12 @@ public class Employee {
         this.name = name;
     }
 
-    public Long getRank() {
-        return rank;
+    public String getAddress() {
+        return address;
     }
 
-    public void setRank(Long rank) {
-        this.rank = rank;
+    public void setAddress(String address) {
+        this.address = address;
     }
 
     public Branch getMainBranch() {
diff --git a/persistence-modules/hibernate-annotations/src/test/java/com/baeldung/hibernate/lazyCollection/LazyCollectionIntegrationTest.java b/persistence-modules/hibernate-annotations/src/test/java/com/baeldung/hibernate/lazyCollection/LazyCollectionIntegrationTest.java
index d5eb9624b2..cec57a3dbf 100644
--- a/persistence-modules/hibernate-annotations/src/test/java/com/baeldung/hibernate/lazyCollection/LazyCollectionIntegrationTest.java
+++ b/persistence-modules/hibernate-annotations/src/test/java/com/baeldung/hibernate/lazyCollection/LazyCollectionIntegrationTest.java
@@ -51,25 +51,25 @@ public class LazyCollectionIntegrationTest {
 
         session.save(branch);
 
-        Employee mainEmployee1 = new Employee("main employee 1", 1L,  branch, null, null);
-        Employee mainEmployee2 = new Employee("main employee 2", 2L, branch, null, null);
-        Employee mainEmployee3 = new Employee("main employee 3", 3L, branch, null, null);
+        Employee mainEmployee1 = new Employee("main employee 1", branch, null, null);
+        Employee mainEmployee2 = new Employee("main employee 2", branch, null, null);
+        Employee mainEmployee3 = new Employee("main employee 3", branch, null, null);
 
         session.save(mainEmployee1);
         session.save(mainEmployee2);
         session.save(mainEmployee3);
 
-        Employee subEmployee1 = new Employee("sub employee 1", 1L, null, branch, null);
-        Employee subEmployee2 = new Employee("sub employee 2", 2L, null, branch, null);
-        Employee subEmployee3 = new Employee("sub employee 3", 3L, null, branch, null);
+        Employee subEmployee1 = new Employee("sub employee 1", null, branch, null);
+        Employee subEmployee2 = new Employee("sub employee 2", null, branch, null);
+        Employee subEmployee3 = new Employee("sub employee 3", null, branch, null);
 
         session.save(subEmployee1);
         session.save(subEmployee2);
         session.save(subEmployee3);
 
-        Employee additionalEmployee1 = new Employee("additional employee 1", 1L, null, null, branch);
-        Employee additionalEmployee2 = new Employee("additional employee 2", 2L, null, null, branch);
-        Employee additionalEmployee3 = new Employee("additional employee 3", 3L, null, null, branch);
+        Employee additionalEmployee1 = new Employee("additional employee 1", null, null, branch);
+        Employee additionalEmployee2 = new Employee("additional employee 2", null, null, branch);
+        Employee additionalEmployee3 = new Employee("additional employee 3", null, null, branch);
 
         session.save(additionalEmployee1);
         session.save(additionalEmployee2);