Skip to content

Commit 2aa2b39

Browse files
committed
feat: update faker
Signed-off-by: Otavio Santana <[email protected]>
1 parent 7d31d49 commit 2aa2b39

File tree

1 file changed

+15
-5
lines changed

1 file changed

+15
-5
lines changed

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

Lines changed: 15 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
package org.soujava.demos.mongodb.document;
22

33
import jakarta.inject.Inject;
4+
import net.datafaker.Faker;
45
import org.assertj.core.api.SoftAssertions;
56
import org.eclipse.jnosql.databases.mongodb.mapping.MongoDBTemplate;
67
import org.eclipse.jnosql.mapping.Database;
@@ -22,6 +23,7 @@
2223
import org.junit.jupiter.params.provider.MethodSource;
2324

2425
import java.util.List;
26+
import java.util.concurrent.ThreadLocalRandom;
2527
import java.util.stream.Stream;
2628

2729

@@ -38,6 +40,8 @@ class RoomServiceTest {
3840
@Inject
3941
private RoomRepository repository;
4042

43+
private static final Faker FAKER = new Faker();
44+
4145
@BeforeEach
4246
void setUP() {
4347

@@ -178,13 +182,19 @@ void shouldFindRoomsNeedingCleaning() {
178182

179183
static Stream<Arguments> room() {
180184
Room room = new RoomBuilder()
181-
.roomNumber(101)
182-
.type(RoomType.VIP_SUITE)
183-
.status(RoomStatus.AVAILABLE)
184-
.cleanStatus(CleanStatus.CLEAN)
185-
.smokingAllowed(false)
185+
.roomNumber(FAKER.number().numberBetween(100, 999))
186+
.type(randomEnum(RoomType.class))
187+
.status(randomEnum(RoomStatus.class))
188+
.cleanStatus(randomEnum(CleanStatus.class))
189+
.smokingAllowed(FAKER.bool().bool())
186190
.build();
187191

188192
return Stream.of(Arguments.of(room));
189193
}
194+
195+
private static <T extends Enum<?>> T randomEnum(Class<T> enumClass) {
196+
T[] constants = enumClass.getEnumConstants();
197+
int index = ThreadLocalRandom.current().nextInt(constants.length);
198+
return constants[index];
199+
}
190200
}

0 commit comments

Comments
 (0)