Cleaner mocking in unit tests