The latest test code followed the code review.

Pls give your view again.