chore: add the GET all budget by account Id

Author: felipemelozx

src/main/java/fun/trackmoney/budget/controller/BudgetsController.java

@@ -33,9 +33,9 @@ public ResponseEntity<ApiResponse<BudgetResponseDTO>> create(@RequestBody Budget
         new ApiResponse<>(true, "Budget created", budgetsService.create(dto), null));
   }
 
-  @GetMapping
-  public ResponseEntity<ApiResponse<List<BudgetResponseDTO>>> findAll() {
-    var list = budgetsService.findAll();
+  @GetMapping("/findAll/{accountId}")
+  public ResponseEntity<ApiResponse<List<BudgetResponseDTO>>> findAllByAccountId(@PathVariable Integer accountId) {
+    var list = budgetsService.findAllByAccountId(accountId);
     return ResponseEntity.status(HttpStatus.OK).body(
         new ApiResponse<>(true, "Get all Budget",list , null));
   }

src/main/java/fun/trackmoney/budget/dtos/BudgetResponseDTO.java

@@ -9,5 +9,6 @@ public record BudgetResponseDTO (Integer budgetId,
                                  CategoryEntity category,
                                  AccountResponseDTO account,
                                  BigDecimal targetAmount,
+                                 BigDecimal currentAmount,
                                  Integer resetDay) {
 }

src/main/java/fun/trackmoney/budget/repository/BudgetsRepository.java

@@ -3,5 +3,8 @@
 import fun.trackmoney.budget.entity.BudgetsEntity;
 import org.springframework.data.jpa.repository.JpaRepository;
 
+import java.util.List;
+
 public interface BudgetsRepository extends JpaRepository<BudgetsEntity, Integer> {
+  List<BudgetsEntity> findAllByAccount_AccountId(Integer accountId);
 }

src/main/java/fun/trackmoney/budget/service/BudgetsService.java

@@ -12,6 +12,7 @@
 import fun.trackmoney.user.service.UserService;
 import org.springframework.stereotype.Service;
 
+import java.math.BigDecimal;
 import java.util.List;
 
 @Service
@@ -47,8 +48,25 @@ public BudgetResponseDTO create(BudgetCreateDTO dto) {
     return budgetMapper.entityToResponseDTO(budgetsRepository.save(budgets));
   }
 
-  public List<BudgetResponseDTO> findAll() {
-    return budgetMapper.entityListToResponseList(budgetsRepository.findAll());
+  public List<BudgetResponseDTO> findAllByAccountId(Integer accountId) {
+    return budgetMapper.entityListToResponseList(
+            budgetsRepository.findAllByAccount_AccountId(accountId)
+        ).stream()
+        .map(budget -> {
+          BigDecimal currentAmount = budget.currentAmount() == null
+              ? BigDecimal.valueOf(100)
+              : budget.currentAmount();
+
+          return new BudgetResponseDTO(
+              budget.budgetId(),
+              budget.category(),
+              budget.account(),
+              budget.targetAmount(),
+              currentAmount,
+              budget.resetDay()
+          );
+        })
+        .toList();
   }
 
   public BudgetResponseDTO findById(Integer id) {

src/test/java/fun/trackmoney/budget/controller/BudgetsControllerTest.java

@@ -62,17 +62,17 @@ void create_shouldReturnCreatedAndBudgetResponse() {
   @Test
   void findAll_shouldReturnOkAndListOfBudgetResponses() {
     List<BudgetResponseDTO> budgetList = Collections.singletonList(responseDTO);
-    when(budgetsService.findAll()).thenReturn(budgetList);
+    when(budgetsService.findAllByAccountId(createDTO.accountId())).thenReturn(budgetList);
 
-    ResponseEntity<ApiResponse<List<BudgetResponseDTO>>> response = budgetsController.findAll();
+    ResponseEntity<ApiResponse<List<BudgetResponseDTO>>> response = budgetsController.findAllByAccountId(createDTO.accountId());
 
     assertEquals(HttpStatus.OK, response.getStatusCode());
     assertNotNull(response.getBody());
     assertTrue(response.getBody().isSuccess());
     assertEquals("Get all Budget", response.getBody().getMessage());
     assertEquals(budgetList, response.getBody().getData());
     assertNull(response.getBody().getErrors());
-    verify(budgetsService, times(1)).findAll();
+    verify(budgetsService, times(1)).findAllByAccountId(createDTO.accountId());
   }
 
   @Test

src/test/java/fun/trackmoney/budget/service/BudgetsServiceTest.java

@@ -114,11 +114,11 @@ void findAll_shouldReturnEmptyList_whenNoBudgetsExist() {
     when(budgetsRepository.findAll()).thenReturn(Collections.emptyList());
     when(budgetMapper.entityListToResponseList(Collections.emptyList())).thenReturn(Collections.emptyList());
 
-    List<BudgetResponseDTO> result = budgetsService.findAll();
+    List<BudgetResponseDTO> result = budgetsService.findAllByAccountId(accountEntity.getAccountId());
 
     assertNotNull(result);
     assertTrue(result.isEmpty());
-    verify(budgetsRepository, times(1)).findAll();
+    verify(budgetsRepository, times(1)).findAllByAccount_AccountId(accountEntity.getAccountId());
     verify(budgetMapper, times(1)).entityListToResponseList(Collections.emptyList());
   }
 
@@ -131,15 +131,15 @@ void findAll_shouldReturnListOfBudgetResponseDTO() {
     List<BudgetsEntity> entities = List.of(entity);
     List<BudgetResponseDTO> dtos = List.of(dto);
 
-    when(budgetsRepository.findAll()).thenReturn(entities);
+    when(budgetsRepository.findAllByAccount_AccountId(accountEntity.getAccountId())).thenReturn(entities);
     when(budgetMapper.entityListToResponseList(entities)).thenReturn(dtos);
 
-    List<BudgetResponseDTO> result = budgetsService.findAll();
+    List<BudgetResponseDTO> result = budgetsService.findAllByAccountId(accountEntity.getAccountId());
 
     assertNotNull(result);
     assertEquals(1, result.size());
     assertEquals(dtos, result);
-    verify(budgetsRepository, times(1)).findAll();
+    verify(budgetsRepository, times(1)).findAllByAccount_AccountId(accountEntity.getAccountId());
     verify(budgetMapper, times(1)).entityListToResponseList(entities);
   }