Split JobServiceTests.java and added ShiftServiceTests

masoudarvishian · May 10, 2024 · 56937ea · 56937ea
1 parent 6350dae
commit 56937ea
Show file tree

Hide file tree

Showing 2 changed files with 77 additions and 52 deletions.
diff --git a/src/test/java/com/zenjob/challenge/application/services/JobServiceTests.java b/src/test/java/com/zenjob/challenge/application/services/JobServiceTests.java
@@ -17,7 +17,6 @@
 import java.time.LocalDate;
 import java.time.ZoneId;
 import java.time.ZonedDateTime;
-import java.util.Optional;
 import java.util.UUID;
 
 @SpringBootTest
@@ -123,55 +122,4 @@ public void a_company_can_only_cancel_its_own_jobs() {
         Assertions.assertThrows(InvalidActionException.class, () ->
                 jobService.cancelJob(companyId, job.getId()));
     }
-
-    @Test
-    public void cancel_a_single_shift_by_company() throws NotFoundException {
-        // given
-        LocalDate startDate = LocalDate.now();
-        LocalDate endDate = LocalDate.now().plusDays(5);
-        Job job = jobService.createJob(UUID.randomUUID(), startDate, endDate);
-        Shift firstShift = job.getShifts().get(0);
-
-        // when
-        shiftService.cancelShift(job.getCompanyId(), firstShift.getId());
-
-        // then
-        Assertions.assertFalse(shiftService.getShift(firstShift.getId()).isPresent());
-    }
-
-    @Test
-    public void a_company_can_only_cancel_its_own_shifts() {
-        // given
-        LocalDate startDate = LocalDate.now();
-        LocalDate endDate = LocalDate.now().plusDays(5);
-        Job job = jobService.createJob(UUID.randomUUID(), startDate, endDate);
-        Shift firstShift = job.getShifts().get(0);
-        UUID companyId = UUID.randomUUID();
-
-        // when - then
-        Assertions.assertThrows(InvalidActionException.class, () ->
-                shiftService.cancelShift(companyId, firstShift.getId()));
-    }
-
-    @Test
-    public void cancel_shift_for_a_talent() throws NotFoundException {
-        // given
-        LocalDate startDate = LocalDate.now();
-        LocalDate endDate = LocalDate.now().plusDays(5);
-        Job job = jobService.createJob(UUID.randomUUID(), startDate, endDate);
-        Shift firstShift = job.getShifts().get(0);
-        Shift secondShift = job.getShifts().get(1);
-        UUID talentId = UUID.randomUUID();
-        shiftService.bookTalent(talentId, firstShift.getId());
-        shiftService.bookTalent(talentId, secondShift.getId());
-
-        // when
-        shiftService.cancelShiftForTalent(job.getCompanyId(), talentId);
-
-        // then
-        Optional<Shift> firstShiftById = shiftService.getShift(firstShift.getId());
-        Optional<Shift> secondShiftById = shiftService.getShift(secondShift.getId());
-        Assertions.assertNull(firstShiftById.get().getTalentId());
-        Assertions.assertNull(secondShiftById.get().getTalentId());
-    }
 }
diff --git a/src/test/java/com/zenjob/challenge/application/services/ShiftServiceTests.java b/src/test/java/com/zenjob/challenge/application/services/ShiftServiceTests.java
@@ -0,0 +1,77 @@
+package com.zenjob.challenge.application.services;
+
+import com.zenjob.challenge.application.interfaces.JobService;
+import com.zenjob.challenge.application.interfaces.ShiftService;
+import com.zenjob.challenge.domain.entity.Job;
+import com.zenjob.challenge.domain.entity.Shift;
+import com.zenjob.challenge.domain.exceptions.InvalidActionException;
+import javassist.NotFoundException;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.Test;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.context.SpringBootTest;
+
+import java.time.LocalDate;
+import java.util.Optional;
+import java.util.UUID;
+
+@SpringBootTest
+public class ShiftServiceTests {
+
+    @Autowired
+    private JobService jobService;
+
+    @Autowired
+    private ShiftService shiftService;
+
+    @Test
+    public void cancel_a_single_shift_by_company() throws NotFoundException {
+        // given
+        LocalDate startDate = LocalDate.now();
+        LocalDate endDate = LocalDate.now().plusDays(5);
+        Job job = jobService.createJob(UUID.randomUUID(), startDate, endDate);
+        Shift firstShift = job.getShifts().get(0);
+
+        // when
+        shiftService.cancelShift(job.getCompanyId(), firstShift.getId());
+
+        // then
+        Assertions.assertFalse(shiftService.getShift(firstShift.getId()).isPresent());
+    }
+
+    @Test
+    public void a_company_can_only_cancel_its_own_shifts() {
+        // given
+        LocalDate startDate = LocalDate.now();
+        LocalDate endDate = LocalDate.now().plusDays(5);
+        Job job = jobService.createJob(UUID.randomUUID(), startDate, endDate);
+        Shift firstShift = job.getShifts().get(0);
+        UUID companyId = UUID.randomUUID();
+
+        // when - then
+        Assertions.assertThrows(InvalidActionException.class, () ->
+                shiftService.cancelShift(companyId, firstShift.getId()));
+    }
+
+    @Test
+    public void cancel_shift_for_a_talent() throws NotFoundException {
+        // given
+        LocalDate startDate = LocalDate.now();
+        LocalDate endDate = LocalDate.now().plusDays(5);
+        Job job = jobService.createJob(UUID.randomUUID(), startDate, endDate);
+        Shift firstShift = job.getShifts().get(0);
+        Shift secondShift = job.getShifts().get(1);
+        UUID talentId = UUID.randomUUID();
+        shiftService.bookTalent(talentId, firstShift.getId());
+        shiftService.bookTalent(talentId, secondShift.getId());
+
+        // when
+        shiftService.cancelShiftForTalent(job.getCompanyId(), talentId);
+
+        // then
+        Optional<Shift> firstShiftById = shiftService.getShift(firstShift.getId());
+        Optional<Shift> secondShiftById = shiftService.getShift(secondShift.getId());
+        Assertions.assertNull(firstShiftById.get().getTalentId());
+        Assertions.assertNull(secondShiftById.get().getTalentId());
+    }
+}