diff --git a/src/test/java/HairSalonTest.java b/src/test/java/HairSalonTest.java index 957fe01..0450ae6 100644 --- a/src/test/java/HairSalonTest.java +++ b/src/test/java/HairSalonTest.java @@ -1,4 +1,6 @@ import org.junit.jupiter.api.Test; +import org.junit.jupiter.params.ParameterizedTest; +import org.junit.jupiter.params.provider.EnumSource; import static org.junit.jupiter.api.Assertions.*; import static org.junit.jupiter.api.Assumptions.assumeTrue; @@ -16,14 +18,16 @@ public void testGetOpeningDays() { assertArrayEquals(expected, actual); } - @Test - public void testSalonIsOpenOnMonday() { - assertTrue(hairSalon.isOpen(HairSalon.Day.MONDAY)); + @ParameterizedTest + @EnumSource(value = HairSalon.Day.class, names = { "MONDAY", "TUESDAY" }) + public void testSalonIsOpenOnMonday(HairSalon.Day day) { + assertTrue(hairSalon.isOpen(day)); } - @Test - public void testSalonIsClosedOnWednesday() { - assertFalse(hairSalon.isOpen(HairSalon.Day.WEDNESDAY)); + @ParameterizedTest + @EnumSource(value = HairSalon.Day.class, names = { "WEDNESDAY", "THURSDAY", "FRIDAY", "SATURDAY", "SUNDAY" }) + public void testSalonIsClosed(HairSalon.Day day) { + assertFalse(hairSalon.isOpen(day)); } @Test