Add mocks for new tests