[스프링] 순수 JDBC, 스프링 통합 테스트

gyeol·2023년 7월 6일

스프링

목록 보기
6/50
post-thumbnail

스프링 강의 5일차😋

김영한님의 '스프링 입문 - 코드로 배우는 스프링 부트, 웹 MVC, DB 접근 기술'을 듣고 적은 글입니다.

✔️ 순수 JDBC

먼저 H2 database를 설치해준다.
https://www.h2database.com

설치해주고 난 뒤 콘솔창을 띄운다. 그리고 h2 database가 깔린 폴더로 경로를 이동해준 뒤 ./h2.bat을 입력해주면 새로운 창이 뜬다. 그 사이트를 통해 DB를 확인해주면 된다.
❌ 이때 절대 콘솔창을 끄면 안된다!! ❌

일단 build.gradle 파일에 jdbc, h2 데이터베이스 관련 라이브러리를 추가해준다.

implementation 'org.springframework.boot:spring-boot-starter-jdbc'
runtimeOnly 'com.h2database:h2'

그리고 resources > application.properties 에 스프링 부트 데이터베이스 연결 설정을 추가해준다.

spring.datasource.url=jdbc:h2:tcp://localhost/~/test
spring.datasource.driver-class-name=org.h2.Driver
spring.datasource.username=sa

그리고 회원 리포지토리를 구현해준다. 파일은 respository > JdbcMemberRepository.java 위치한다.

package hello.hellospring.repository;
import hello.hellospring.domain.Member;
import org.springframework.jdbc.datasource.DataSourceUtils;
import javax.sql.DataSource;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Optional;
public class JdbcMemberRepository implements MemberRepository {
    private final DataSource dataSource;
    
    public JdbcMemberRepository(DataSource dataSource) {
        this.dataSource = dataSource;
        //DB 받을 수 있음.
    }
    
    @Override
    public Member save(Member member) {
        String sql = "insert into member(name) values(?)";
        //쿼리문을 통해 값 받기

        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);

            pstmt.setString(1, member.getName());

            pstmt.executeUpdate();
            rs = pstmt.getGeneratedKeys(); //RETURN_GENERATED_KEYS와 매칭

            if (rs.next()) {
                member.setId(rs.getLong(1));
                // 값 꺼내기
            } else {
                throw new SQLException("id 조회 실패");
            }

            return member;
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs); //자원 꼭 close, 역순으로
        }
    }

    @Override
    public Optional<Member> findById(Long id) { //조회
        String sql = "select * from member where id = ?";

        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            pstmt.setLong(1, id);

            rs = pstmt.executeQuery();
            if(rs.next()) { //값이 있으면 멤버 객체 생성
                Member member = new Member();
                
                member.setId(rs.getLong("id"));
                member.setName(rs.getString("name"));
                return Optional.of(member);
            } else {
                return Optional.empty();
            }
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs);
        }
    }

    @Override
    public List<Member> findAll() { // 전체 조회
        String sql = "select * from member";
        
        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;
        
        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            rs = pstmt.executeQuery();

            List<Member> members = new ArrayList<>();
            while(rs.next()) {
                Member member = new Member();
                member.setId(rs.getLong("id"));
                member.setName(rs.getString("name"));
                members.add(member);
            }
            return members;
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs);
        }
    }

	//이름으로 회원 찾기
    @Override
    public Optional<Member> findByName(String name) {
        String sql = "select * from member where name = ?";

        Connection conn = null;
        PreparedStatement pstmt = null;
        ResultSet rs = null;

        try {
            conn = getConnection();
            pstmt = conn.prepareStatement(sql);
            pstmt.setString(1, name);
            rs = pstmt.executeQuery();
            
            if(rs.next()) {
                Member member = new Member();
                member.setId(rs.getLong("id"));
                member.setName(rs.getString("name"));
                return Optional.of(member);
            }
            return Optional.empty();
        } catch (Exception e) {
            throw new IllegalStateException(e);
        } finally {
            close(conn, pstmt, rs);
        }
    }

    //일일히 getConnection시키는게 아님. 스프링 통해 dataConnection 쓸 때 DataSourceUtil통해 커넥션해야함.
    private Connection getConnection() {
        return DataSourceUtils.getConnection(dataSource);
    }
    private void close(Connection conn, PreparedStatement pstmt, ResultSet rs) {
        try {
            if (rs != null) {
                rs.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        try {
            if (pstmt != null) {
                pstmt.close();
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
        try {
            if (conn != null) {
                close(conn);
            }
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
    private void close(Connection conn) throws SQLException {
        DataSourceUtils.releaseConnection(conn, dataSource);
    }
}

그리고 service > SpringConfig.java 파일에서 스프링 설정을 변경해준다.

package hello.hellospring;

import hello.hellospring.repository.JdbcMemberRepository;
import hello.hellospring.repository.MemberRepository;
import hello.hellospring.repository.MemoryMemberRepository;
import hello.hellospring.service.MemberService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.sql.DataSource;
import javax.xml.crypto.Data;

@Configuration
public class SpringConfig {

    private DataSource dataSource;

    @Autowired
    public SpringConfig(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    @Bean
    public MemberService memberService(){
        return new MemberService(memberRepository());
    }

    @Bean
    public MemberRepository memberRepository(){
        //return new MemoryMemberRepository();
        return new JdbcMemberRepository(dataSource);
    }
}

여기서 DataSource는 데이터베이스 커넥션을 획득할 때 사용하는 객체이다. 스프링 부트는 데이터베이스 커넥션 정보를 바탕으로 DataSource를 생성하고 스프링 빈으로 만들어두기에 DI를 받을 수 있다.

spring 장점 :

객체지향적 사고. 다형성 활용.
구현체 바꿔끼우기 편리하게 가능(스프링 컨테이너가 지원)
➡️어셈블리 코드만 손보면 된다. ( 어플리케이션 코드는 손보지 않아도 됨 )

개방-폐쇄 원칙 (OCP) :
확장엔 열려있고 수정 변경에는 닫혀있다.
다형성 활용 -> 기능 완전히 변경해도 애플리케이션 전체 수정 필요가 없다.
스프링의 DI를 활용하면 기존 코드를 전혀 손대지않고, 설정만으로 구현 클래스 변경이 가능하다.

✔️스프링 통합 테스트

일단 먼저 테스트 코드를 만들어준다.

package hello.hellospring.service;
import hello.hellospring.domain.Member;
import hello.hellospring.repository.MemberRepository;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.transaction.annotation.Transactional;
import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
@SpringBootTest //스프링 컨테이너와 함께 테스트 실행 
@Transactional //이 친구 덕분에 aftereach 필요 없음

class MemberServiceIntegrationTest {
    @Autowired MemberService memberService;
    @Autowired MemberRepository memberRepository;

    @Test
    public void 회원가입() throws Exception {
        //Given
        Member member = new Member();
        member.setName("spring");
        //When
        Long saveId = memberService.join(member);
        //Then
        Member findMember = memberRepository.findById(saveId).get();
        assertEquals(member.getName(), findMember.getName());
    }
    @Test
    public void 중복_회원_예외() throws Exception {
        //Given
        Member member1 = new Member();
        member1.setName("spring");
        Member member2 = new Member();
        member2.setName("spring");
		//When
        memberService.join(member1);
        IllegalStateException e = assertThrows(IllegalStateException.class,
                () -> memberService.join(member2));//예외가 발생해야 한다.
        assertThat(e.getMessage()).isEqualTo("이미 존재하는 회원입니다.");
    }
}

앞서 만들어 놓은 테스트 케이스와 흐름은 똑같다.
맨 위 어노테이션 @Transactional 을 지우고 테스트 케이스를 2번 실행하면 오류가 발생한다. 왜냐면 spring이 첫 번째 테스트 케이스 때 생성되고 두 번째 케이스 때 생성되면 중복 회원이기에 오류가 발생하는 것이다.
이때 @Transactional 를 이용해 이 문제를 해결할 수 있다.

DB에는 transaction이라는 개념이 존재한다.( 커밋해야 DB에 적용 )

@Transaction :

테스트 케이스에 이 애노테이션이 존재하면, 테스트 시작 전 트랜잭션을 시작하고, 테스트 완료 후에 롤백한다. 이렇게 하면 DB에 데이터가 남지 않기에 다음 테스트에 영향을 주지 않는다.

그래서 다시 원래대로 @Transactional 을 적어주면 아무런 오류없이 잘 돌아가는 것을 확인할 수 있다.

profile
공부 기록 공간 '◡'

0개의 댓글