[Misc] Convert test to Mockito