Added JUnit test cases