fix: 초대코드 인증 정보를 저장할 때 status도 저장한다. (#54)

YAPP-Github · Jan 22, 2024 · 5b3552b · 5b3552b
1 parent 67f3c3f
commit 5b3552b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/main/java/com/moneymong/domain/invitationcode/entity/InvitationCodeCertification.java b/src/main/java/com/moneymong/domain/invitationcode/entity/InvitationCodeCertification.java
@@ -41,10 +41,11 @@ public class InvitationCodeCertification {
     @Column(name = "updated_at")
     private ZonedDateTime updatedAt;
 
-    public static InvitationCodeCertification of(Long userId, Long agencyId) {
+    public static InvitationCodeCertification of(Long userId, Long agencyId, CertificationStatus status) {
         return InvitationCodeCertification.builder()
                 .userId(userId)
                 .agencyId(agencyId)
+                .status(status)
                 .build();
     }
 

diff --git a/src/main/java/com/moneymong/domain/invitationcode/service/InvitationCodeService.java b/src/main/java/com/moneymong/domain/invitationcode/service/InvitationCodeService.java
@@ -6,6 +6,7 @@
 import com.moneymong.domain.invitationcode.api.request.CertifyInvitationCodeRequest;
 import com.moneymong.domain.invitationcode.api.response.CertifyInvitationCodeResponse;
 import com.moneymong.domain.invitationcode.api.response.InvitationCodeResponse;
+import com.moneymong.domain.invitationcode.entity.CertificationStatus;
 import com.moneymong.domain.invitationcode.entity.InvitationCode;
 import com.moneymong.domain.invitationcode.entity.InvitationCodeCertification;
 import com.moneymong.domain.invitationcode.repository.InvitationCodeCertificationRepository;
@@ -69,7 +70,7 @@ public CertifyInvitationCodeResponse certify(CertifyInvitationCodeRequest reques
 
     private InvitationCodeCertification getCertification(Long userId, Long agencyId) {
         return invitationCodeCertificationRepository.findByUserIdAndAgencyId(userId, agencyId)
-                .orElseGet(() -> InvitationCodeCertification.of(userId, agencyId));
+                .orElseGet(() -> InvitationCodeCertification.of(userId, agencyId, CertificationStatus.DONE));
     }
 
     private AgencyUser getAgencyUser(Long userId, Long agencyId) {