Skip to content

Commit 0fe0943

Browse files
committed
refactor: replace stream filtering with findByNumber in HotelRoomSteps for clarity
1 parent c8d7d3a commit 0fe0943

File tree

1 file changed

+3
-8
lines changed

1 file changed

+3
-8
lines changed

src/test/java/org/soujava/demos/mongodb/HotelRoomSteps.java

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -70,10 +70,7 @@ public void a_room_with_number_is_registered_as(Integer number, String statusNam
7070
@When("I mark the room {int} as {word}")
7171
public void i_mark_the_room_as(Integer number, String newStatusName) {
7272
RoomStatus newStatus = RoomStatus.valueOf(newStatusName);
73-
Optional<Room> roomOpt = repository.findAll()
74-
.stream()
75-
.filter(r -> r.getNumber() == number)
76-
.findFirst();
73+
Optional<Room> roomOpt = repository.findByNumber(number);
7774

7875
Assertions.assertThat(roomOpt)
7976
.as("Room %s should exist", number)
@@ -88,10 +85,8 @@ public void i_mark_the_room_as(Integer number, String newStatusName) {
8885
@Then("the room {int} should be marked as {word}")
8986
public void the_room_should_be_marked_as(Integer number, String expectedStatusName) {
9087
RoomStatus expectedStatus = RoomStatus.valueOf(expectedStatusName);
91-
Optional<Room> roomOpt = repository.findAll()
92-
.stream()
93-
.filter(r -> r.getNumber() == number)
94-
.findFirst();
88+
List<Room> all = repository.findAll();
89+
Optional<Room> roomOpt = repository.findByNumber(number);
9590

9691
Assertions.assertThat(roomOpt)
9792
.as("Room %s should exist", number)

0 commit comments

Comments
 (0)