[JWT] SpringSecurity + JWT를 사용하여 로그인, 회원가입 구현 - 2

Donghoon Jeong·2024년 2월 21일

목록 보기

3/4

전 포스팅에서 Spring Security와 JWT를 사용하여 회원가입과 로그인 기능을 구현해 보았는데
이번 포스팅에서는 Postman을 사용하여 해당 기능들이 잘 작동하는지 테스트를 해보겠습니다.

1. DTO 생성

MemberRequestDTO.java

회원가입 및 로그인 요청을 보내기 위해 사용할 MemberRequestDTO를 생성합니다.

@Getter
@AllArgsConstructor
@NoArgsConstructor
public class MemberRequestDTO {

    @NotBlank
    private String username;

    @NotBlank
    private String password;

    public Member toMember(PasswordEncoder passwordEncoder) {
        return Member.builder()
                .username(username)
                .password(passwordEncoder.encode(password))
                .roles(Collections.singletonList("ROLE_USER"))
                .build();
    }
}

TokenRequestDTO.java

AccessToken을 재발급 받을 때, 사용할 TokenRequestDTO를 생성합니다.

@Getter
@NoArgsConstructor
public class RefreshTokenDTO {

    @NotEmpty
    String refreshToken;
}

2.SecurityUtil 구현

SecurityUtil.java

@Slf4j
@RequiredArgsConstructor
public class SecurityUtil {

    // SecurityContext 에 유저 정보가 저장되는 시점
    // Request 가 들어올 때 JwtFilter 의 doFilter 에서 저장
    public static Long getCurrentMemberId() {
        final Authentication authentication = SecurityContextHolder.getContext().getAuthentication();

        if (authentication == null || authentication.getName() == null) {
            throw  new RuntimeException("Security Context 에 인증 정보가 없습니다.");
        }

        return Long.parseLong(authentication.getName());
    }
}

static method인 getCurrentUsername()을 호출하여 SecurityContext에 저장된 현재 요청을 보낸 회원의 username을 간단하게 얻을 수 있습니다. 이 메서드를 활용하여 해당 회원의 특정 TODO LIST를 조회하는 작업등을 진행하였습니다.

3. MemberController 구현

테스트를 하기 위해 다음과 같은 과정을 진행하겠습니다.

회원가입
로그인을 통해 AccessToken 발급
발급받은 AccessToken을 header에 넣어 API 요청
AccessToken 만료 시, RefresToken을 통해 재발급

다음과 같은 테스트를 진행하기 위해 다음과 같은 MemberController를 구현하였습니다.

MemberController.java

@RestController
@RequestMapping("/auth")
@RequiredArgsConstructor
public class AuthController {
    private final AuthService authService;

    @PostMapping("/signup")
    public ResponseEntity<MemberResponseDTO> signup(@RequestBody @Valid MemberRequestDTO memberRequestDTO) {
        return ResponseEntity.ok(authService.signup(memberRequestDTO));
    }

    @PostMapping("/login")
    public ResponseEntity<TokenDTO> login(@RequestBody @Valid MemberRequestDTO memberRequestDTO) {
        return ResponseEntity.ok(authService.login(memberRequestDTO));
    }

    @PostMapping("/reissue")
    public ResponseEntity<TokenDto> reissue(@RequestBody @Valid RefreshTokenDto refreshTokenDto) {
        return ResponseEntity.ok(authService.reissue(refreshTokenDto));
    }
}