Moved mock to mocked object's test