Added mock tests