test: add test

f-lab-clone · Nov 1, 2023 · be5fb98 · be5fb98
1 parent aaf384e
commit be5fb98
Show file tree

Hide file tree

Showing 4 changed files with 49 additions and 13 deletions.
diff --git a/src/integrationTest/kotlin/com/group4/ticketingservice/Reservation/ReservationTest.kt b/src/integrationTest/kotlin/com/group4/ticketingservice/Reservation/ReservationTest.kt
@@ -1,20 +1,26 @@
 package com.group4.ticketingservice.Reservation
 
 import com.group4.ticketingservice.AbstractIntegrationTest
+import com.group4.ticketingservice.dto.QueueResponseDTO
 import com.group4.ticketingservice.entity.Event
 import com.group4.ticketingservice.entity.User
 import com.group4.ticketingservice.repository.EventRepository
 import com.group4.ticketingservice.repository.ReservationRepository
 import com.group4.ticketingservice.repository.UserRepository
 import com.group4.ticketingservice.service.ReservationService
+import io.mockk.every
+import io.mockk.mockk
 import org.assertj.core.api.Assertions.assertThat
 import org.junit.jupiter.api.AfterEach
 import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.RepeatedTest
 import org.junit.jupiter.api.Test
 import org.springframework.beans.factory.annotation.Autowired
+import org.springframework.http.HttpMethod
+import org.springframework.http.ResponseEntity
 import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder
 import org.springframework.test.context.TestPropertySource
+import org.springframework.web.client.RestTemplate
 import java.time.Duration.ofHours
 import java.time.OffsetDateTime
 import java.util.concurrent.CountDownLatch
@@ -50,6 +56,12 @@ class ReservationTest @Autowired constructor(
         maxAttendees = 100
     )
 
+    private val queueSuccess: QueueResponseDTO = QueueResponseDTO(
+        status = true,
+        message = "",
+        data = null
+    )
+
     @BeforeEach fun addUserAndEvent() {
         userRepository.save(sampleUser)
         eventRepository.save(sampleEvent)
@@ -67,6 +79,9 @@ class ReservationTest @Autowired constructor(
     @RepeatedTest(3)
     @Test
     fun `ReservationService_createReservation should not exceed the limit in the concurrency test`() {
+        val restTemplate: RestTemplate = mockk()
+        every { restTemplate.exchange(any() as String, HttpMethod.DELETE, null, QueueResponseDTO::class.java) } returns ResponseEntity.ok(queueSuccess)
+        reservationService.restTemplate = restTemplate
         val threadCount = 1000
         val executorService = Executors.newFixedThreadPool(32)
         val countDownLatch = CountDownLatch(threadCount)

diff --git a/src/main/kotlin/com/group4/ticketingservice/dto/QueueDTO.kt b/src/main/kotlin/com/group4/ticketingservice/dto/QueueDTO.kt
@@ -1,16 +1,17 @@
 package com.group4.ticketingservice.dto
 
-class QueueResponseDTO {
-    val status: Boolean = false
-    val message: String = ""
+class QueueResponseDTO(
+    val status: Boolean = false,
+    val message: String = "",
     val data: TicketInfo? = null
-}
-class TicketInfo {
-    var eventId: String = ""
-    var userId: String = ""
-    var isWaiting = true
-    var offset = 0
-}
+)
+
+class TicketInfo(
+    var eventId: String = "",
+    var userId: String = "",
+    var isWaiting: Boolean = true,
+    var offset: Int = 0
+)
 data class TicketRequest(
     val eventId: Int,
     val userId: Int

diff --git a/src/main/kotlin/com/group4/ticketingservice/service/ReservationService.kt b/src/main/kotlin/com/group4/ticketingservice/service/ReservationService.kt
@@ -24,7 +24,7 @@ class ReservationService @Autowired constructor(
     private val userRepository: UserRepository,
     private val eventRepository: EventRepository,
     private val reservationRepository: ReservationRepository,
-    private val restTemplate: RestTemplate,
+    var restTemplate: RestTemplate,
     @Value("\${ticketing.queue.server.url}")
     private val queueServerURL: String
 

diff --git a/src/test/kotlin/com/group4/ticketingservice/reservation/ReservationServiceTest.kt b/src/test/kotlin/com/group4/ticketingservice/reservation/ReservationServiceTest.kt
@@ -1,5 +1,6 @@
 package com.group4.ticketingservice.reservation
 
+import com.group4.ticketingservice.dto.QueueResponseDTO
 import com.group4.ticketingservice.entity.Event
 import com.group4.ticketingservice.entity.Reservation
 import com.group4.ticketingservice.entity.User
@@ -14,6 +15,10 @@ import io.mockk.mockk
 import io.mockk.verify
 import org.junit.jupiter.api.Test
 import org.junit.jupiter.api.assertThrows
+import org.springframework.http.HttpMethod
+import org.springframework.http.HttpStatus
+import org.springframework.http.ResponseEntity
+import org.springframework.web.client.HttpClientErrorException
 import org.springframework.web.client.RestTemplate
 import java.time.Duration.ofHours
 import java.time.OffsetDateTime
@@ -32,7 +37,6 @@ class ReservationServiceTest() {
         queueServerURL = "url",
         restTemplate = restTemplate
     )
-
     val sampleUserId = 1
 
     val sampleUser = User(
@@ -67,13 +71,18 @@ class ReservationServiceTest() {
         user = sampleUser,
         event = sampleEvent
     )
+    private val queueSuccess: QueueResponseDTO = QueueResponseDTO(
+        status = true,
+        message = "",
+        data = null
+    )
 
     @Test
     fun `ReservationService_createReservation invoke ReservationRepository_save`() {
         every { userRepository.getReferenceById(any()) } returns sampleUser
         every { eventRepository.findByIdWithPesimisticLock(any()) } returns sampleEvent
         every { eventRepository.findByIdWithOptimisicLock(any()) } returns sampleEvent
-
+        every { restTemplate.exchange(any() as String, HttpMethod.DELETE, null, QueueResponseDTO::class.java) } returns ResponseEntity.ok(queueSuccess)
         every { eventRepository.saveAndFlush(any()) } returns sampleEvent
         every { reservationRepository.saveAndFlush(any()) } returns sampleReservation
         reservationService.createReservation(1, 1, "김해군", "010-1234-5678", 1, "서울")
@@ -84,7 +93,18 @@ class ReservationServiceTest() {
     fun `ReservationService_createReservation throw custom exception when time not allowed`() {
         every { userRepository.getReferenceById(any()) } returns sampleUser
         every { eventRepository.findByIdWithPesimisticLock(any()) } returns sampleWrongEvent
+        every { restTemplate.exchange(any() as String, HttpMethod.DELETE, null, QueueResponseDTO::class.java) } returns ResponseEntity.ok(queueSuccess)
+        every { eventRepository.saveAndFlush(any()) } returns sampleEvent
+        every { reservationRepository.saveAndFlush(any()) } returns sampleReservation
 
+        assertThrows<CustomException> { reservationService.createReservation(1, 1, "김해군", "010-1234-5678", 1, "서울") }
+    }
+
+    @Test
+    fun `ReservationService_createReservation throw custom exception when doesn't have waiting ticket`() {
+        every { userRepository.getReferenceById(any()) } returns sampleUser
+        every { eventRepository.findByIdWithPesimisticLock(any()) } returns sampleWrongEvent
+        every { restTemplate.exchange(any() as String, HttpMethod.DELETE, null, QueueResponseDTO::class.java) } throws HttpClientErrorException(HttpStatus.NOT_FOUND)
         every { eventRepository.saveAndFlush(any()) } returns sampleEvent
         every { reservationRepository.saveAndFlush(any()) } returns sampleReservation