adding unit tests, mockito dependency