스프링부트 게시판 API 만들기 (13) - 게시글 서비스 구현, 파일 저장 서비스 구현

 

 

 

 

 

이번 글에서는 Post에 대한 서비스를 개발하고, 파일을 저장하는 File Service까지 작성해보도록 하겠습니다.

오늘은 제가 부스터샷을 맞아 몸상태가 좋지 않은 관계로 테스트코드까지 작성할 힘은 없어서, 테스트코드는 이후 포스팅에서 작성하도록 하겠습니다.

 

• 시큐리티를 이용한 JSON 데이터로 로그인 (완료)
• JWT를 이용한 인증 (완료)
• 도메인, 테이블 설계, 엔티티 생성 (완료)
• 댓글 삭제 로직 구현 (완료)
• 회원가입 + 정보수정 등 회원 서비스 구현 (완료)
• 게시판 서비스 구현 (진행 중)
• 댓글 서비스 구현 (1댓글 -> *(무한) 대댓글 구조)
• 예외 처리 (완료)
• 예외 메세지 국제화
• 카테고리별 게시판 분류
• 게시글 페이징
• 동적인 검색 조건을 사용한 검색
• 사용자 간 쪽지 기능
• 무한 쪽지 스크롤
• 게시물 & 댓글에 대한 알람
• 쪽지에 대한 알람
• 접속한 사용자 간 실시간 채팅
• 회원가입 시 검증(예: XX대학교 XX과가 아니면 가입할 수 없게)
• Swagger를 사용한 API 문서 만들기
• 신고 & 블랙리스트 기능
• AOP를 통한 로그
• 어드민 페이지
• 캐시
• 배포 (+ 무중단 배포)
• 배포 자동화
• 포트와 어댑터 설계를 따르는 패키지 구조 설계하기
• ...

 

 

 

 

 

 

 

 

 

 

게시글(Post) 서비스 구현하기

현재 Post 클래스는 다음과 같습니다.

@Table(name = "POST")
@Getter
@NoArgsConstructor(access = AccessLevel.PROTECTED)
@Entity
public class Post extends BaseTimeEntity {

    @Id @GeneratedValue(strategy = IDENTITY)
    @Column(name = "post_id")
    private Long id;


    @ManyToOne(fetch = LAZY)
    @JoinColumn(name = "writer_id")
    private Member writer;


    @Column(length = 40, nullable = false)
    private String title;

    @Lob
    @Column(nullable = false)
    private String content;

    @Column(nullable = true)
    private String filePath;

    @Builder
    public Post(String title, String content) {
        this.title = title;
        this.content = content;
    }

    //== 게시글을 삭제하면 달려있는 댓글 모두 삭제 ==//
    @OneToMany(mappedBy = "post", cascade = ALL, orphanRemoval = true)
    private List<Comment> commentList = new ArrayList<>();




    //== 연관관계 편의 메서드 ==//
    public void confirmWriter(Member writer) {
        //writer는 변경이 불가능하므로 이렇게만 해주어도 될듯
        this.writer = writer;
        writer.addPost(this);
    }
    public void addComment(Comment comment){
        //comment의 Post 설정은 comment에서 함
        commentList.add(comment);
    }



    //== 내용 수정 ==//
    public void updateTitle(String title) {
        this.title = title;
    }

    public void updateContent(String content) {
        this.content = content;
    }

    public void updateFilePath(String filePath) {
        this.filePath = filePath;
    }
}

수정된 코드는 없으니, 지금까지 작성한 코드가 맞는지 확인만 해주시면 됩니다.

이제 Post 서비스 개발을 시작해보도록 하겠습니다.

 

 

 

 

 

 

 

 

PostRepository 구현

Jpa를 사용할 것이므로 JpaRepository를 상속받아 구현하겠습니다.

위치는 아래와 같습니다.

public interface PostRepository extends JpaRepository<Post, Long> {
}

 

 

 

 

 

 

 

 

 

 

 

 

 

PostService 구현

PostService는 게시글의 CRUD 기능을 가지고 있습니다. 이후 검색 조건을 추가하여 조건에 따라 검색하고, 페이징 처리를 하게끔 구현하겠지만, 우선은 댓글 기능을 만들기 전이기 때문에, 조회 기능은 구현하지 않고, 생성과 수정, 삭제에 대해서만 구현하겠습니다.

 

위치는 다음과 같습니다.

 

구현

public interface PostService {

    /**
     * 게시글 등록
     */
    void save(PostSaveDto postSaveDto) throws FileException;

    /**
     * 게시글 수정
     */
    void update(Long id, PostUpdateDto postUpdateDto);

    /**
     * 게시글 삭제
     */
    void delete(Long id);

    /**
     * 게시글 1개 조회
     */
    PostInfoDto getPostInfo(Long id);


    /**
     * 검색 조건에 따른 게시글 리스트 조회 + 페이징
     */
    PostPagingDto getPostList(Pageable pageable, PostSearchCondition postSearchCondition);
}

Pageable은 이후 Controller에서 Pageable 객체를 바로 받아와서 사용할 수 있습니다.

page와 size에 대한 정보를 가지고 있습니다.

 

 

이제 DTO들과, 검색 조건인 PostSearchCondition을 생성해 주도록 하겠습니다.

먼저 DTO부터 생성하도록 하겠습니다

 

 

 

 

 

 

 

 

 

 

 

 

DTO 생성

위치는 다음과 같습니다.

 

 

PostSaveDto 생성

public record PostSaveDto(@NotBlank(message = "제목을 입력해주세요") String title,
                          @NotBlank(message = "내용을 입력해주세요") String content, 
                          Optional<MultipartFile> uploadFile) {

    public Post toEntity() {
        return Post.builder().title(title).content(content).build();
    }
}

제목(title)과 내용(content), 그리고 업로드할 파일(uploadFile)을 가진 record 입니다. 파일을 여러개 업로드할 수 있게 만드시고 싶으시다면 List로 받아오시면 됩니다.

 

 

 

 

 

 

 

 

PostUpdateDto 생성

public record PostUpdateDto(
        Optional<String> title,
        Optional<String> content,
        Optional<MultipartFile> uploadFile) {
}

저장하는 경우와 마찬가지로, 바꿀 제목과 내용, 업로드할 파일을 받아옵니다.

 

 

 

 

 

 

 

 

PostInfoDto, PostPagingDto 생성

이들은 이후 댓글과 대댓글을 구현하면, 댓글과 대댓글을 추가할 것이기에, 잠시 클래스만 생성해 둔 후 구현은 하지 않도록 하겠습니다.

public class PostInfoDto {
}

public class PostPagingDto {
}

 

 

 

 

 

 

 

 

PostSearchCondition 생성

검색 조건도, 댓글과 대댓글을 모두 구현하고 난 후 생각해 보도록 하겠습니다. 우선 클래스만 만들겠습니다.

위치는 다음과 같습니다.

 

 

public class PostSearchCondition {
}

 

 

 

 

 

 

 

 

 

이제 마지막으로 PostService를 구현해 보도록 하겠습니다.

PostServiceImpl 구현

@Service
@RequiredArgsConstructor
@Transactional
public class PostServiceImpl implements PostService{

    private final PostRepository postRepository;
    private final MemberRepository memberRepository;
    private final FileService fileService;

    @Override
    public void save(PostSaveDto postSaveDto) throws FileException {
        Post post = postSaveDto.toEntity();

        post.confirmWriter(memberRepository.findByUsername(SecurityUtil.getLoginUsername())
                .orElseThrow(() -> new MemberException(MemberExceptionType.NOT_FOUND_MEMBER)));

        postSaveDto.uploadFile().ifPresent(
                file ->  post.updateFilePath(fileService.save(file))
        );
        
        postRepository.save(post);
    }



    @Override
    public void update(Long id, PostUpdateDto postUpdateDto) {

        Post post = postRepository.findById(id).orElseThrow(() ->
                new PostException(PostExceptionType.POST_NOT_POUND));

        checkAuthority(post,PostExceptionType.NOT_AUTHORITY_UPDATE_POST );

        postUpdateDto.title().ifPresent(post::updateTitle);
        postUpdateDto.content().ifPresent(post::updateContent);


        if(post.getFilePath() !=null){
            fileService.delete(post.getFilePath());//기존에 올린 파일 지우기
        }
        
        postUpdateDto.uploadFile().ifPresentOrElse(
                multipartFile ->  post.updateFilePath(fileService.save(multipartFile)),
                () ->  post.updateFilePath(null)
                );
    }


    @Override
    public void delete(Long id) {

        Post post = postRepository.findById(id).orElseThrow(() ->
                new PostException(PostExceptionType.POST_NOT_POUND));

        checkAuthority(post,PostExceptionType.NOT_AUTHORITY_DELETE_POST);


        if(post.getFilePath() !=null){
            fileService.delete(post.getFilePath());//기존에 올린 파일 지우기
        }

        postRepository.delete(post);

    }


    private void checkAuthority(Post post, PostExceptionType postExceptionType) {
        if(!post.getWriter().getUsername().equals(SecurityUtil.getLoginUsername()))
            throw new PostException(postExceptionType);
    }


    @Override
    public PostInfoDto getPostInfo(Long id) {
        return null;
    }

    @Override
    public PostPagingDto getPostList(Pageable pageable, PostSearchCondition postSearchCondition) {
        return null;
    }
}

getPostInfo와 getPostList는 이후 댓글과 대댓글을 구현한 후에 구현하도록 하겠습니다.

 

설명

save 메소드 하나만 살펴보겠습니다.

 

먼저 PostSaveDto에서 title과 content를 가지고 Post 객체를 생성합니다.

이후 로그인한 사용자를 작성자로 매핑시킵니다. (로그인 한 작성자가 쓴 글이기 때문입니다.)

이후 FileService를 이용해 File을 서버에 저장시킵니다. (현재는 컴퓨터상에 저장되게 했습니다만, 이후 AWS를 사용하여, S3 등에 저장하도록 바꾸겠습니다.)

그 다음 저장된 경로를 save의 반환값으로 가져와 이를 post의 uploadFilePath에 저장합니다.

이후 해당 Post 엔티티를 DB에 저장시켜줍니다.

 

 

 

 

 

 

 

 

 

 

 

FileService 구현

이제 file을 서버에 저장하는 역할을 담당하는 fileService를 구현하도록 하겠습니다.

 

위치는 global -> file -> service 입니다.

 

 

public interface FileService {
    
    //저장된 파일 경로 반환
    String save(MultipartFile multipartFile) throws FileException;

    void delete(String filePath);

}

 

 

 

이를 구현하기 전에 파일 저장과 관련하여 설정을 조금 하고 넘어가도록 하겠습니다.

 

 

 

 

 

 

 

 

 

application.yml 수정

해당 코드를 추가해줍니다. 

dir에는 파일을 저장할 폴더의 경로를 지정해 주시면 됩니다.

저는 아래 위치해 추가해 주도록 하겠습니다. (폴더를 미리 만들어 주셔야 합니다.)

file:
  dir: D:\files\

 

 

 

 

전체 application.yml 파일은 다음과 같습니다.

spring:
  datasource:
    driver-class-name: org.h2.Driver
    url: jdbc:h2:tcp://localhost/~/jpa
    username: sa
    password: 1


  jpa:
    properties:
      hibernate:
        format_sql: true
        user_sql_cooments: true

    hibernate:
      ddl-auto: create


  profiles:
    include: jwt



logging:
  level:
    org:
      apache:
        coyote:
          http11: debug

      hiberante:
        SQL: debug

    boardexample:
      myboard: info


file:
  dir: D:\files\

 

 

이제 구현을 해보도록 하겠습니다.

 

 

 

 

 

 

 

 

 

FileServiceImpl 구현

@Service
public class FileServiceImpl implements FileService{

    @Value("${file.dir}")//application.yml 파일에 있는 file.dir의 내용을 가져옴
    private String fileDir;



    //TODO : 테스트코드 작성해야 함

    @Override
    public String save(MultipartFile multipartFile)  {
        String filePath = fileDir + UUID.randomUUID();
        try {
            multipartFile.transferTo(new File(filePath));
        }catch (IOException e){
            //파일 저장 에러!
            throw new FileException(FileExceptionType.FILE_CAN_NOT_SAVE);
        }


        return filePath;
    }

    @Override
    public void delete(String filePath) {
            File file = new File(filePath);

            //존재하지 않는데 굳이 지우나..?
            if(!file.exists()) return;

            if(!file.delete()) throw new FileException(FileExceptionType.FILE_CAN_NOT_DELETE);

    }
}

 

우선 @Value를 사용한 코드부터 살펴보겠습니다.

예전에 JWT를 발급할 때 다뤘듯이 application.yml 파일에서 설정한 값을 가져와서 사용하는 코드입니다.

 

이후 save 메서드를 보시면, 파일이 저장될 경로인 filePath는 우리가 지정한 fileDir에, UUID를 발급받아 파일 이름을 겹치지 않고 랜덤하게 저장해주도록 하였습니다.

이후 transferTo 메서드를 사용하여 파일을 컴퓨터에 저장해준 후, 저장된 경로를 반환합니다.

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

예외 처리

기본적인 코드 작성은 끝났고, 이제 Post와 File Service에 대한 예외 처리를 진행해주도록 하겠습니다.

 

 

FileException 구현

위치는 아래와 같습니다.

 

public class FileException extends BaseException {
    private BaseExceptionType exceptionType;


    public FileException(BaseExceptionType exceptionType) {
        this.exceptionType = exceptionType;
    }

    @Override
    public BaseExceptionType getExceptionType() {
        return exceptionType;
    }
}

 

 

 

 

 

 

public enum FileExceptionType implements BaseExceptionType {

    FILE_CAN_NOT_SAVE(10000, HttpStatus.BAD_REQUEST, "파일 저장에 실패했습니다."),
    FILE_CAN_NOT_DELETE(10001, HttpStatus.BAD_REQUEST, "파일 삭제에 실패했습니다.");




    private int errorCode;
    private HttpStatus httpStatus;
    private String errorMessage;

    FileExceptionType(int errorCode, HttpStatus httpStatus, String errorMessage) {
        this.errorCode = errorCode;
        this.httpStatus = httpStatus;
        this.errorMessage = errorMessage;
    }

    @Override
    public int getErrorCode() {
        return this.errorCode;
    }

    @Override
    public HttpStatus getHttpStatus() {
        return this.httpStatus;
    }

    @Override
    public String getErrorMessage() {
        return this.errorMessage;
    }
}

 

파일 저장에 실패했을 때와, 파일 삭제에 실패했을 경우 발생하는 IOException을 커스텀 에러로 전환해주기 위해서 작성했습니다.

 

 

 

 

 

 

 

 

 

 

 

 

 

 

PostException 구현

위치는 아래와 같습니다.

 

public class PostException extends BaseException {

    private BaseExceptionType baseExceptionType;


    public PostException(BaseExceptionType baseExceptionType) {
        this.baseExceptionType = baseExceptionType;
    }

    @Override
    public BaseExceptionType getExceptionType() {
        return this.baseExceptionType;
    }
}

 

 

 

 

 

public enum PostExceptionType implements BaseExceptionType {

    POST_NOT_POUND(700, HttpStatus.NOT_FOUND, "찾으시는 포스트가 없습니다"),
    NOT_AUTHORITY_UPDATE_POST(701, HttpStatus.FORBIDDEN, "포스트를 업데이트할 권한이 없습니다."),
    NOT_AUTHORITY_DELETE_POST(702, HttpStatus.FORBIDDEN, "포스트를 삭제할 권한이 없습니다.");


    private int errorCode;
    private HttpStatus httpStatus;
    private String errorMessage;

    PostExceptionType(int errorCode, HttpStatus httpStatus, String errorMessage) {
        this.errorCode = errorCode;
        this.httpStatus = httpStatus;
        this.errorMessage = errorMessage;
    }


    @Override
    public int getErrorCode() {
        return this.errorCode;
    }

    @Override
    public HttpStatus getHttpStatus() {
        return this.httpStatus;
    }

    @Override
    public String getErrorMessage() {
        return this.errorMessage;
    }
}

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

테스트코드 작성 - PostService

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertThrows;


@SpringBootTest
@Transactional
class PostServiceImplTest {


    @Autowired private EntityManager em;

    @Autowired
    private PostService postService;

    @Autowired
    private MemberService memberService;

    private static final String USERNAME = "username";
    private static final String PASSWORD = "PASSWORD123@@@";



    private void clear(){
        em.flush();
        em.clear();
    }

    private void deleteFile(String filePath) {
        File files = new File(filePath);
        files.delete();
    }

    private MockMultipartFile getMockUploadFile() throws IOException {
        return new MockMultipartFile("file", "file.jpg", "image/jpg", new FileInputStream("C:/Users/user/Desktop/tistory/diary.jpg"));
    }



    @BeforeEach
    private void signUpAndSetAuthentication() throws Exception {
        memberService.signUp(new MemberSignUpDto(USERNAME,PASSWORD,"name","nickName",22));
        SecurityContext emptyContext = SecurityContextHolder.createEmptyContext();
        emptyContext.setAuthentication(
                new UsernamePasswordAuthenticationToken(
                        User.builder()
                                .username(USERNAME)
                                .password(PASSWORD)
                                .roles(Role.USER.toString())
                                .build(),
                        null)
        );
        SecurityContextHolder.setContext(emptyContext);
        clear();
    }




    @Test
    public void 포스트_저장_성공_업로드_파일_없음() throws Exception {
        //given
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title, content, Optional.empty());


        //when
        postService.save(postSaveDto);
        clear();


        //then
        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        Post post = em.find(Post.class, findPost.getId());
        assertThat(post.getContent()).isEqualTo(content);
        assertThat(post.getWriter().getUsername()).isEqualTo(USERNAME);
        assertThat(post.getFilePath()).isNull();
    }





    @Test
    public void 포스트_저장_성공_업로드_파일_있음() throws Exception {
        //given

        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title,content, Optional.ofNullable(getMockUploadFile()));

        //when
        postService.save(postSaveDto);
        clear();

        //then
        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        Post post = em.find(Post.class, findPost.getId());
        assertThat(post.getContent()).isEqualTo(content);
        assertThat(post.getWriter().getUsername()).isEqualTo(USERNAME);
        assertThat(post.getFilePath()).isNotNull();

        deleteFile(post.getFilePath());
        //올린 파일 삭제
    }




    @Test
    public void 포스트_저장_실패_제목이나_내용이_없음() throws Exception {
        //given
        String title = "제목";
        String content = "내용";

        PostSaveDto postSaveDto = new PostSaveDto(null,content, Optional.empty());
        PostSaveDto postSaveDto2 = new PostSaveDto(title,null, Optional.empty());


        //when,then
        assertThrows(Exception.class, () -> postService.save(postSaveDto));
        assertThrows(Exception.class, () -> postService.save(postSaveDto2));

    }




    @Test
    public void 포스트_업데이트_성공_업로드파일_없음TO없음() throws Exception {
        //given
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title,content, Optional.empty());
        postService.save(postSaveDto);
        clear();

        //when
        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        PostUpdateDto postUpdateDto = new PostUpdateDto(Optional.ofNullable("바꾼제목"),Optional.ofNullable("바꾼내용"), Optional.empty());
        postService.update(findPost.getId(),postUpdateDto);
        clear();

        //then
        Post post = em.find(Post.class, findPost.getId());
        assertThat(post.getContent()).isEqualTo("바꾼내용");
        assertThat(post.getWriter().getUsername()).isEqualTo(USERNAME);
        assertThat(post.getFilePath()).isNull();

    }



    @Test
    public void 포스트_업데이트_성공_업로드파일_없음TO있음() throws Exception {
        //given
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title,content, Optional.empty());
        postService.save(postSaveDto);
        clear();


        //when
        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();

        PostUpdateDto postUpdateDto = new PostUpdateDto(Optional.ofNullable("바꾼제목"),Optional.ofNullable("바꾼내용"), Optional.ofNullable(getMockUploadFile()));
        postService.update(findPost.getId(),postUpdateDto);
        clear();


        //then
        Post post = em.find(Post.class, findPost.getId());
        assertThat(post.getContent()).isEqualTo("바꾼내용");
        assertThat(post.getWriter().getUsername()).isEqualTo(USERNAME);
        assertThat(post.getFilePath()).isNotNull();

        deleteFile(post.getFilePath());
        //올린 파일 삭제
    }




    @Test
    public void 포스트_업데이트_성공_업로드파일_있음TO없음() throws Exception {
        //given
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title,content, Optional.ofNullable(getMockUploadFile()));
        postService.save(postSaveDto);

        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        assertThat(findPost.getFilePath()).isNotNull();
        clear();

        //when
        PostUpdateDto postUpdateDto = new PostUpdateDto(Optional.ofNullable("바꾼제목"),Optional.ofNullable("바꾼내용"), Optional.empty());
        postService.update(findPost.getId(),postUpdateDto);
        clear();

        //then
        findPost = em.find(Post.class, findPost.getId());
        assertThat(findPost.getContent()).isEqualTo("바꾼내용");
        assertThat(findPost.getWriter().getUsername()).isEqualTo(USERNAME);
        assertThat(findPost.getFilePath()).isNull();
    }




    @Test
    public void 포스트_업데이트_성공_업로드파일_있음TO있음() throws Exception {
        //given
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title,content, Optional.empty());
        postService.save(postSaveDto);

        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        Post post = em.find(Post.class, findPost.getId());
        String filePath = post.getFilePath();
        clear();

        //when
        PostUpdateDto postUpdateDto = new PostUpdateDto(Optional.ofNullable("바꾼제목"),Optional.ofNullable("바꾼내용"), Optional.ofNullable(getMockUploadFile()));
        postService.update(findPost.getId(),postUpdateDto);
        clear();

        //then
        post = em.find(Post.class, findPost.getId());
        assertThat(post.getContent()).isEqualTo("바꾼내용");
        assertThat(post.getWriter().getUsername()).isEqualTo(USERNAME);
        assertThat(post.getFilePath()).isNotEqualTo(filePath);
        deleteFile(post.getFilePath());
        //올린 파일 삭제
    }




    private void setAnotherAuthentication() throws Exception {
        memberService.signUp(new MemberSignUpDto(USERNAME+"123",PASSWORD,"name","nickName",22));
        SecurityContext emptyContext = SecurityContextHolder.createEmptyContext();
        emptyContext.setAuthentication(
                new UsernamePasswordAuthenticationToken(
                        User.builder()
                                .username(USERNAME+"123")
                                .password(PASSWORD)
                                .roles(Role.USER.toString())
                                .build(),
                        null)
        );
        SecurityContextHolder.setContext(emptyContext);
        clear();
    }


    @Test
    public void 포스트_업데이트_실패_권한이없음() throws Exception {
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title, content, Optional.empty());

        postService.save(postSaveDto);
        clear();


        //when, then
        setAnotherAuthentication();
        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        PostUpdateDto postUpdateDto = new PostUpdateDto(Optional.ofNullable("바꾼제목"),Optional.ofNullable("바꾼내용"), Optional.empty());


        assertThrows(PostException.class, ()-> postService.update(findPost.getId(),postUpdateDto));

    }




    @Test
    public void 포스트삭제_성공() throws Exception {
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title, content, Optional.empty());
        postService.save(postSaveDto);
        clear();

        //when
        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        postService.delete(findPost.getId());

        //then
        List<Post> findPosts = em.createQuery("select p from Post p", Post.class).getResultList();
        assertThat(findPosts.size()).isEqualTo(0);
    }



    @Test
    public void 포스트삭제_실패() throws Exception {
        String title = "제목";
        String content = "내용";
        PostSaveDto postSaveDto = new PostSaveDto(title, content, Optional.empty());

        postService.save(postSaveDto);
        clear();

        //when, then
        setAnotherAuthentication();
        Post findPost = em.createQuery("select p from Post p", Post.class).getSingleResult();
        assertThrows(PostException.class, ()-> postService.delete(findPost.getId()));
    }

}

 

MockMultipartFile의 생성자에는 순서대로 name,  originalFilename, contentType, FileInputStream("파일경로")를 넣어주시면 됩니다.

제 코드를 참고하셔서 작성해주시면 됩니다.

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

테스트코드 작성 - FileService

import static org.assertj.core.api.Assertions.*;


@SpringBootTest
class FileServiceTest {

    @Autowired FileService fileService;


    private MockMultipartFile getMockUploadFile() throws IOException {
        return new MockMultipartFile("file", "file.jpg", "image/jpg", new FileInputStream("C:/Users/user/Desktop/tistory/diary.jpg"));
    }

    @Test
    public void 파일저장_성공() throws Exception {
        //given, when
        String filePath = fileService.save(getMockUploadFile());

        //then
        File file = new File(filePath);
        assertThat(file.exists()).isTrue();

        //finally
        file.delete();//파일 삭제


    }

    @Test
    public void 파일삭제_성공() throws Exception {
        //given, when
        String filePath = fileService.save(getMockUploadFile());
        fileService.delete(filePath);

        //then
        File file = new File(filePath);
        assertThat(file.exists()).isFalse();
        
    }

}

 

 

 

 

 

 

 

 

 

 

이렇게 해서 간단한 Post에 대한 서비스와, 파일을 저장하는 코드를 작성해 보았습니다. 다음번에는 댓글과 대댓글을 구현해보도록 하겠습니다. 감사합니다.

 

 

 

전체 코드는 깃허브에서 확인하실 수 있습니다.

(추가로 제가 정신이 없어서 고치고 반영을 하지 못한 부분이 있을 수도 있기에, 코드가 제대로 작동하지 않는다면 언제든지 댓글이나 메일 보내주시면 확인하도록 하겠습니다.)

 

https://github.com/ShinDongHun1/SpringBoot-Board-API

GitHub - ShinDongHun1/SpringBoot-Board-API

 

 

 

 

 

 

 

 

 

이상하거나 오류가 나는 부분이 있다면 댓글로 알려주시면 감사하겠습니다.

광고도 한번씩 눌러주시면 감사하겠습니다  :)