[Misc] Convert unit test to mockito/rule