Add more mock test