Mockito 测试迭代器在 hasNext 上总是返回 false

Question

我有这个方法：

    public void executeShadowingCF(String outputPath) throws Exception {
            try{

                Iterator<Fatt0Gdpr> notProcessedGDPRIterator = gdprDAO.getGdprRowCF(em1Fatt0);

                while(notProcessedGDPRIterator.hasNext()){
                    Fatt0Gdpr element = notProcessedGDPRIterator.next();

                    String codiceFiscale = element.getCodFiscale();

                    if(codiceFiscale != null && !codiceFiscale.equalsIgnoreCase("")){
                        //some operation

                    }
                    else{
                        gdprDAO.updateProcessedRow(emFatt0Update, element.getIdTecnico(), "99");
                    }

                }               

            }
            catch(Exception e ){
                log.error(e);
            }
            finally{
                emFatt0.close();
            }
        }

我正在尝试使用 mockito 对其进行测试，但是当我执行测试时，notProcessedGDPRIterator.hasNext() 始终返回 false，因此我无法进入 while 循环。 这是我的测试方法：

       @Test
        public void testExecuteShadowingCF() {    
            try {
                EntityManager em1Fatt0 = Mockito.mock(EntityManager.class);
                EntityManager emFatt0 = Mockito.mock(EntityManager.class);
                Fatt0GdprService service = new Fatt0GdprService();          
                Fatt0GdprDAO fatt0GdprDAO = new Fatt0GdprDAO(); 
                Query queryFind = Mockito.mock(Query.class);
                Iterator<Fatt0Gdpr> mockIterator = Mockito.mock(Iterator.class);
                Mockito.when(
                        emFatt0.createNativeQuery(Mockito.anyString(),
                                eq(Fatt0Gdpr.class))).thenReturn(queryFind);    
                Mockito.when(fatt0GdprDAO.getGdprRowNDG(emFatt0)).thenReturn(mockIterator); 

                Mockito.when(mockIterator.hasNext()).thenReturn(true).thenReturn(false);
                Mockito.when(mockIterator.next()).thenReturn(new Fatt0Gdpr());

                service.setEmFatt0Updater(emFatt0);
                EntityTransaction trxMock = Mockito.mock(EntityTransaction.class);
                Mockito.when(service.getEmFatt0Updater().getTransaction())
                        .thenReturn(trxMock);

                String outputPath = "U:/RTC_U0E7069/fatt0_SRC/SRC/fatt0-gdpr-batch/target/diraliases/BATCHROOT/ITT/out";

                service.executeShadowingCF(outputPath);

            } catch (Exception e) {
                e.printStackTrace();
            }
        }

有人可以帮我解决这个问题吗？ 谢谢

Answer 1

不要模拟迭代器。

您应该只在行为太难以其他方式创建时才使用模拟。来自Wikipedia：

如果一个对象具有以下任何特征，则在其位置使用模拟对象可能会很有用：

• 对象提供不确定的结果（例如当前时间或当前温度）；
• 它具有难以创建或重现的状态（例如网络错误）；
• 速度很慢（例如，一个完整的数据库，必须在测试前初始化）；
• 它尚不存在或可能会改变行为；
• 它必须包含专门用于测试目的（而不是用于其实际任务）的信息和方法。

迭代器没有任何这些属性（必须）：只需让返回迭代器的方法使用Arrays.asList().iterator()，或类似的。

List<Fatt0Gdpr> fakeData = Arrays.asList(new Fatt0Gdpr());
Mockito.when(fatt0GdprDAO.getGdprRowNDG(emFatt0)).thenReturn(fakeData.iterator()); 

Answer 2

我猜你应该模拟 gdprDAO 并定义 getGdprRowCF 的行为并返回 mockIterator

Mockito.when(gdprDAO.getGdprRowCF(em1Fatt0)).thenReturn(mockIterator); 

---

定义List 并将其作为迭代器而不是mockIterator 发送的最佳实践

List<Fatt0Gdpr> list = new ArrayList<>();
list.add(new Fatt0Gdpr());
Mockito.when(gdprDAO.getGdprRowCF(em1Fatt0)).thenReturn(list.iterator());