[Misc] Convert unit test to use Mockito