Java嘲笑DriverManager

Question

我正在尝试编写一个名为DriverManager.getConnection()的方法的单元测试。我正在使用PowerMock 1.5和easymock。不知何故，Powermock无法模拟DriverManager并最终调用真正的DriverManager。 这是我的代码看起来像： - 如下图所示

@Test 
public void checkConnection() { 
    try { 
     String url = "jdbc:oracle:thin:@//myhost:1521/orcl"; 
     String password = "@55"; 
     String dbName = "Halloween"; 
     String username = "Dracula"; 

     PowerMock.mockStatic(DriverManager.class); 

     Connection connection = EasyMock.createMock(Connection.class); 
     List<Datasource> allDs = new ArrayList<Datasource>(2); 
     Datasource d1 = new Datasource(); 
     d1.setUrl(url); 
     allDs.add(d1); 
     allDs.add(new Datasource()); 

     EasyMock.expect(hibernateTemplate.find("from Datasource ds where ds.dsName = ?", dbName)).andReturn(allDs); 

     EasyMock.expect(DriverManager.getConnection(d1.getUrl(), username, password)).andReturn(connection); 

     PowerMock.replay(DriverManager.class); 
     EasyMock.replay(hibernateTemplate); 

     Connection con = testee.authenticateUserForDatabase(dbName, username, password); 
     Assert.assertNotNull(con); 

     PowerMock.verify(DriverManager.class); 
     EasyMock.verify(hibernateTemplate); 
    } catch (Exception e) { 
     Assert.fail(e.getMessage()); 
    } 
} 

我的类代码： -

@Override 
public Connection getConnection(String dbName, String username, String password) { 
    Connection connection = null; 
    @SuppressWarnings("unchecked") 
    List<Datasource> ds = hibernateTemplate.find("from Datasource ds where ds.dsName = ?", dbName); 
    if (ds == null || ds.isEmpty()) { 
     throw new ProviderException("Invalid datasource name [" + dbName + "]"); 
    } 
    Datasource d = (Datasource) ds.get(0); 
    int retryCount = 0; 
    boolean connected = false; 
    while (retryCount < 2 && !connected) { 
     try { 
      connection = DriverManager.getConnection(d.getUrl(), username, password); 
      LOG.info("Connected successfully to [{}]", dbName); 
      connected = true; 
     } catch (SQLException e) { 
      LOG.warn("Error: [{}] occured. Going to retry, attempt # [{}]", e.getMessage(), (retryCount + 1)); 
      retryCount++; 
      connected = false; 
     } 
    } 
    return connection; 
} 

Answer 1

您可能会改用javax.sql.DataSource中，而不是使用的DriverManager的。

http://docs.oracle.com/javase/6/docs/api/javax/sql/DataSource.html

这样，您与您可以模拟接口工作。现在大多数应用程序都使用DataSource接口，并且不像这样与DriverManager交互。

所以，你的代码可能会再有一个API，如：

public class DataSourceFactory { 
    public DataSource getDataSource(String url, String usr, String pwd); 
} 

而且你可以使用的DriverManager实现真正的数据源，如果你愿意的话，然后你的单元测试模拟DataSourceFactory。

Answer 2

如果您不能模拟DriverManager.getConnection静态调用，可能的解决方法就是跳过它。

将静态调用提取到您的sut中的受保护方法，并使用与静态方法相同的签名。现在

protected Connection driverManagerGetConnection(String url, String user, String password){ 
    return DriverManager.getConnection(url, user, password); 
} 

，在您的测试，使用您的SUT的EasyMock to create a partial mock，嘲讽driverManagerGetConnection方法。

public void testFoo{ 

    // setup: sut 
    YourSutClass sut = EasyMock.createMockBuilder(YourSutClass.class) 
       .addMockedMethod("driverManagerGetConnection").createMock(); 

    ... 

    // setup: expectations 
    EasyMock.expect(sut.driverManagerGetConnection()).andReturn(mockConnection); 
    ... 

    // exercise 
    EasyMock.replay(sut, mockConnection); 
    sut.whatever(); 

    // verify 
    EasyMock.verify(sut, mockConnection); 
} 

使用这种方法，您唯一没有测试的行是对静态方法的调用。