From ba92a3d331ac5e57874ada126ecfb54795e20034 Mon Sep 17 00:00:00 2001 From: VoHoangAn <63175067+VoHoangAn@users.noreply.github.com> Date: Thu, 26 Sep 2024 15:03:16 +0700 Subject: [PATCH] Unit test for transaction (#127) * add test * add test * fix duplicate * fix in out to vietnamese * fix choose a user to vietnamese * fix validation and to vietnamese * fix validation to vietnamese * fix default is admin * fix edit modal * add statistic * fix return active category * fix checkstyle * add test * add test * fix --- .../sunrise/constraints/AfterAndUntilNow.java | 2 +- .../StringMustBeDigitWithFraction.java | 21 ++ ...tringMustBeDigitWithFractionValidator.java | 36 ++ .../controllers/TransactionController.java | 24 +- .../CreateOrUpdateTransactionRequest.java | 2 + .../dtos/responses/StatisticResponse.java | 13 + .../repositories/TransactionRepository.java | 15 + .../sunrise/services/TransactionService.java | 3 + .../services/impl/CategoryServiceImpl.java | 17 +- .../services/impl/TransactionServiceImpl.java | 56 +-- .../java/com/fjb/sunrise/utils/Constants.java | 1 + .../templates/transaction/index.html | 339 +++++++++--------- .../services/TransactionServiceTest.java | 160 ++++++++- 13 files changed, 468 insertions(+), 221 deletions(-) create mode 100644 src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFraction.java create mode 100644 src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFractionValidator.java create mode 100644 src/main/java/com/fjb/sunrise/dtos/responses/StatisticResponse.java diff --git a/src/main/java/com/fjb/sunrise/constraints/AfterAndUntilNow.java b/src/main/java/com/fjb/sunrise/constraints/AfterAndUntilNow.java index d0b7bc52..19ae75be 100644 --- a/src/main/java/com/fjb/sunrise/constraints/AfterAndUntilNow.java +++ b/src/main/java/com/fjb/sunrise/constraints/AfterAndUntilNow.java @@ -11,7 +11,7 @@ @Target({ElementType.FIELD, ElementType.METHOD}) @Retention(RetentionPolicy.RUNTIME) public @interface AfterAndUntilNow { - String message() default "must be after {value} and before now"; + String message() default "Thời gian tạo phải sau 01/01/2020 và trước hiện tại"; Class[] groups() default {}; Class[] payload() default {}; String value(); diff --git a/src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFraction.java b/src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFraction.java new file mode 100644 index 00000000..11da82c2 --- /dev/null +++ b/src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFraction.java @@ -0,0 +1,21 @@ +package com.fjb.sunrise.constraints; + +import jakarta.validation.Constraint; +import jakarta.validation.Payload; +import java.lang.annotation.ElementType; +import java.lang.annotation.Retention; +import java.lang.annotation.RetentionPolicy; +import java.lang.annotation.Target; + +@Constraint(validatedBy = StringMustBeDigitWithFractionValidator.class) +@Target({ElementType.FIELD, ElementType.METHOD}) +@Retention(RetentionPolicy.RUNTIME) +public @interface StringMustBeDigitWithFraction { + String message() default "Số tiền phải là chữ số với số thập phân là {fraction}"; + + Class[] groups() default {}; + + Class[] payload() default {}; + + int fraction(); +} diff --git a/src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFractionValidator.java b/src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFractionValidator.java new file mode 100644 index 00000000..80ab204d --- /dev/null +++ b/src/main/java/com/fjb/sunrise/constraints/StringMustBeDigitWithFractionValidator.java @@ -0,0 +1,36 @@ +package com.fjb.sunrise.constraints; + + +import jakarta.validation.ConstraintValidator; +import jakarta.validation.ConstraintValidatorContext; +import org.springframework.util.StringUtils; + +public class StringMustBeDigitWithFractionValidator + implements ConstraintValidator { + private int fraction = 0; + + @Override + public void initialize(StringMustBeDigitWithFraction constraintAnnotation) { + ConstraintValidator.super.initialize(constraintAnnotation); + fraction = constraintAnnotation.fraction(); + } + + @Override + public boolean isValid(String string, ConstraintValidatorContext constraintValidatorContext) { + String ignoreComma = string.replaceAll(",", ""); + String ignoreDot = ignoreComma.replace(".", ""); + if (ignoreDot.chars().allMatch(Character::isDigit)) { + if (fraction > 0) { + int startOfFraction = ignoreComma.indexOf("."); + if (startOfFraction == -1) { + return true; + } + if (ignoreComma.length() - startOfFraction - 1 > fraction) { + return false; + } + } + return true; + } + return false; + } +} diff --git a/src/main/java/com/fjb/sunrise/controllers/TransactionController.java b/src/main/java/com/fjb/sunrise/controllers/TransactionController.java index 641b580d..6aa6645d 100644 --- a/src/main/java/com/fjb/sunrise/controllers/TransactionController.java +++ b/src/main/java/com/fjb/sunrise/controllers/TransactionController.java @@ -1,9 +1,5 @@ package com.fjb.sunrise.controllers; -import static com.fjb.sunrise.utils.Constants.ApiConstant.CATEGORIES; -import static com.fjb.sunrise.utils.Constants.ApiConstant.TRANSACTION_INDEX; -import static com.fjb.sunrise.utils.Constants.ApiConstant.USERS; - import com.fjb.sunrise.dtos.base.DataTableInputDTO; import com.fjb.sunrise.dtos.requests.CreateOrUpdateTransactionRequest; import com.fjb.sunrise.dtos.responses.TransactionFullPageResponse; @@ -13,6 +9,7 @@ import com.fjb.sunrise.services.CategoryService; import com.fjb.sunrise.services.TransactionService; import com.fjb.sunrise.services.UserService; +import com.fjb.sunrise.utils.Constants; import jakarta.validation.Valid; import java.text.ParseException; import lombok.RequiredArgsConstructor; @@ -42,21 +39,24 @@ public class TransactionController { @GetMapping("/create") public String getCreate(@ModelAttribute("request") CreateOrUpdateTransactionRequest request, Model model) { - model.addAttribute(CATEGORIES, categoryService.findCategoryByAdminAndUser()); - model.addAttribute(USERS, userService.findAllNormalUser()); - return TRANSACTION_INDEX; + model.addAttribute(Constants.ApiConstant.CATEGORIES, categoryService.findCategoryByAdminAndUser()); + model.addAttribute(Constants.ApiConstant.USERS, userService.findAllNormalUser()); + model.addAttribute(Constants.ApiConstant.STATISTIC, transactionService.statistic()); + return Constants.ApiConstant.TRANSACTION_INDEX; } @PostMapping("/create") public String postCreate(@ModelAttribute("request") @Valid CreateOrUpdateTransactionRequest request, BindingResult result, Model model) throws ParseException { - model.addAttribute(CATEGORIES, categoryService.findCategoryByAdminAndUser()); - model.addAttribute(USERS, userService.findAllNormalUser()); + model.addAttribute(Constants.ApiConstant.CATEGORIES, categoryService.findCategoryByAdminAndUser()); + model.addAttribute(Constants.ApiConstant.USERS, userService.findAllNormalUser()); + model.addAttribute(Constants.ApiConstant.STATISTIC, transactionService.statistic()); if (result.hasErrors()) { - model.addAttribute(CATEGORIES, categoryService.findCategoryByAdminAndUser()); - model.addAttribute(USERS, userService.findAllNormalUser()); - return TRANSACTION_INDEX; + model.addAttribute(Constants.ApiConstant.CATEGORIES, categoryService.findCategoryByAdminAndUser()); + model.addAttribute(Constants.ApiConstant.USERS, userService.findAllNormalUser()); + model.addAttribute(Constants.ApiConstant.STATISTIC, transactionService.statistic()); + return Constants.ApiConstant.TRANSACTION_INDEX; } Transaction transaction = transactionService.create(request); return "redirect:/transaction/create"; diff --git a/src/main/java/com/fjb/sunrise/dtos/requests/CreateOrUpdateTransactionRequest.java b/src/main/java/com/fjb/sunrise/dtos/requests/CreateOrUpdateTransactionRequest.java index b4b625cc..32fdedfa 100644 --- a/src/main/java/com/fjb/sunrise/dtos/requests/CreateOrUpdateTransactionRequest.java +++ b/src/main/java/com/fjb/sunrise/dtos/requests/CreateOrUpdateTransactionRequest.java @@ -6,6 +6,7 @@ import com.fasterxml.jackson.datatype.jsr310.deser.LocalDateTimeDeserializer; import com.fasterxml.jackson.datatype.jsr310.ser.LocalDateTimeSerializer; import com.fjb.sunrise.constraints.AfterAndUntilNow; +import com.fjb.sunrise.constraints.StringMustBeDigitWithFraction; import com.fjb.sunrise.enums.ETrans; import jakarta.validation.constraints.NotBlank; import java.io.Serializable; @@ -16,6 +17,7 @@ public class CreateOrUpdateTransactionRequest implements Serializable { private Long id; @NotBlank(message = "Amount must not be blank") + @StringMustBeDigitWithFraction(fraction = 2) private String amount; private ETrans transactionType; diff --git a/src/main/java/com/fjb/sunrise/dtos/responses/StatisticResponse.java b/src/main/java/com/fjb/sunrise/dtos/responses/StatisticResponse.java new file mode 100644 index 00000000..04907d59 --- /dev/null +++ b/src/main/java/com/fjb/sunrise/dtos/responses/StatisticResponse.java @@ -0,0 +1,13 @@ +package com.fjb.sunrise.dtos.responses; + +import java.io.Serializable; +import lombok.Data; + + + +@Data +public class StatisticResponse implements Serializable { + private String totalThisYear; + private String totalInputThisYear; + private String totalThisMonth; +} diff --git a/src/main/java/com/fjb/sunrise/repositories/TransactionRepository.java b/src/main/java/com/fjb/sunrise/repositories/TransactionRepository.java index 2fcc85b5..c158d12d 100644 --- a/src/main/java/com/fjb/sunrise/repositories/TransactionRepository.java +++ b/src/main/java/com/fjb/sunrise/repositories/TransactionRepository.java @@ -1,13 +1,28 @@ package com.fjb.sunrise.repositories; + +import com.fjb.sunrise.enums.ETrans; import com.fjb.sunrise.models.Transaction; +import java.time.LocalDateTime; import org.springframework.data.domain.Page; import org.springframework.data.domain.Pageable; import org.springframework.data.jpa.repository.JpaRepository; import org.springframework.data.jpa.repository.JpaSpecificationExecutor; +import org.springframework.data.jpa.repository.Query; + + public interface TransactionRepository extends JpaRepository, JpaSpecificationExecutor { Page findAll(Pageable pageable); + + @Query("select sum(t.amount) from Transaction t " + + "where t.updatedAt between ?1 and ?2") + Double sumAmountInRange(LocalDateTime start, LocalDateTime end); + + @Query("select sum(t.amount) from Transaction t " + + "where t.transactionType = ?1 and t.updatedAt between ?2 and ?3 ") + Double sumTransactionTypeINInThisYear(ETrans transactionType, LocalDateTime start, LocalDateTime end); + } diff --git a/src/main/java/com/fjb/sunrise/services/TransactionService.java b/src/main/java/com/fjb/sunrise/services/TransactionService.java index 7765677f..dfd07a51 100644 --- a/src/main/java/com/fjb/sunrise/services/TransactionService.java +++ b/src/main/java/com/fjb/sunrise/services/TransactionService.java @@ -2,6 +2,7 @@ import com.fjb.sunrise.dtos.base.DataTableInputDTO; import com.fjb.sunrise.dtos.requests.CreateOrUpdateTransactionRequest; +import com.fjb.sunrise.dtos.responses.StatisticResponse; import com.fjb.sunrise.models.Transaction; import java.text.ParseException; import org.springframework.data.domain.Page; @@ -12,4 +13,6 @@ public interface TransactionService { Page getTransactionList(DataTableInputDTO payload, String email); Transaction update(CreateOrUpdateTransactionRequest request) throws ParseException; + + StatisticResponse statistic(); } diff --git a/src/main/java/com/fjb/sunrise/services/impl/CategoryServiceImpl.java b/src/main/java/com/fjb/sunrise/services/impl/CategoryServiceImpl.java index 7f1dc135..2df3197f 100644 --- a/src/main/java/com/fjb/sunrise/services/impl/CategoryServiceImpl.java +++ b/src/main/java/com/fjb/sunrise/services/impl/CategoryServiceImpl.java @@ -148,7 +148,7 @@ public List findCategoryByAdminAndUser() { List orders = new ArrayList<>(); orders.add(Sort.Order.asc("owner.role")); Sort sortOpt = Sort.by(orders); - Specification specs = findAllByUser(); + Specification specs = findAllForTransaction(); return categoryRepository.findAll(specs, sortOpt); } @@ -169,6 +169,21 @@ private Specification findAllByUser() { }); } + private Specification findAllForTransaction() { + return Specification.where((root, query, builder) -> { + Join userJoin = root.join("owner"); + User dbUser = userRepository.findById(getCurrentUserId()).orElseThrow(); + if (dbUser.getRole() == ERole.ADMIN) { + return builder.equal(root.get("status"), EStatus.ACTIVE); + } + Predicate hasRoleAdmin = builder.equal(userJoin.get("role"), "ADMIN"); + Predicate isOwner = builder.equal(userJoin.get("id"), getCurrentUserId()); + Predicate active = builder.equal(root.get("status"), EStatus.ACTIVE); + Predicate or = builder.or(hasRoleAdmin, isOwner); + return builder.and(or, active); + }); + } + public Long getCurrentUserId() { Authentication auth = SecurityContextHolder.getContext().getAuthentication(); org.springframework.security.core.userdetails.User securityUser = diff --git a/src/main/java/com/fjb/sunrise/services/impl/TransactionServiceImpl.java b/src/main/java/com/fjb/sunrise/services/impl/TransactionServiceImpl.java index 3f541439..122b556b 100644 --- a/src/main/java/com/fjb/sunrise/services/impl/TransactionServiceImpl.java +++ b/src/main/java/com/fjb/sunrise/services/impl/TransactionServiceImpl.java @@ -2,6 +2,7 @@ import com.fjb.sunrise.dtos.base.DataTableInputDTO; import com.fjb.sunrise.dtos.requests.CreateOrUpdateTransactionRequest; +import com.fjb.sunrise.dtos.responses.StatisticResponse; import com.fjb.sunrise.enums.ETrans; import com.fjb.sunrise.models.Category; import com.fjb.sunrise.models.Transaction; @@ -11,8 +12,6 @@ import com.fjb.sunrise.services.TransactionService; import jakarta.persistence.criteria.Join; import jakarta.persistence.criteria.Predicate; -import jakarta.persistence.criteria.Root; -import jakarta.persistence.criteria.Subquery; import jakarta.transaction.Transactional; import java.text.DecimalFormat; import java.text.DecimalFormatSymbols; @@ -21,16 +20,7 @@ import java.time.LocalDateTime; import java.time.Month; import java.time.Year; -import java.time.ZoneId; import java.time.ZoneOffset; -import java.time.format.DateTimeFormatter; -import java.util.ArrayList; -import java.util.Calendar; -import java.util.Collection; -import java.util.Date; -import java.util.List; -import java.util.Locale; -import java.util.TimeZone; import lombok.RequiredArgsConstructor; import lombok.extern.slf4j.Slf4j; import org.apache.logging.log4j.util.Strings; @@ -65,12 +55,6 @@ public Transaction create(CreateOrUpdateTransactionRequest request) throws Parse return transactionRepository.save(transaction); } - public Transaction findById(Long id) { - return transactionRepository.findById(id).orElseThrow( - () -> new IllegalArgumentException("Invalid transaction id") - ); - } - @Override public Page getTransactionList(DataTableInputDTO payload, String email) { Sort sortOpt = Sort.by(Sort.Direction.ASC, "id"); @@ -100,9 +84,15 @@ public Page getTransactionList(DataTableInputDTO payload, String em // "%" + keyword + "%" specs = specs.and(((root, query, builder) -> { Join categoryJoin = root.join("category"); + String search = payload.getSearch().getOrDefault("value", "").toLowerCase(); + if (search.equals("thu")) { + search = "IN"; + } else if (search.equals("chi")) { + search = "OUT"; + } Predicate predictTransactionType = builder.like(builder.lower(root.get("transactionType")), String.format("%%%s%%", - payload.getSearch().getOrDefault("value", "").toLowerCase())); + search.toLowerCase())); Predicate predictCategory = builder.like(builder.lower(categoryJoin.get("name")), String.format("%%%s%%", payload.getSearch().getOrDefault("value", "").toLowerCase() @@ -121,7 +111,6 @@ public Transaction update(CreateOrUpdateTransactionRequest request) throws Parse transaction.setUpdatedAt(request.getCreatedAt()); transaction.setTransactionType(request.getTransactionType()); transaction.setAmount(convertMoneyStringWithCommaToDouble(request.getAmount())); - transaction.setUser(userRepository.findByEmailOrPhone(getCurrentUserName())); transaction.setCategory(categoryRepository.findById(request.getCategory()).orElse(null)); log.error("update: {}", transaction.toString()); @@ -130,14 +119,27 @@ public Transaction update(CreateOrUpdateTransactionRequest request) throws Parse return transaction1; } - private String changeFormatFromFullDateToMonthDate(LocalDateTime dateTime) throws ParseException { - SimpleDateFormat monthDate = new SimpleDateFormat("dd-MM", Locale.ENGLISH); - SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd"); - - - Date date = sdf.parse(dateTime.format(DateTimeFormatter.ofPattern("yyyy-MM-dd"))); - monthDate.format(date); - return monthDate.format(date); + @Override + public StatisticResponse statistic() { + StatisticResponse response = new StatisticResponse(); + final LocalDateTime firstDay = getFirstOrLastDateOfThisYear(false); + final LocalDateTime lastDay = getFirstOrLastDateOfThisYear(true); + final LocalDateTime firstDayOfThisMonth = getFirstDayOfThisMonth(); + + + response.setTotalThisMonth( + convertDoubleWithScientificNotationToDouble( + transactionRepository.sumAmountInRange(firstDayOfThisMonth, lastDay) + )); + + response.setTotalThisYear( + convertDoubleWithScientificNotationToDouble( + transactionRepository.sumAmountInRange(firstDay, lastDay) + )); + response.setTotalInputThisYear(convertDoubleWithScientificNotationToDouble( + transactionRepository.sumTransactionTypeINInThisYear(ETrans.IN, firstDay, lastDay) + )); + return response; } private String convertDoubleWithScientificNotationToDouble(Double amount) { diff --git a/src/main/java/com/fjb/sunrise/utils/Constants.java b/src/main/java/com/fjb/sunrise/utils/Constants.java index 7b1a18b8..e7ff7c80 100644 --- a/src/main/java/com/fjb/sunrise/utils/Constants.java +++ b/src/main/java/com/fjb/sunrise/utils/Constants.java @@ -36,6 +36,7 @@ private ApiConstant() {} public static final String TRANSACTION_INDEX = "transaction/index"; public static final String CATEGORIES = "categories"; public static final String USERS = "users"; + public static final String STATISTIC = "statistic"; public static final String AUTH_REDIRECT_LOGIN = "redirect:/auth/login"; public static final String AUTH_VIEW = "auth/loginAndRegister"; diff --git a/src/main/resources/templates/transaction/index.html b/src/main/resources/templates/transaction/index.html index a5667990..0633a0f0 100644 --- a/src/main/resources/templates/transaction/index.html +++ b/src/main/resources/templates/transaction/index.html @@ -51,10 +51,11 @@

Tổng quan

- +
@@ -64,172 +65,165 @@

Tổng quan

-
- - -
-
-
-
-
-
- Thu nhập (Tháng Này) +
+
+ +
+
+
+
+
+
+ Thu nhập (Tháng Này) +
+
+
+
+
-
20,000,000 VNĐ
-
-
-
-
- -
-
-
-
-
-
- Thu nhập (Năm nay) + +
+
+
+
+
+
+ Thu nhập (Năm nay) +
+
+
+
+
-
240,000,000 VNĐ
-
-
-
-
- -
-
-
-
-
-
Tổng chi - (Tháng này) -
-
-
-
12,000,000 - VNĐ -
+ +
+
+
+
+
+
+ Tổng chi (Tháng này)
- +
+
+
+
-
-
-
-
- - -
-
-
-
-
-
- Tổng nợ (Tháng này) + +
+
+
+
+
+
+ Tổng nợ (Tháng này) +
+
0 VNĐ
+
+
+
-
0 VNĐ
-
-
-
-
+
-
-
- -
-
Biểu đồ thu nhập (Tháng này)
-
- -
-
Dropdown Header:
- Action - Another action -
- Something else here -
-
-
- -
-
- -
-
-
-
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + -
-
- -
-
Thu nhập theo danh mục (Tháng này) -
-
- -
-
Dropdown Header:
- Action - Another action -
- Something else here -
-
-
- -
-
- -
-
- - Lương - - - Chứng khoán - - - Bitcoin - - - Khác - -
-
-
-
+ + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + + +
@@ -318,23 +312,24 @@
-
+
+
- -
+ value="IN" th:checked="true"> @@ -362,7 +357,8 @@ - +
@@ -396,7 +392,7 @@ - diff --git a/src/test/java/com/fjb/sunrise/services/TransactionServiceTest.java b/src/test/java/com/fjb/sunrise/services/TransactionServiceTest.java index 20d71267..40a4a809 100644 --- a/src/test/java/com/fjb/sunrise/services/TransactionServiceTest.java +++ b/src/test/java/com/fjb/sunrise/services/TransactionServiceTest.java @@ -1,10 +1,14 @@ package com.fjb.sunrise.services; +import static org.instancio.Select.field; +import static org.mockito.ArgumentMatchers.*; + import com.fjb.sunrise.dtos.base.DataTableInputDTO; import com.fjb.sunrise.dtos.requests.CategoryCreateDto; import com.fjb.sunrise.dtos.requests.CategoryUpdateDto; import com.fjb.sunrise.dtos.requests.CreateOrUpdateTransactionRequest; import com.fjb.sunrise.dtos.responses.CategoryResponseDto; +import com.fjb.sunrise.dtos.responses.StatisticResponse; import com.fjb.sunrise.enums.ERole; import com.fjb.sunrise.enums.EStatus; import com.fjb.sunrise.enums.ETrans; @@ -15,35 +19,70 @@ import com.fjb.sunrise.repositories.TransactionRepository; import com.fjb.sunrise.repositories.UserRepository; import com.fjb.sunrise.services.impl.TransactionServiceImpl; + +import java.text.ParseException; +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.Optional; +import java.util.Set; + +import org.instancio.Instancio; +import org.junit.jupiter.api.Assertions; import org.junit.jupiter.api.BeforeEach; +import org.junit.jupiter.api.DisplayName; import org.junit.jupiter.api.Nested; import org.junit.jupiter.api.Test; +import org.junit.jupiter.api.extension.ExtendWith; import org.mockito.InjectMocks; import org.mockito.Mock; +import org.mockito.Mockito; import org.mockito.MockitoAnnotations; import java.time.LocalDateTime; +import org.mockito.junit.jupiter.MockitoExtension; +import org.springframework.beans.factory.annotation.Autowired; +import org.springframework.boot.test.context.SpringBootTest; +import org.springframework.boot.test.mock.mockito.MockBean; +import org.springframework.data.domain.Page; +import org.springframework.data.domain.PageImpl; +import org.springframework.data.domain.Pageable; +import org.springframework.data.jpa.domain.Specification; +import org.springframework.security.authentication.UsernamePasswordAuthenticationToken; +import org.springframework.security.core.Authentication; +import org.springframework.security.core.authority.SimpleGrantedAuthority; +import org.springframework.security.core.context.SecurityContext; +import org.springframework.security.core.context.SecurityContextHolder; +import org.springframework.security.core.userdetails.UserDetails; +import org.springframework.test.context.junit.jupiter.SpringExtension; + +@ExtendWith(SpringExtension.class) +@SpringBootTest public class TransactionServiceTest { - @InjectMocks - private TransactionServiceImpl transactionService; + @Autowired + TransactionServiceImpl transactionService; - @Mock - private TransactionRepository transactionRepository; - @Mock - private CategoryRepository categoryRepository; - @Mock - private UserRepository userRepository; + @MockBean + TransactionRepository transactionRepository; + @MockBean + CategoryRepository categoryRepository; + @MockBean + UserRepository userRepository; + @MockBean + DataTableInputDTO dataTableInputDTO; private Transaction transaction; private Category category; private User user; private CreateOrUpdateTransactionRequest create; private CreateOrUpdateTransactionRequest update; + private StatisticResponse statisticResponse; @BeforeEach public void init() { - MockitoAnnotations.openMocks(this); + MockitoAnnotations.initMocks(this); category = Category.builder() .id(1L) .name("Category-Test") @@ -76,14 +115,103 @@ public void init() { update.setAmount("200000"); update.setCategory(1L); update.setTransactionType(ETrans.IN); + + statisticResponse = new StatisticResponse(); + statisticResponse.setTotalThisMonth("100000"); + statisticResponse.setTotalInputThisYear("200000"); + statisticResponse.setTotalThisYear("100000"); + } + + @DisplayName("Junit test for createTransaction method") + @Test + public void createTransaction_whenCreateOrUpdateDto_returnTransaction() throws ParseException { + UserDetails appUserDetails = new org.springframework.security.core.userdetails.User("an@gmail.com", "123", + List.of(new SimpleGrantedAuthority("ROLE_USER"))); + + Authentication auth = new UsernamePasswordAuthenticationToken(appUserDetails, null); + + SecurityContext securityContext = Mockito.mock(SecurityContext.class); + Mockito.when(securityContext.getAuthentication()).thenReturn(auth); + SecurityContextHolder.setContext(securityContext); + + Mockito.when(categoryRepository.findById(anyLong())).thenReturn(Optional.of(category)); + Mockito.when(userRepository.findByEmailOrPhone(anyString())).thenReturn(user); + Mockito.when(transactionRepository.save(any())).thenReturn(transaction); + + Transaction transactionTest = transactionService.create(create); + Assertions.assertEquals(transactionTest, transaction); } -// @Nested -// class CreateCategoryTest { -// @Test -// void create_shouldReturnTransaction() { -// -// } -// } + @DisplayName("Junit test for getTransactionList method with keyword") + @Test + public void getTransactionList_whenDataTableInputDTOWithKeyword_returnPageOfTransaction() { + Mockito.when(dataTableInputDTO.getSearch()).thenReturn(Map.of("value", "thu")); + Mockito.when(dataTableInputDTO.getStart()).thenReturn(10); + Mockito.when(dataTableInputDTO.getLength()).thenReturn(10); + List transactions = Instancio.ofList(Transaction.class).create(); + Page page = new PageImpl(transactions); + Mockito.when(userRepository.findByEmailOrPhone(anyString())).thenReturn(user); + Mockito.when(transactionRepository.findAll(any(Specification.class), any(Pageable.class))).thenReturn(page); + + Page transactionPage = transactionService.getTransactionList(dataTableInputDTO, "an@gmail.com"); + + Assertions.assertEquals(transactionPage, page); + } + + @DisplayName("Junit test for getTransactionList method without keyword") + @Test + public void getTransactionList_whenDataTableInputDTO_returnPageOfTransaction() { + List> orders = new ArrayList<>(); + Map order = new HashMap<>(); + order.put("column", "2"); + order.put("dir", "asc"); + order.put("colName", "category"); + orders.add(order); + Map search = new HashMap<>(); + search.put("value", ""); + search.put("regex", "false"); + DataTableInputDTO dataTable = new DataTableInputDTO(1, null, orders, 10, 10, search); + List transactions = Instancio.ofList(Transaction.class).create(); + Page page = new PageImpl(transactions); + Mockito.when(userRepository.findByEmailOrPhone(anyString())).thenReturn(user); + + Mockito.when(transactionRepository.findAll(any(Specification.class), any(Pageable.class))).thenReturn(page); + + Page transactionPage = transactionService.getTransactionList(dataTable, "an@gmail.com"); + + Assertions.assertEquals(transactionPage, page); + } + + @DisplayName("Junit test for update method") + @Test + public void updateTransaction_whenCreateOrUpdateDto_returnTransaction() throws ParseException { + UserDetails appUserDetails = new org.springframework.security.core.userdetails.User("an@gmail.com", "123", + List.of(new SimpleGrantedAuthority("ROLE_USER"))); + + Authentication auth = new UsernamePasswordAuthenticationToken(appUserDetails, null); + + SecurityContext securityContext = Mockito.mock(SecurityContext.class); + Mockito.when(securityContext.getAuthentication()).thenReturn(auth); + SecurityContextHolder.setContext(securityContext); + + Mockito.when(transactionRepository.getReferenceById(anyLong())).thenReturn(transaction); + Mockito.when(categoryRepository.findById(anyLong())).thenReturn(Optional.of(category)); + Mockito.when(userRepository.findByEmailOrPhone(anyString())).thenReturn(user); + Mockito.when(transactionRepository.save(any())).thenReturn(transaction); + + Transaction transactionTest = transactionService.update(update); + Assertions.assertEquals(transactionTest, transaction); + } + + @DisplayName("Junit test for statistic method") + @Test + public void statistic_returnStatisticResponse() throws ParseException { + Mockito.when(transactionRepository.sumAmountInRange(any(LocalDateTime.class), any(LocalDateTime.class))) + .thenReturn(100000.0); + Mockito.when(transactionRepository.sumTransactionTypeINInThisYear(any(ETrans.class), any(LocalDateTime.class), any(LocalDateTime.class))) + .thenReturn(200000.0); + StatisticResponse response = transactionService.statistic(); + Assertions.assertEquals(statisticResponse, response); + } }