单元测试的威力

为一段代码补充单元测试后出现了让人郁闷的red bar和蓝色的Failures，这是一段中古时期的代码，读取.properties文件并返回Properties句柄：

1. public class LoadProp {
3.     /**
4.      * 读取.properties配置文件,返回Properties
5.      * @param path 文件路径
6.      * @return Properties
7.      */
8.     public Properties getProp(String path) {
9.         InputStream inStream = null;
10.         
11.         try {
12.             inStream = getInStream(path);
13.             Properties prop = new Properties();
14.             prop.load(inStream);
15.             inStream.close();
16.             
17.             return prop;
18.         } catch (IOException e) {
19.             e.printStackTrace();
20.             return null;
21.         } finally {
22.             if(inStream != null) {
23.                 try {
24.                     inStream.close();
25.                 } catch (IOException e) {
26.                     // TODO Auto-generated catch block
27.                     e.printStackTrace();
28.                 }
29.             }
30.         }
31.     }
32.     
33.     protected InputStream getInStream(String path) {
34.         return this.getClass().getClassLoader().getResourceAsStream(path);
35.     }
36. }

当然，这段代码还有很大的改进余地，可以使用ICO和method factory进行重构，抛开其结构优劣不提，仅对功能进行测试，以下是单元测试代码：

1、MockInputStream

2. public class MockInputStream  extends InputStream {
4.     private int position = 0;
5.     private int closeCount = 0;
6.     private String buffer;
7.     
8.     public void setBuffer(String buffer) {
9.         this.buffer = buffer;
10.     }
11.     
12.     public int read() throws IOException {
13.         if(position == this.buffer.length()) {
14.             return -1;
15.         }
16.         
17.         return this.buffer.charAt(position++);
18.     }
19.     
20.     public void close() throws IOException {
21.         this.closeCount++;
22.         super.close();
23.     }
24.     
25.     public void verify() throws AssertionFailedError {
26.         if(closeCount != 1) {
27.             throw new AssertionFailedError("close() should have been called once and once only!" +
28.                     " This close is " + closeCount);
29.         }
30.     }
31. }

2、TestTableLoadProp

1. public class TestTableLoadProp extends LoadProp {
3.     private InputStream inputStream;
5.     public InputStream getInStream(String path) {
6.         
7.         return inputStream;
8.     }
10.     public void setInputStream(InputStream inputStream) {
11.         this.inputStream = inputStream;
12.     }
13. }

3、TestLoadProp

1. public class TestLoadProp extends TestCase {
3.     private MockInputStream mockInputStream;
4.     private TestTableLoadProp prop;
5.     
6.     public void setUp() {
7.         mockInputStream = new MockInputStream();
8.         mockInputStream.setBuffer("invoiceManager.roleId = 5862");
9.         prop = new TestTableLoadProp();
10.         prop.setInputStream(mockInputStream);
11.     }
12.     
13.     public void testGetProp() {
14.         this.assertNotNull(prop.getProp(Variable.FREEZETIME));
15.     }
16.     
17.     public void tearDown() {
18.         mockInputStream.verify();
19.     }
20. }

运行TestLoadProp将会提示测试失败：“close() should have been called once and once only! This close is 2”

问题再明显不过。

 

这个问题很隐蔽，而且并非人人都喜欢掘墓，挖出中古时期的代码，由此可见单元测试的威力。