테스트 코드 작성 및 소스 정리