이 부분에 대한 테스트 코드도 부탁드려도 될까요.