如何在 Spring 3.0.x 中正确测试 @Async 邮件程序？

Question

我有几个邮件程序，我很想在它们上使用 @Async 并获得异步邮件的好处。

我遇到的唯一问题是我不知道如何正确测试它们，并且我想轻松地采用我当前测试它们的方法，只需修改它而不对测试代码进行大量更改，如果这是可能的。

例如，在测试类中，我将定义两个自动装配的 bean。一个是邮件服务，它负责执行所有类似邮件的事务，另一个是 JavaMailSender - 但它是一个模拟。然后，我将模拟放入服务中，这样它实际上不会发送真正的电子邮件；）

@Autowired
Mailer mailer;

MockJavaMailSender mailSender;

@Before
public void setup() {
    mailSender = new MockJavaMailSender();
    mailer.setMailSender(mailSender);
}

这种方法效果非常好，因为我可以问我的模拟问题或从中获取数据以确保我的邮件程序代码有效：

UserAccount userAccount = userAccountDao.find(1);

mailer.sendRetrievePassword(userAccount);

mailSender.assertTimesSent(1);
String text = mailSender.getMimeMessage().buildText();

// Do tests on text.

@Async 的问题是 mailSender 尚未填充，因此测试将失败。

这是使用 @Async 的代码：

@Async
@Override
public void sendRetrievePassword(UserAccount userAccount) {
    mailSender.send(new MimeMessageSupport(velocityEngine)
        .setTitle("Retrieve Password")
        .setTemplate("mail/retrievePassword.vm")
        .setToEmailAddress(userAccount.getEmailAddress())
        .addObject("userAccount", userAccount));
}

有没有一种非常简单的方法来纠正这个问题？

Answer 1

嗯，看来这可能是解决方案。我真的不想返回 mime 消息，因为我的应用程序不需要...但它有效：

@Async
@Override
public Future<MimeMessageSupport> sendRetrievePassword(UserAccount userAccount) {
    MimeMessageSupport mimeMessage = new MimeMessageSupport(velocityEngine)
        .setTitle("Retrieve Password")
        .setTemplate("mail/retrievePassword.vm")
        .setToEmailAddress(userAccount.getEmailAddress())
        .addObject("userAccount", userAccount);

    mailSender.send(mimeMessage);

    return new AsyncResult<MimeMessageSupport>(mimeMessage);
}

下面是使其通过的测试：

@Test
public void sendRetrievePassword() throws ExecutionException, InterruptedException {
    UserAccount userAccount = userAccountDao.find(1);

    Future<MimeMessageSupport> future = mailer.sendRetrievePassword(userAccount);

    String text = future.get().buildText();

    assertTrue(text.contains(userAccount.getEmailAddress()));
    assertTrue(text.contains(userAccount.getPassword()));

    mailSender.assertTimesSent(1);
}