Code related to the mini-article BAEL-3386

persistence-modules/spring-data-jpa-4/src/test/java/com/baeldung/elementcollection/ElementCollectionIntegrationTest.java

@@ -0,0 +1,63 @@
+package com.baeldung.elementcollection;
+
+import com.baeldung.elementcollection.model.Employee;
+import com.baeldung.elementcollection.model.Phone;
+import com.baeldung.elementcollection.repository.EmployeeRepository;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+import org.springframework.test.context.junit4.SpringRunner;
+import org.springframework.transaction.annotation.Transactional;
+
+import java.util.Arrays;
+
+import static org.hamcrest.core.Is.is;
+import static org.junit.Assert.assertThat;
+
+@RunWith(SpringRunner.class)
+@SpringBootTest(classes = ElementCollectionApplication.class)
+public class ElementCollectionIntegrationTest {
+
+    @Autowired
+    private EmployeeRepository employeeRepository;
+
+    @Before
+    public void init() {
+        Employee employee = new Employee(1, "Fred");
+        employee.setPhones(Arrays.asList(new Phone("work", "+55", "99999-9999"), new Phone("home", "+55", "98888-8888")));
+        employeeRepository.save(employee);
+    }
+
+    @After
+    public void clean() {
+        employeeRepository.remove(1);
+    }
+
+    @Test(expected = org.hibernate.LazyInitializationException.class)
+    public void whenAccessLazyCollection_thenThrowLazyInitializationException() {
+        Employee employee = employeeRepository.findById(1);
+        assertThat(employee.getPhones().size(), is(2));
+    }
+
+    @Test
+    public void whenUseJPAQL_thenFetchResult() {
+        Employee employee = employeeRepository.findByJPQL(1);
+        assertThat(employee.getPhones().size(), is(2));
+    }
+
+    @Test
+    public void whenUseEntityGraph_thenFetchResult() {
+        Employee employee = employeeRepository.findByEntityGraph(1);
+        assertThat(employee.getPhones().size(), is(2));
+    }
+
+    @Test
+    @Transactional
+    public void whenUseTransaction_thenFetchResult() {
+        Employee employee = employeeRepository.findById(1);
+        assertThat(employee.getPhones().size(), is(2));
+    }
+}