add test in junit