Add logic around mocking framework.