Added mock to the robotClass test