添加生成测试类的方法