feat : 예약 테스트용 데이터 추가

dev-hooon · Jan 9, 2024 · 3a30484 · 3a30484
1 parent 3b28425
commit 3a30484
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/test/java/com/prgrms/catchtable/reservation/fixture/ReservationFixture.java b/src/test/java/com/prgrms/catchtable/reservation/fixture/ReservationFixture.java
@@ -3,6 +3,7 @@
 import static com.prgrms.catchtable.reservation.domain.ReservationStatus.COMPLETED;
 
 import com.prgrms.catchtable.common.data.shop.ShopData;
+import com.prgrms.catchtable.member.domain.Member;
 import com.prgrms.catchtable.reservation.domain.Reservation;
 import com.prgrms.catchtable.reservation.domain.ReservationStatus;
 import com.prgrms.catchtable.reservation.domain.ReservationTime;
@@ -95,4 +96,16 @@ public static ModifyReservationStatusRequest getModifyReservationStatusRequest(
             .build();
     }
 
+    public static Reservation getReservation(ReservationTime reservationTime, Member member) {
+        if (!reservationTime.isOccupied()) {
+            reservationTime.setOccupiedTrue();
+        }
+        return Reservation.builder()
+            .status(COMPLETED)
+            .peopleCount(4)
+            .reservationTime(reservationTime)
+            .member(member)
+            .build();
+    }
+
 }