Adding junit test cases