Mockito-无法模拟结果集

我正在编写一个测试用例,试图模拟一个Resultset。为此我已经准备好了

  val mockedResultSet = mock[ResultSet]
  val mockedRow = mock[Row]

  doReturn(mockedRow).when(mockedResultSet).one()

或

  doReturn(true).when(mockedResultSet).isExhausted

但是,有些方法中,我直接在resultSet上应用map函数,而不是在resultSet上应用.all()函数。例如:-

   val results = executeDBStatement(dBConnection, queryBuilderStmt)
    if (!results.isExhausted) {
      val res = results.map(row => {        
      // iterate over the result and create a list of case classes
      }
      )
    }

在这里,我无法模拟ResultSet的映射函数行为。请让我知道在这种情况下我如何嘲笑结果集。提前谢谢!!!