Added mock objects feature for tests