72 lines
2.5 KiB
Java
72 lines
2.5 KiB
Java
package com.baeldung.sqlresultsetmapping;
|
|
|
|
import static org.junit.Assert.*;
|
|
|
|
import org.junit.jupiter.api.*;
|
|
|
|
import javax.persistence.EntityManager;
|
|
import javax.persistence.EntityManagerFactory;
|
|
import javax.persistence.Persistence;
|
|
import javax.persistence.Query;
|
|
import java.util.Collections;
|
|
import java.util.List;
|
|
|
|
public class SqlResultSetMappingUnitTest {
|
|
|
|
private static EntityManager em;
|
|
private static EntityManagerFactory emFactory;
|
|
|
|
@BeforeAll
|
|
public static void setup() {
|
|
emFactory = Persistence.createEntityManagerFactory("java-jpa-scheduled-day");
|
|
em = emFactory.createEntityManager();
|
|
}
|
|
|
|
@Test
|
|
public void whenNamedQuery_thenColumnResult() {
|
|
List<Long> employeeIds = em.createNamedQuery("FridayEmployees").getResultList();
|
|
assertEquals(2, employeeIds.size());
|
|
}
|
|
|
|
@Test
|
|
public void whenNamedQuery_thenConstructorResult() {
|
|
List<ScheduledDay> scheduleDays = Collections.checkedList(em.createNamedQuery("Schedules", ScheduledDay.class).getResultList(), ScheduledDay.class);
|
|
assertEquals(3, scheduleDays.size());
|
|
assertTrue(scheduleDays.stream().allMatch(c -> c.getHourIn().longValue() == 8));
|
|
}
|
|
|
|
@Test
|
|
public void whenNamedQuery_thenSingleEntityResult() {
|
|
List<Employee> employees = Collections.checkedList(em.createNamedQuery("Employees").getResultList(), Employee.class);
|
|
assertEquals(3, employees.size());
|
|
assertTrue(employees.stream().allMatch(c -> c.getClass() == Employee.class));
|
|
}
|
|
|
|
@Test
|
|
public void whenNamedQuery_thenMultipleEntityResult() {
|
|
final Query query = em.createNativeQuery("SELECT e.id, e.name, d.id, d.employeeId, "
|
|
+ " d.dayOfWeek, d.hourIn, d.hourOut "
|
|
+ " FROM employee e, schedule_days d "
|
|
+ " WHERE e.id = d.employeeId", "EmployeeScheduleResults");
|
|
List<Object[]> results = query.getResultList();
|
|
assertEquals(4, results.size());
|
|
assertTrue(results.get(0).length == 2);
|
|
|
|
Employee emp = (Employee) results.get(1)[0];
|
|
ScheduledDay day = (ScheduledDay) results.get(1)[1];
|
|
|
|
assertTrue(day.getEmployeeId() == emp.getId());
|
|
}
|
|
|
|
@AfterAll
|
|
public static void destroy() {
|
|
|
|
if (em != null) {
|
|
em.close();
|
|
}
|
|
if (emFactory != null) {
|
|
emFactory.close();
|
|
}
|
|
}
|
|
}
|