[Misc] Convert test to Mockito + add some unit tests + really make it a unit test (and not an integration test)