Add mocks to the tests