Add Junit test case