Add tests

test for operation WITHDRAW, unsupported operation and not found wallet uuid
bifrurcated · Jan 13, 2024 · 1609abf · 1609abf
1 parent 4566887
commit 1609abf
Showing 1 changed file with 56 additions and 0 deletions.
diff --git a/src/test/java/com/bifurcated/wallet/controller/WalletControllerTest.java b/src/test/java/com/bifurcated/wallet/controller/WalletControllerTest.java
@@ -88,6 +88,54 @@ record WalletRequest(UUID valletId, String operationType, Float amount){}
                 .andExpect(status().isNotFound());
     }
 
+    @Test
+    public void testWalletWithdraw() throws Exception {
+        record WalletRequest(UUID valletId, String operationType, Float amount){}
+        record WalletResponse(UUID id, Float amount) {}
+        UUID id = UUID.fromString("b3919077-79e6-4570-bfe0-980ef18f3731");
+        WalletRequest walletRequest = new WalletRequest(
+                id, "WITHDRAW", 1000F);
+        WalletResponse walletResponse = new WalletResponse(
+                id, 1000F
+        );
+        String requestBody = objectMapper.writeValueAsString(walletRequest);
+        String response = objectMapper.writeValueAsString(walletResponse);
+        this.mockMvc.perform(post(END_POINT_PATH+"/wallet")
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content(requestBody))
+                .andDo(print())
+                .andExpect(status().isOk())
+                .andExpect(content().json(response));
+    }
+
+    @Test
+    public void testWalletWithdrawNotEnoughMoney() throws Exception {
+        record WalletRequest(UUID valletId, String operationType, Float amount){}
+        UUID id = UUID.fromString("bc10fad7-94f1-4047-be4e-311247eed5fb");
+        WalletRequest walletRequest = new WalletRequest(
+                id, "WITHDRAW", 1000F);
+
+        String requestBody = objectMapper.writeValueAsString(walletRequest);
+        this.mockMvc.perform(post(END_POINT_PATH+"/wallet")
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content(requestBody))
+                .andDo(print())
+                .andExpect(status().isPaymentRequired());
+    }
+
+    @Test
+    public void testWalletWithdrawUnsupportedOperationType() throws Exception {
+        record WalletRequest(UUID valletId, String operationType, Float amount){}
+        UUID id = UUID.fromString("bc10fad7-94f1-4047-be4e-311247eed5fb");
+        WalletRequest walletRequest = new WalletRequest(id, "WWWWWWW", 1000F);
+        String requestBody = objectMapper.writeValueAsString(walletRequest);
+        this.mockMvc.perform(post(END_POINT_PATH+"/wallet")
+                        .contentType(MediaType.APPLICATION_JSON)
+                        .content(requestBody))
+                .andDo(print())
+                .andExpect(status().isUnsupportedMediaType());
+    }
+
     @Test
     public void testWalletGetAmount() throws Exception {
         record BalanceResponse(Float amount){}
@@ -99,4 +147,12 @@ record BalanceResponse(Float amount){}
                 .andExpect(status().isOk())
                 .andExpect(content().json(response));
     }
+
+    @Test
+    public void testWalletGetAmountNotFound() throws Exception {
+        String id = "b3919077-79e6-4570-bfe0-980ef18f3700";
+        this.mockMvc.perform(get(END_POINT_PATH+"/wallets/"+id))
+                .andDo(print())
+                .andExpect(status().isNotFound());
+    }
 }