增加对单元测试的支持