Added cancel_shift_for_a_talent unit test

Author: masoudarvishian

src/main/java/com/zenjob/challenge/repository/ShiftRepository.java

@@ -2,6 +2,8 @@
 
 import com.zenjob.challenge.entity.Shift;
 import org.springframework.data.jpa.repository.JpaRepository;
+import org.springframework.data.jpa.repository.Query;
+import org.springframework.data.repository.query.Param;
 import org.springframework.stereotype.Repository;
 
 import java.util.List;
@@ -10,4 +12,5 @@
 @Repository
 public interface ShiftRepository extends JpaRepository<Shift, UUID> {
     List<Shift> findAllByJobId(UUID jobId);
+    List<Shift> findAllByTalentId(UUID talentId);
 }

src/main/java/com/zenjob/challenge/service/IJobService.java

@@ -16,4 +16,5 @@ public interface IJobService {
     Optional<Job> getJob(UUID id);
     void cancelShift(UUID companyId, UUID shiftId);
     Optional<Shift> getShift(UUID id);
+    void cancelShiftForTalent(UUID companyId, UUID talentId, UUID shiftId);
 }

src/main/java/com/zenjob/challenge/service/JobService.java

@@ -56,9 +56,9 @@ public List<Shift> getShifts(UUID id) {
         return shiftRepository.findAllByJobId(id);
     }
 
-    public void bookTalent(UUID talent, UUID shiftId) {
+    public void bookTalent(UUID talentId, UUID shiftId) {
         Optional<Shift> shiftById = shiftRepository.findById(shiftId);
-        shiftById.map(shift -> shiftRepository.save(shift.setTalentId(talent)));
+        shiftById.map(shift -> shiftRepository.save(shift.setTalentId(talentId)));
     }
 
     @Override
@@ -88,4 +88,20 @@ public void cancelShift(UUID companyId, UUID shiftId) {
     public Optional<Shift> getShift(UUID id) {
         return shiftRepository.findById(id);
     }
+
+    @Override
+    public void cancelShiftForTalent(UUID companyId, UUID talentId, UUID shiftId) {
+        List<Shift> shifts = getShiftsByTalentIdAndCompanyId(talentId, companyId);
+        shifts.forEach(shift -> {
+            shift.setTalentId(null);
+            shiftRepository.save(shift);
+        });
+    }
+
+    private List<Shift> getShiftsByTalentIdAndCompanyId(UUID talentId, UUID companyId) {
+        List<Shift> shifts = shiftRepository.findAllByTalentId(talentId);
+        return shifts.stream()
+                .filter(shift -> companyId.equals(shift.getJob().getCompanyId()))
+                .collect(Collectors.toList());
+    }
 }

src/test/java/com/zenjob/challenge/JobServiceTests.java

@@ -15,6 +15,7 @@
 import java.time.LocalDate;
 import java.time.ZoneId;
 import java.time.ZonedDateTime;
+import java.util.Optional;
 import java.util.UUID;
 
 @SpringBootTest
@@ -146,4 +147,26 @@ public void a_company_can_only_cancel_its_own_shifts() {
         Assertions.assertThrows(InvalidActionException.class, () ->
                 jobService.cancelShift(companyId, firstShift.getId()));
     }
+
+    @Test
+    public void cancel_shift_for_a_talent() {
+        // 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();
+        jobService.bookTalent(talentId, firstShift.getId());
+        jobService.bookTalent(talentId, secondShift.getId());
+
+        // when
+        jobService.cancelShiftForTalent(job.getCompanyId(), talentId, firstShift.getId());
+
+        // then
+        Optional<Shift> firstShiftById = jobService.getShift(firstShift.getId());
+        Optional<Shift> secondShiftById = jobService.getShift(secondShift.getId());
+        Assertions.assertNull(firstShiftById.get().getTalentId());
+        Assertions.assertNull(secondShiftById.get().getTalentId());
+    }
 }