require should
Refactor mock