update TransactionController with withdrawal

src/test/kotlin/ltd/lulz/controller/TransactionControllerTest.kt

@@ -6,11 +6,13 @@ import java.math.BigDecimal
 import java.util.UUID
 import ltd.lulz.controller.AccountControllerTest.Companion.name
 import ltd.lulz.exception.AccountNotFoundException
+import ltd.lulz.exception.InsufficientFundsException
 import ltd.lulz.model.AccountEntity
 import ltd.lulz.model.Transaction
 import ltd.lulz.service.TransactionService
 import org.junit.jupiter.api.BeforeEach
 import org.junit.jupiter.api.Test
+import org.springframework.http.HttpStatus.NOT_ACCEPTABLE
 import org.springframework.http.MediaType.APPLICATION_JSON
 import org.springframework.test.web.reactive.server.WebTestClient
 import reactor.core.publisher.Mono
@@ -102,4 +104,94 @@ class TransactionControllerTest {
         // then
         result.expectStatus().isNotFound
     }
+
+    @Test
+    fun `withdrawal success`() {
+        // given
+        val request = Transaction.Request(uuid, amount)
+
+        every { transactionService.withdrawal(any(), any()) } returns Mono.just(
+            AccountEntity(id = uuid, name = name, amount = amount),
+        )
+
+        // when
+        val result = webTestClient.post()
+            .uri("/withdrawal")
+            .contentType(APPLICATION_JSON)
+            .bodyValue(request)
+            .exchange()
+
+        // then
+        result.expectStatus().isCreated
+    }
+
+    @Test
+    fun `withdrawal fail amount to small`() {
+        // given
+        val request = Transaction.Request(uuid, BigDecimal.valueOf(0.009))
+
+        // when
+        val result = webTestClient.post()
+            .uri("/withdrawal")
+            .contentType(APPLICATION_JSON)
+            .bodyValue(request)
+            .exchange()
+
+        // then
+        result.expectStatus().isBadRequest
+    }
+
+    @Test
+    fun `withdrawal insufficient Funds`() {
+        // given
+        val request = Transaction.Request(uuid, amount)
+
+        every { transactionService.withdrawal(any(), any()) } returns Mono.error(InsufficientFundsException())
+
+        // when
+        val result = webTestClient.post()
+            .uri("/withdrawal")
+            .contentType(APPLICATION_JSON)
+            .bodyValue(request)
+            .exchange()
+
+        // then
+        result.expectStatus().isEqualTo(NOT_ACCEPTABLE)
+    }
+
+    @Test
+    fun `withdrawal account not found`() {
+        // given
+        val request = Transaction.Request(uuid, amount)
+
+        every { transactionService.withdrawal(any(), any()) } returns Mono.error(AccountNotFoundException())
+
+        // when
+        val result = webTestClient.post()
+            .uri("/withdrawal")
+            .contentType(APPLICATION_JSON)
+            .bodyValue(request)
+            .exchange()
+
+        // then
+        result.expectStatus().isNotFound
+    }
+
+    @Test
+    fun `withdrawal default error`() {
+        // given
+        val request = Transaction.Request(uuid, amount)
+
+        every { transactionService.withdrawal(any(), any()) } returns Mono.error(RuntimeException())
+
+        // when
+        val result = webTestClient.post()
+            .uri("/withdrawal")
+            .contentType(APPLICATION_JSON)
+            .bodyValue(request)
+            .exchange()
+
+        // then
+        result.expectStatus().is5xxServerError
+    }
 }