Write unit tests using junit/mockito

Complete the test cases for all the classes in the src/test/java directory.

Ensure each method being tested has all conditions tested.

Ensure you mock the dependencies and verify with Mockito that the proper methods were called