Question

我正在尝试使用PowerMockito来模拟DBUtil。与典型的测试用例不同，我不想完全模拟db调用。每当调用Dbutil.getConnection（）时。我想将连接对象返回到本地数据库。

当我从@BeforeClass方法调用时，下面的简单jdbc连接代码无效。但是当我从java类调用时它可以工作。

public static Connection getConnection() throws Exception {
    System.out.println("-------- Connecting to  " + Constants.CONNECTION_STR + "  ------");

    try {
        Class.forName(Constants.ORACLE_DRIVER_NAME);
    } 
    catch (ClassNotFoundException e) {          
        throw new Exception("JDBC Driver not found... " + e);           
    }
    catch (Exception e) {
        // TODO: handle exception
        System.out.println("getConnection :: exp :: "+ e);
    }

    System.out.println("Oracle JDBC Driver Registered Sucessfully!");

    Connection connection = null;

    try {
        connection = DriverManager.getConnection(Constants.CONNECTION_STR, Constants.USERNAME, Constants.PASSWORD);
    } 
    catch (SQLException e) {
        throw new Exception("Connection Failed!",e);                        
    }   

    if (connection != null) {
        System.out.println("Connected to Database Sucessfully, take control your database now!");
        return connection;
    } 

    System.out.println("Failed to make connection!");
    return null;
}

我的测试类

@RunWith (PowerMockRunner.class)
@PrepareForTest(DbUtil.class)
public class MyUtilTest {

@Mock
private DbUtil dbUtil;

@InjectMocks
private MyUtil myUtil;

private static Connection myDBConn;


@BeforeClass
public static void beforeClass() throws Exception {           
    myDBConn = OracleJDBCConnetion.getConnection(); // This always throws invalid username/password exception.
}

@Before
public void setUp() throws Exception {
    MockitoAnnotations.initMocks(this);
}

@Test
public void testIsAdminUser() throws Throwable{


    PowerMockito.mockStatic(DbUtil.class);          
    PowerMockito.when(DbUtil.getConnection()).thenReturn(myDBConn);     

    String accId= "TH123" ;
    boolean  isAdmin = MyUtil.isAdminUser(cloudAccGuid);        
    System.out.println("isAdmin  : " + isAdmin);

    //then
    PowerMockito.verifyStatic(Mockito.times(1));
    DbUtil.getConnection();
    assertTrue(isAdmin);


    //Finally I am closing my connection.
    if(myDBConn!=null &&  !myDBConn.isClosed())
        OracleJDBCConnetion.closeConnection(myDBConn);

}

}

beforeClass方法总是抛出预期。

Connection Failed! java.sql.SQLException: ORA-01017: invalid username/password; logon denied

但是当我从普通的Java类尝试时，相同的代码可以工作。

任何人都可以帮助理解这里的错误吗？

我使用的是ojdbc6.jar和powermokito-1.5.6，我的Oracle数据库版本是11.2.0.4.0

感谢。

编辑：我发现@PrepareForTest注释导致错误。没有注释连接是成功的但模拟不起作用。谁能帮我理解发生的事情？我对这些嘲弄的东西很新。

Answer 1

@PrepareForTest注释的问题是，它以递归方式为所有依赖类创建存根。由于DBUtil类使用java.sql.Connection类，因此也为Connection类创建了存根。

所以，当我尝试创建连接时，它指的是存根类并抛出预期。

将@PowerMockIgnore注释添加到类中，以避免它。 @PowerMockIgnore注释告诉powermock不要为属于给定包的类创建。

@RunWith (PowerMockRunner.class)
@PrepareForTest({DbUtil.class})
@PowerMockIgnore({"java.sql.*"})
public class MyUtilTest {
  ... 
}

这对我有用。

JDBC连接无法与PowerMockito

1 个答案: