单元测试的威力

来源：互联网发布：锦衣卫同知编辑：程序博客网时间：2024/04/27 15:18

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

public class LoadProp {
    /**
     * 读取.properties配置文件,返回Properties
     * @param path 文件路径
     * @return Properties
     */
    public Properties getProp(String path) {
        InputStream inStream = null;
        
        try {
            inStream = getInStream(path);
            Properties prop = new Properties();
            prop.load(inStream);
            inStream.close();
            
            return prop;
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        } finally {
            if(inStream != null) {
                try {
                    inStream.close();
                } catch (IOException e) {
                    // TODO Auto-generated catch block 
                    e.printStackTrace();
                }
            }
        }
    }
    
    protected InputStream getInStream(String path) {
        return this.getClass().getClassLoader().getResourceAsStream(path);
    }
}

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

1、MockInputStream

public class MockInputStream  extends InputStream {
    private int position = 0;
    private int closeCount = 0;
    private String buffer;
    
    public void setBuffer(String buffer) {
        this.buffer = buffer;
    }
    
    public int read() throws IOException {
        if(position == this.buffer.length()) {
            return -1;
        }
        
        return this.buffer.charAt(position++);
    }
    
    public void close() throws IOException {
        this.closeCount++;
        super.close();
    }
    
    public void verify() throws AssertionFailedError {
        if(closeCount != 1) {
            throw new AssertionFailedError("close() should have been called once and once only!" +
                    " This close is " + closeCount);
        }
    }
}

2、TestTableLoadProp

public class TestTableLoadProp extends LoadProp {
    private InputStream inputStream;
    public InputStream getInStream(String path) {
        
        return inputStream;
    }
    public void setInputStream(InputStream inputStream) {
        this.inputStream = inputStream;
    }
}

3、TestLoadProp

public class TestLoadProp extends TestCase {
    private MockInputStream mockInputStream;
    private TestTableLoadProp prop;
    
    public void setUp() {
        mockInputStream = new MockInputStream();
        mockInputStream.setBuffer("invoiceManager.roleId = 5862");
        prop = new TestTableLoadProp();
        prop.setInputStream(mockInputStream);
    }
    
    public void testGetProp() {
        this.assertNotNull(prop.getProp(Variable.FREEZETIME));
    }
    
    public void tearDown() {
        mockInputStream.verify();
    }
}

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

问题再明显不过。

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