test: update room repository

Signed-off-by: Otavio Santana <[email protected]>

Author: otaviojava

src/main/java/org/soujava/demos/mongodb/document/RoomRepository.java

@@ -1,6 +1,7 @@
 package org.soujava.demos.mongodb.document;
 
 import jakarta.data.repository.Delete;
+import jakarta.data.repository.Param;
 import jakarta.data.repository.Query;
 import jakarta.data.repository.Repository;
 import jakarta.data.repository.Save;
@@ -26,4 +27,7 @@ public interface RoomRepository {
     void save(List<Room> rooms);
 
     void deleteBy();
+
+    @Query("WHERE type = :type")
+    List<Room> findByType(@Param("type") RoomType type);
 }

src/test/java/org/soujava/demos/mongodb/document/RoomServiceTest.java

@@ -16,6 +16,8 @@
 import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.EnumSource;
 
 import java.util.List;
 
@@ -146,4 +148,11 @@ void shouldFindRoomsNeedingCleaning() {
            softly.assertThat(rooms).allMatch(room -> !room.getStatus().equals(RoomStatus.OUT_OF_SERVICE));
        });
    }
+
+   @ParameterizedTest(name = "should find rooms by type {0}")
+   @EnumSource(RoomType.class)
+   void shouldFindRoomByType(RoomType type) {
+         List<Room> rooms = this.repository.findByType(type);
+         SoftAssertions.assertSoftly(softly -> softly.assertThat(rooms).allMatch(room -> room.getType().equals(type)));
+   }
 }