Added mocked test as well