Added heavily mocked testin