java - 如何使用 Mockito 模拟 For 循环

Question

我是 mockito 的新手。我的问题是如何使用 Mockito 模拟 for 循环？

例如:这是主类:

import java.util.HashSet;
import java.util.Set;

    public class stringConcatination {

        public static void main(String[] args) {
            Set<String> stringSet = new HashSet();
            stringSet.add("Robert");
            stringSet.add("Jim");
            for (String s:stringSet) {
                s = "hi " + s;
            }
        }

}

这是测试类:

import java.util.HashSet;
import java.util.Set;

import org.junit.Test;
import static org.mockito.Mockito.mock;

public class stringConcatinationTest {

    @Test
    public void testMain() {
        Set mockSet = mock(HashSet.class);
        // --  How to mock For Loop --
    }

}

我看到了这个related question .但我无法理解，如何模拟 for 循环。

Answer 1

由于 for 循环只是 iterator() 循环的语法糖，您可以只 stub 该方法并返回模拟的 Iterator 实例