updated test class too junit test case