[Spring] JUnit - 예외조건에 대한 테스트

💡 만약 메소드에 전달된 파리미터 값이 없을 때 어떻게 해결해야 할까?

 

해결 방법 :

1. null과 같은 특별한 값을 리턴
2. 값에 해당하는 정보를 찾을 수 없다고 예외를 던진다.

 

1번은 구현하기 간단하니 2번을 천천히 구현해보자.

 

두번째 방법을 위해 해당하는 정보가 없다는 의미를 가진 예외 클래스가 하나 필요한데

스프링이 미리 정의해 놓은 예외가 존재하며 가져다 써도 된다.

 

지금부터 UserDao의 get() 메소드에 전달된 id값에 해당하는 사용자 정보가 없을 때

예외를 던져 테스트 해보자.

 

예외 클래스로는 EmptyResultDataAccessException을 사용할건데,

먼저 java.org.springframework.dao 라이브러리가 있는지 확인을 해줘야한다.

 

일반적인 테스트는 다 원활이 수행되면 테스트를 성공인데,

예외조건에 대한 테스트는 다르다.

 

이 테스트는 반대로 테스트 진행 중에 특정 예외가 던져지면 테스트가 성공한 것이고,

예외가 던져지지 않고 정상적으로 작업을 마치면 테스트가 실패했다고 판단한다.

 

✏️ 구현 방법

 

@Test 애노테이션의 expected 앨리먼트를 사용해주면 되는데,

expected에 테스트 메소드 실행 중에 발생하리라 기대하는 예외 클래스를 넣어주면 된다.

@Test(expected=ExceptionClass)

 

expected를 추가하면 보통의 테스트와 반대로

정상적으로 메소드를 마치면 테스트가 실패하고, expected에서 지정한 예외가 던져지면 테스트가 성공한다.

 

예외조건에 대한 getUserFailure 테스트 메소드

@Test(expected=EmptyResultDataAccessException.class)
	public void getUserFailure() throws SQLException, ClassNotFoundException {
		ApplicationContext context = new AnnotationConfigApplicationContext(DaoFactory.class);
		
		UserDao dao = context.getBean("userDao", UserDao.class);
		dao.deleteAll();
		assertThat(dao.getCount(), is(0));
		
		dao.get("unknown_id"); 
		
	}

위와 같이 변경해도 아직 테스트는 성공하지 않는다.

 

왜냐하면,

userDao에서 rs.next()를 실행 할 때 가져올 로우가 없기 때문에 SQLException이 발생해

테스트 코드는 EmptyResultDataAccessException이 아닌 SQLException을 받아 테스트가 실패했다고 뜰 것이다.

 

따라서 UserDao의 get() 메소드의 코드도 아래와 같이 수정을 해줘야 한다.

 

public User get(String id) throws ClassNotFoundException, SQLException {
	
		Connection c = dataSource.getConnection();
        PreparedStatement ps = c.prepareStatement(
                "select * from users where id = ?");
		ps.setString(1,  id);
		
		ResultSet rs = ps.executeQuery();
		
		User user = null;
		
		if(rs.next()) {
			user = new User();
			user.setId(rs.getString("id"));
			user.setName(rs.getString("name"));
			user.setPassword(rs.getString("password"));
		}
		
		
		rs.close();
		ps.close();
		c.close();
		
		if(user==null) throw new EmptyResultDataAccessException(1);
        //user에 아무값도 없을 시 EmptyResultDataAccessException 예외를 던져주도록 하자
       
        
        return user;
	}

 

📃 정리

 

이렇게 간단한 DAO는 다양한 테스트를 하지 않아도 괜찮을지도 모르지만,

코드 내용에 상관 없이 포괄적인 테스트를 만들어두는 편히 훨씬 안전하고 유용

 

또한 개발자는 테스트를 직접 만들 때 성공하는 테스트만 골라서 만들지 않도록 주의해야 한다.

"내 PC에서는 잘 되는데"라는 변명을 하지 않도록 해야한다.

 

스프링의 창시자인 로드 존슨이 한 말 처럼 '항상 네거티브 테스트를 먼저 만들라' 를 기억하며,

 

테스트를 작성할 때 부정적인 케이스를 먼저 만드는 습관을 들이자!!

 

코드 저장소 : https://github.com/withseungryu

withseungryu - Overview

참고 문서 : 

토비의 스프링3 (이일민)