Added JUnit tests for use case testing