Add mockito for testing
Revenge of the unit tests