Add mocked tests