用Mockito模拟单身人士

Question

我需要测试一些遗留代码，该遗留代码在方法调用中使用单例。该测试的目的是确保clas
sunder测试调用单例方法。我在SO上也看到过类似的问题，但是所有答案都需要其他依赖项（不同的测试框架）-不幸的是，我仅限于使用Mockito和JUnit，但这在如此流行的框架下应该是完全可能的。

单例：

public class FormatterService {

    private static FormatterService INSTANCE;

    private FormatterService() {
    }

    public static FormatterService getInstance() {
        if (INSTANCE == null) {
            INSTANCE = new FormatterService();
        }
        return INSTANCE;
    }

    public String formatTachoIcon() {
        return "URL";
    }

}

被测课程：

public class DriverSnapshotHandler {

    public String getImageURL() {
        return FormatterService.getInstance().formatTachoIcon();
    }

}

单元测试：

public class TestDriverSnapshotHandler {

    private FormatterService formatter;

    @Before
    public void setUp() {

        formatter = mock(FormatterService.class);

        when(FormatterService.getInstance()).thenReturn(formatter);

        when(formatter.formatTachoIcon()).thenReturn("MockedURL");

    }

    @Test
    public void testFormatterServiceIsCalled() {

        DriverSnapshotHandler handler = new DriverSnapshotHandler();
        handler.getImageURL();

        verify(formatter, atLeastOnce()).formatTachoIcon();

    }

}

这个想法是配置可怕的单例的预期行为，因为被测类将调用它的getInstance，然后调用formatTachoIcon方法。不幸的是，这失败并显示一条错误消息：

when() requires an argument which has to be 'a method call on a mock'.

Answer 1

您的要求是不可能的，因为您的旧代码依赖于静态方法，getInstance()而Mockito不允许模拟静态方法，因此以下行将不起作用

when(FormatterService.getInstance()).thenReturn(formatter);

有两种方法可以解决此问题：

1. 使用其他模拟工具，例如PowerMock，该工具可以模拟静态方法。

2. 重构代码，以免依赖静态方法。我能想到的最小侵入性方法是通过向其DriverSnapshotHandler注入FormatterService依赖项的构造函数。该构造函数将仅在测试中使用，并且您的生产代码将继续使用实际的单例实例。

   public static class DriverSnapshotHandler {
   
   private final FormatterService formatter;
   
   //used in production code
   public DriverSnapshotHandler() {
       this(FormatterService.getInstance());
   }
   
   //used for tests
   DriverSnapshotHandler(FormatterService formatter) {
       this.formatter = formatter;
   }
   
   public String getImageURL() {
       return formatter.formatTachoIcon();
   }
   

   }

然后，您的测试应如下所示：

FormatterService formatter = mock(FormatterService.class);
when(formatter.formatTachoIcon()).thenReturn("MockedURL");
DriverSnapshotHandler handler = new DriverSnapshotHandler(formatter);
handler.getImageURL();
verify(formatter, atLeastOnce()).formatTachoIcon();