java - 需要访问存储库的测试方法

Question

我尝试学习编写测试。

@Service
public class MailServiceImpl {
  @Autowired
  private SettingRepository settingRepository;

  public String buildTemplate(Template template, Map<String, String> templateParameters) {

    final String templateHeader = settingRepository.findByKey("test1").getValue();
    final String templateFooter = settingRepository.findByKey("test2").getValue();

    String content = replaceParameters(template.getContent(), templateParameters);

    StringBuilder builder = new StringBuilder();
    builder.append(templateHeader);
    builder.append(content);
    builder.append(templateFooter);

    return builder.toString();
  }

  private String replaceParameters(String content, Map<String, String> templateParameters) {
    for (Entry<String, String> parameter : templateParameters.entrySet()) {
      content = content.replace(
          String.format("{{ %s }}", parameter.getKey()),
          parameter.getValue()
      );
    }
    return content;
  }
}

public class MailServiceImplTest {

  @InjectMocks
  MailServiceImpl mailService;

  @Before
  public void setup() {
    initMocks(this);
  }

  @Test
  public void buildTemplate() {

    Template template = new Template();
    template.setContent("lorem ipsum");

    String content = mailService.buildTemplate(template, new HashMap<>());

    // assertion
  }
}

我的问题是在这部分:

final String templateHeader = settingRepository.findByKey("test1").getValue();

我收到异常“NullPointerException”，是否可以在不更改任何代码的情况下测试此方法？

数据库是 MongoDB，所以我需要创建一些带有键“test1”、“test2”的“虚拟”文档吗？

@Repository
public interface SettingRepository extends PagingAndSortingRepository<Setting, String> {

  Setting findByKey(String id);
}

感谢您的帮助

<小时/>

解决方案:

  @Test
  public void buildTemplate() {
Mockito.when(settingRepository.findByKey("test1")).thenReturn(new Setting());
    Mockito.when(settingRepository.findByKey("test2")).thenReturn(new Setting());
}

Answer 1

您需要模拟 settingRepository 的实例。简单地使用 @InjectMocks 不会神奇地注入(inject) MailServiceImpl 的依赖项，因为如果您不指定，Mockito 不知道需要注入(inject)什么。

public class MailServiceImplTest {

  @InjectMocks
  private MailServiceImpl mailService;

  @Mock
  private SettingRepository settingRepository;

我不会解释 @InjectMocks 和 @Mock 之间的区别，因为它超出了您的具体问题的范围，但如果您想了解两者之间的区别二、可以查看this question .