fix: 더미데이터 사용하도록 테스트 수정

codesquad-members-2021 · May 31, 2021 · bf8b057 · bf8b057
1 parent 3cf2ad9
commit bf8b057
Showing 1 changed file with 4 additions and 10 deletions.
diff --git a/...rc/test/java/com/codesquad/airbnb/accommodation/service/AccommodationServiceImplTest.java b/...rc/test/java/com/codesquad/airbnb/accommodation/service/AccommodationServiceImplTest.java
@@ -17,7 +17,6 @@
 
 import java.time.LocalDate;
 import java.util.List;
-import java.util.stream.Collectors;
 import java.util.stream.Stream;
 
 import static org.assertj.core.api.BDDAssertions.then;
@@ -52,21 +51,16 @@ static Stream<Arguments> readAllProvider() {
         return Stream.of(
                 Arguments.of(
                         AccommodationRequest.builder().build(),
-                        AccommodationDummyDataFactory.listWithId().stream()
-                                .collect(Collectors.toList()),
-                        AccommodationResponseDummyDataFactory.listWithIdTypeOneNight().stream()
-                                .collect(Collectors.toList())
+                        AccommodationDummyDataFactory.listWithId(),
+                        AccommodationResponseDummyDataFactory.listWithIdTypeOneNight()
                 ),
                 Arguments.of(
                         AccommodationRequest.builder()
                                 .checkinDate(LocalDate.now())
                                 .checkoutDate(LocalDate.now().plusDays(2))
                                 .build(),
-                        AccommodationDummyDataFactory.listWithId().stream()
-                                .collect(Collectors.toList()),
-                        AccommodationDummyDataFactory.listWithId().stream()
-                                .map(accommodation -> AccommodationResponse.of(accommodation, 2))
-                                .collect(Collectors.toList())
+                        AccommodationDummyDataFactory.listWithId(),
+                        AccommodationResponseDummyDataFactory.listWithIdTypeTwoNights()
                 )
         );
     }