java如何使用MockitoJUnitRunner模拟不可修改的映射

Question

Class Test {

 private A a;

 private B b; 

}

Class B{

 private final Map<String, Integer> sampleMap = new HashMap();

 public Map<String, Integer> getSampleMap() {

      return Collections.unmodifiableMap(this.sampleMap);
  }

}

如何使用模拟不可修改的getSampleMap模拟测试对象。我需要创建和设置键/值对

Answer 1

您可以将b协作者（类型为B）注入Test类。 这样做你可以模仿b.getSampleMap()

莫基托：

import org.junit.Rule;
import org.mockito.Mock;
import org.mockito.junit.MockitoJUnit;
import org.mockito.junit.MockitoRule;

import java.util.Collections;
import java.util.HashMap;
import java.util.Map;

import static org.hamcrest.Matchers.is;
import static org.junit.Assert.assertThat;
import static org.mockito.Mockito.when;

public class StackoverflowTest {

  public static class Test {

    private A a;
    private B b;

    public Test(B b) {
      this.b = b;
    }
  }

  public static class A {}

  public static class B {

    private final Map<String, Integer> sampleMap = new HashMap<>();

    public Map<String, Integer> getSampleMap() {
      return Collections.unmodifiableMap(this.sampleMap);
    }
  }

  @Rule
  public MockitoRule mockitoRule = MockitoJUnit.rule();

  @Mock
  private B b;

  @org.junit.Test
  public void testWithMock() {

    Map<String, Integer> expectedMap = new HashMap<>();
    expectedMap.put("a", 1);
    expectedMap.put("b", 2);
    expectedMap.put("c", 3);

    when(b.getSampleMap()).thenReturn(expectedMap);

    Test underTest = new Test(b);

    assertThat(underTest.b.getSampleMap(), is(expectedMap));
  }
}

请注意，断言（assertThat(underTest.b.getSampleMap(), is(expectedMap))）在实际案例中可能没有用处。我把它放在那里只是为了验证我的解决方案