단위 테스트

기능을 개발한후 해당 기능 단위들이 기대한대로 작동하는지 간편하게 테스트하고 싶을때가 있다.
main메서드나 Controller를 통해 실행하는데에는 많은 시간이 소요되니 JUnit이라는 프레임 워크를 사용해서 실행하곤한다.

Member객체

package com.playground2.build_result.domain;

public class Member
{
    private Long id;
    private String name;
    public Long getId() {
        return id;
    }
    public void setId(Long id) {
        this.id = id;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
}

회원 Repository 구현체

package com.playground2.build_result.Repository;

import com.playground2.build_result.domain.Member;

import java.util.*;
/**
 * 동시성 문제가 고려되어 있지 않음, 실무에서는 ConcurrentHashMap, AtomicLong 사용 고려
 */
public class MemoryMemberRepository implements MemberRepository {
    private static Map<Long, Member> store = new HashMap<>();
    private static long sequence = 0L;
    @Override
    public Member save(Member member) {
        member.setId(++sequence);
        store.put(member.getId(), member);
        return member;
    }
    @Override
    public Optional<Member> findById(Long id) {
        return Optional.ofNullable(store.get(id));
    }
    @Override
    public List<Member> findAll() {
        return new ArrayList<>(store.values());
    }
    @Override
    public Optional<Member> findByName(String name) {
        return store.values().stream()
                .filter(member -> member.getName().equals(name))
                .findAny();
    }
    public void clearStore() {
        store.clear();
    }
}

위 리포지토리 클래스가 내가 원하는데로 작동할지 테스트를 해보자.

package com.playground2.build_result.repository;

import com.playground2.build_result.Repository.MemoryMemberRepository;
import com.playground2.build_result.domain.Member;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.Test;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

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

public class MemoryMemberRepositoryTest
{

    MemoryMemberRepository repository = new MemoryMemberRepository();


    //각 테스트 종료될때마다 이기능을 실행하는 어노테이션
    @AfterEach
    public void afterEach()
    {
        //DB에 저장된 데이터를 삭제
        repository.clearStore();
    }

    @Test
    public void save()
    {
        Member member = new Member();
        member.setName("spring");

        //when
        repository.save(member);

        //then
        Member result = repository.findById(member.getId()).get();
        assertThat(result).isEqualTo(member);
    }

    @Test
    public void findByName()
    {
        //given
        Member member1 = new Member();
        member1.setName("spring1");
        repository.save(member1);

        Member member2 = new Member();
        member2.setName("spring2");
        repository.save(member2);

        //when
        Member result = repository.findByName("spring1").get();

        //then
        assertThat(result).isEqualTo(member1);
    }


    @Test
    public void findAll()
    {
        //given
        Member member1 = new Member();
        member1.setName("spring1");
        repository.save(member1);

        Member member2 = new Member();
        member2.setName("spring2");
        repository.save(member2);

        //when
        List<Member> result = repository.findAll();

        //then
        assertThat(result.size()).isEqualTo(2);

    }
    
}

위코드중@AfterEach라는게 있는데 이는 단위테스트를 마칠때마다 실행하는 코드를 설정한다.
여기에서는 Repository가 관리하는 저장소를 비워주는 코드다 (회원 Repository 구현체 에 정의해놓음)
단위테스트마다 저장소를 비워주는 이유는 테스트 간 독립성을 지켜주기 위함이다.
그렇다면 독립성이 지켜지지 않았을경우 테스트가 어떻게 꼬이는지 보여주겠다.

@AfterEach를 쓰지 않았을 경우

위 사진에 필기 해놓았듯이 findAll()테스트에 저장되었던 객체가 findByName()테스트까지 남아있어서
assertThat()코드가 틀리는걸 볼 수 있다.