refactoring test
add Junit