Mockito-mocking數據庫調用

Question

我試圖用Mockito模擬以下方法，它返回用戶的詳細信息。

它正在對數據庫進行實際調用，而不是返回模擬結果。

我的方法 -

public User getUserById(String userId){ 
    if (userId == null) { 
     throw new IllegalArgumentException("AssociateId cannot be null"); 
    } 
    User user = new User(); 
    preparedStatement = null; 
    try { 
     connection = dbConnection.openConnection(properties, inputStream); 
     query = queryReader 
       .getQuery(RelationshipManagerConstants.selectUser); 
     preparedStatement = connection.prepareStatement(query); 
     preparedStatement.setString(I_User.associateId, userId); 
     resultSet = preparedStatement.executeQuery(); 
     if (resultSet.next()) { 
      user.setAssociateId(resultSet.getString(I_User.associateId)); 
      user.setAssociatePassword(resultSet 
        .getString(I_User.associatePassword)); 
      user.setAssociateRole(resultSet.getInt(I_User.associateRole)); 
      user.setAssociateIsActive(resultSet 
        .getBoolean(I_User.associateIsActive)); 
      user.setAssociateEmail(resultSet 
        .getString(I_User.associateEmail)); 
     } 
    } catch (ClassNotFoundException e) { 
     LOGGER.warning("Cannot return User Details. ClassNotFoundException occured."); 
    } catch (SQLException e) { 
     LOGGER.warning("Cannot return User Details. SQLException occured."); 
    } catch (IOException e) { 
     LOGGER.warning("Cannot return User Details. IOException occured."); 
    } finally { 
     if (resultSet != null) { 
      try { 
       resultSet.close(); 
      } catch (SQLException e) { 
       LOGGER.warning("Failed to close resultSet."); 
      } 
     } 
     if (preparedStatement != null) { 
      try { 
       preparedStatement.close(); 
      } catch (SQLException e) { 
       LOGGER.warning("Failed to close statement."); 
      } 
     } 
     if (connection != null) { 
      try { 
       connection.close(); 
      } catch (SQLException e) { 
       LOGGER.warning("Failed to close connection."); 
      } 
     } 
    } 
    return user; 
} 

我的測試 -

@Test 
public void testGetUserById() throws Exception { 
    mockConnection = Mockito.mock(Connection.class); 
    Properties mockProperties = Mockito.mock(Properties.class); 
    InputStream mockInputStream = Mockito.mock(InputStream.class); 
    DBConnection mockDbConnection = Mockito.mock(DBConnection.class); 
    PreparedStatement mockPreparedStatement = Mockito.mock(PreparedStatement.class); 
    ResultSet mockResultSet = Mockito.mock(ResultSet.class); 
    QueryReader mockQueryReader = Mockito.mock(QueryReader.class); 


    PowerMockito.whenNew(DBConnection.class).withNoArguments() 
    .thenReturn(mockDbConnection); 
    PowerMockito.whenNew(QueryReader.class).withNoArguments() 
    .thenReturn(mockQueryReader); 


    String query = "select * from User where AssociateID=?;"; 
    Mockito.when(mockDbConnection.openConnection(mockProperties, mockInputStream)).thenReturn(mockConnection); 
    Mockito.when(mockQueryReader.getQuery("sqlScript_selectUser.sql")).thenReturn("query"); 
    Mockito.when(mockConnection.prepareStatement("query")).thenReturn(mockPreparedStatement); 
    Mockito.when(mockPreparedStatement.executeQuery()).thenReturn(mockResultSet); 
    Mockito.when(mockResultSet.next()).thenReturn(true); 

    Mockito.when(mockResultSet.getString(1)).thenReturn("message"); 
    User u=userDAO.getUserById("AB1234"); 
    assertEquals("EX112233", u.getAssociateId()); 
} 

，因爲我回來 「消息」 不過，我與 「EX112233」

斷言我的測試應該失敗

但它打電話給數據庫而不是嘲笑。

在此先感謝。

Answer 1

你準備好了UserDAO以便在PowerMockito using @PrepareForTest中測試嗎？

請記住，您需要準備爲那些構造函數存根調用new DBConnection()和new QueryReader()才能生效的類。