Add mocking testing