EasyMock无法在java.sql.Connection对象上模拟close（）方法

Question

我无法模仿close()对象上的java.sql.Connection方法。我的最终目标是在运行JUnit测试时进行连接池。这将消除单元测试期间的连接开销，因为测试中的代码为每个SQL请求执行connect / close。

我有代码片段来演示这个问题。它成功地模拟了connect（），但未能模拟close()调用。连接和关闭都必须模拟到托管池连接。

由于close()方法未返回值，因此我使用EasyMock.expectLastCall。

以下代码定义了mock，但是close()的print语句永远不会执行，我的连接也会关闭。

public class ConnectionPoolTest {

    public static final String DATASOURCE_REF = "jdbc/cloudexDS";
    public static final String ref = "osgi:service/javax.sql.DataSource/(osgi.jndi.service.name=" + DATASOURCE_REF + ")";

    @BeforeClass
    public static void setUpBeforeClass() throws Exception 
    {
        //
        //  set up a simple database connection (e.g. DB2SimpleDataSource) 
        //
        DataSource dbds = connect("localhost", "dbname", "userid", "password", 50000);

        //
        //  now set up mock for the datasource to use the above connection
        //
        Context context = EasyMock.createNiceMock(Context.class);
        System.setProperty(Context.INITIAL_CONTEXT_FACTORY, CEXInitialContextFactory.class.getName());
        CEXInitialContextFactory.setCurrentContext(context);

        try {
            expect(context.lookup(ref)).andReturn(dbds).anyTimes();
            replay(context);
        } catch (NamingException e1) {
            e1.printStackTrace();
        }

        //
        //  mock the close()
        //
        Connection conn = EasyMock.createNiceMock(Connection.class);
        try {
            conn.close();
            expectLastCall().andAnswer(new IAnswer<Object>() {
                public Object answer() {
                    System.out.println("Close connection");
                    return null;
                }
            });
            replay(conn);
        } catch (Exception e1) {
            e1.printStackTrace();
        }
    }

    @Test
    public void testMakeRequest() 
    {
        try {
            Connection conn = getConnection();
            ResultSet rset = conn.createStatement().executeQuery("SELECT COUNT(*) FROM SYSCAT.TABLES");
            if(rset.next()) {
                System.out.println("Count=" + rset.getString(1));
            }
            conn.close();
        } catch (Exception ex) {
            fail("Exception thrown, msg=" + ex.getMessage());
        }
    }

    /**
     * get a connection to the database
     * 
     * Note:  The ic.lookup is mocked and will return a database connection
     * 
     * @return  connection object
     * @throws NamingException
     * @throws SQLException
     */
    private Connection getConnection() throws NamingException, SQLException 
    {
        DataSource cloudexds = null;

        try {
            InitialContext ic = new InitialContext();
            cloudexds = (DataSource) ic.lookup(ref);
        } catch (NamingException e) {
            e.printStackTrace();
        }
        return cloudexds.getConnection();
    }

    /**
     * setup a connection to the database.  This will be used with mock
     * 
     * @param hostname
     * @param dbname
     * @param schema
     * @param userid
     * @param password
     * @param port
     * @return
     */
    private static DataSource connect(String hostname, String dbname, String userid, String password, Integer port)
    {
        DB2SimpleDataSource dbds = new DB2SimpleDataSource();
        //
        //  local 'docker' database
        //
        dbds.setDatabaseName(dbname);
        dbds.setUser(userid);
        dbds.setPassword(password);
        dbds.setServerName(hostname);
        dbds.setPortNumber(port);
        dbds.setDriverType(4);

        try {
            dbds.getConnection();
        } catch (Exception ex) {
            fail("Cannot connect to the " + dbname + " database, msg=" + ex.getMessage());
        }
        return dbds;
    }
}

我正在使用EasyMock 3.3。

感谢任何建议。