单元测试中如何改良项目代码的整体结构?

单元测试中如何改良项目代码的整体结构?

具有良好整体结构的代码，应该符合“低耦合”的特性，形象的说，就是“各家自扫门前雪、不管他人瓦上霜”，每一个函数、每一个类、每一个模块，都应该只做自己该做的事，不要把应该由“其他人”做的事扯进来。具有良好整体结构的代码就具有“可测性”，否则就不具有“可测性”。

具体来说，如果代码包含不当耦合，当这些代码加入测试工程后，会产生编译错误，或者需要打桩才能测试，从而将不当耦合暴露出来。发现问题后，重构代码、消除不当耦合一般不难，消除不当耦合后，单元测试就可以顺利进行了。

下面是几种典型的不当耦合：

把代码写在界面类中

问题：如果把业务代码写在界面类中，测试时把界面类加入测试工程，会产生编译错误。

解决：把业务代码独立出来，写到相应的实体类中，对这些实体类进行测试。界面类只负责数据的显示和接受用户的输入，具体的计算由实体类负责。

说明：把业务代码写在界面类中，这些代码将很难管理和维护，复用就更谈不上了，这是一定要避免的。

实体类混合了边界代码

问题：例如，一个表示用户的类CUser，它的对象的数据保存在数据库中，如果在CUser类中直接读写数据库，就是实体类混合了边界代码，这也是一种不当耦合，测试CUser类时要察看数据库，或者要打桩，甚至会产生编译错误。

解决：把执行数据库操作的代码写到CDatabase类中，CUser类和CDatabase类没有任何关联，由界面类或用于协调各对象工作的控制类来操作这两个类的对象进行数据读写，现在对CUser类的测试就完全和数据库无关了。

说明：经过重构后，代码的可扩展性、可复用性都有了很大提高，也便于进行单元测试和将来的维护。

无意中形成了高耦合

问题：例如，CMyClass类中一个函数中有这样的语句:

CDemoApp* pApp = (CDemoApp*)AfxGetApp();

UNIT var = pApp->GetXXX()->Getxxxx();

由于CDemoApp是工程的最高层类，可能跟工程中的所有类关联，这两行代码就造成被测试对象跟工程中的所有类耦合。耦合具有扩散性，纵向来说，CMyClass类的所有子类也可能跟工程中的所有类关联，横向来说，所有使用了CMyClass类的类，也可能跟工程中的所有类关联，从而形成了盘根错节的关系。这种问题往往很难发现，但把代码加入测试工程后，一般来说是通不过编译的，从而使问题暴露。

解决：修改被测试代码，去除耦合。把需要从pApp指针中取得的数据改为由参数传进来，这两行代码移到客户程序(即调用被测试函数的代码)中，如果客户程序也是实体类，则继续往上移，直到界面类。

说明：这类无意中形成的耦合非常常见，一般的代码审查、静态扫描都很难发现这类问题，但把代码加入测试工程后，编译器却能轻松发现。这类问题只要及时发现了，消除并不难，经过简单的代码重构后，就可以有效地提高代码质量，特别是提高代码的可复用性，也使单元测试可以顺利进行。

开发工具生成的代码形成了不当耦合

有时候，开发工具生成的代码也形成了不当耦合，例如，VC6.0生成的类代码中，源文件会添加下面的代码：

#include "ProjectName.h" //ProjectName是工程名

如果在另一个不同名的工程中复用代码，这一行就会产生编译错误。进行单元测试时，这一行形成的不当耦合也可能会导致编译错误，因此应删除它。