update mock test and todo