chore : test code modify

backend-tech-forge · Apr 13, 2024 · e2cd9a9 · e2cd9a9
1 parent f826eed
commit e2cd9a9
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/...roller/src/test/java/org/benchmarker/bmcontroller/mail/controller/MailControllerTest.java b/...roller/src/test/java/org/benchmarker/bmcontroller/mail/controller/MailControllerTest.java
@@ -59,7 +59,7 @@ public void mailSender() throws Exception {
 
 
         // when
-        when(mailSender.sendMail(any())).thenReturn(res);
+        when(mailSender.sendMail(any(), any())).thenReturn(res);
 
         // then
         mockMvc.perform(post("/api/mail/certification")
@@ -94,7 +94,7 @@ public void mailFormatErrorTest() throws Exception {
 
 
         // when
-        when(mailSender.sendMail(any())).thenReturn(res);
+        when(mailSender.sendMail(any(), any())).thenReturn(res);
 
         // then
         mockMvc.perform(post("/api/mail/certification")

diff --git a/...ontroller/src/test/java/org/benchmarker/bmcontroller/mail/service/MailSenderImplTest.java b/...ontroller/src/test/java/org/benchmarker/bmcontroller/mail/service/MailSenderImplTest.java
@@ -3,6 +3,8 @@
 import jakarta.mail.MessagingException;
 import jakarta.mail.internet.MimeMessage;
 import jakarta.transaction.Transactional;
+import org.benchmarker.bmcontroller.mail.common.factory.EmailBodyGenerator;
+import org.benchmarker.bmcontroller.mail.common.factory.EmailVerificationFactory;
 import org.benchmarker.bmcontroller.mail.controller.dto.EmailCertificationDto;
 import org.benchmarker.bmcontroller.mail.controller.dto.EmailResDto;
 import org.benchmarker.bmcontroller.mail.service.impl.MailSenderImpl;
@@ -43,10 +45,12 @@ public void sendMailTest() {
                 .certificationCode(authNum)
                 .build();
 
-        when(mailSender.sendMail(any())).thenReturn(res);
+        EmailBodyGenerator emailBodyGenerator = new EmailVerificationFactory();
+
+        when(mailSender.sendMail(any(), any())).thenReturn(res);
 
         // When
-        EmailResDto emailResDto = mailSender.sendMail(emailCertificationDto);
+        EmailResDto emailResDto = mailSender.sendMail(emailCertificationDto, emailBodyGenerator);
 
         // Then
         assertThat(emailResDto.getMail()).isEqualTo(emailCertificationDto.getEmail());