Add mocking helper for better testablility
0.2.1