为 Java DAO 类中的方法创建测试

Question

For the following Java class DAO I created in InteliJ, I need to create a test that:

"Returns two reservations for test parkId 1."

I need help with writing the logic itself for a possible test

Here is the method:

@Override
public List<UpcomingReservation> getAllUpcomingReservations() {
    List<UpcomingReservation> reservations = new ArrayList<>();
    String sql = "SELECT * FROM reservation WHERE from_date >= CURRENT_DATE AND from_date <= CURRENT_DATE + 30";
    SqlRowSet results = jdbcTemplate.queryForRowSet(sql);
    while (results.next()) {
        reservations.add(mapRowToUpcomingReservation(results));
    }
    return reservations;
}

Answer 1

The correct SQL query was supposed to be as follows which would have included the parkID:

}
public List<Reservation> getAllReservationsNext30Days(int parkID) {
    List<Reservation> reservations = new ArrayList<>();
    String sql = "SELECT * FROM park JOIN campground ON park.park_id = campground.park_id JOIN site ON campground.campground_id = site.campground_id JOIN reservation ON site.site_id = reservation.site_id WHERE from_date >= CURRENT_DATE AND from_date <= CURRENT_DATE + 30";
    SqlRowSet results = jdbcTemplate.queryForRowSet(sql);
    while (results.next()) {
        reservations.add(mapRowToReservation(results));
    }
    return reservations;
}

And the correct test I was trying to create was:

@Test
public void getUpcomingReservations_Should_ReturnNext30DayReservations(){
    List<Reservation> upcomingReservation = dao.getAllReservationsNext30Days(1);
    assertEquals(2, upcomingReservation.size());
}