添加测试代码，准备测试