Updated for JUnit Test

Signed-off-by: jzz128 <jimmyzhangca@gmail.com>