update the junit test code