Add mocked unit tests