java - 如何使用jmockt在Enum类中模拟呢？

我正在尝试为下面的旧式遗留枚举类编写单元测试。我尝试进行单元测试的方法是-toLocalPookString。

每当我运行单元测试代码时，它总是在if statement方法的toLocalPookString内部，因为this总是解析为CORP。

public enum DatacenterEnum {
    CORP, PHX, SLC, LVS;

    private static final DatacenterEnum ourlocation = compareLocation();

    private static DatacenterEnum compareLocation() {
        // some code here
    }

    private String toLocalPookString() {
        if (this == CORP || !(TestUtils.getEnvironmentName().equalsIgnoreCase("production"))) {
            return "/pp/dc/phx";
        }

        return "/pp/dc/" + name().toLowerCase();
    }


    public static final String BETA_POOK_STRING = ourlocation.toLocalPookString();

}

除了this之外，还有什么方法可以模拟PHX到SLC或LVS或CORP，以便在toLocalPookString中不应该进入if statement？我在这里使用jmockit。

new MockUp<TestUtils>() {
    @Mock
    public String getEnvironmentName() {
    return "production";
    }
};

String ss = DatacenterEnum.BETA_POOK_STRING;
System.out.println(ss);

这很简单，但是以某种方式我无法理解该怎么做？有什么想法吗？

最佳答案

好吧，您可以轻松地模拟枚举，如下所示：

new MockUp<DatacenterEnum>() {
    @Mock DatacenterEnum compareLocation() { return DatacenterEnum.LVS; }
};

但是，由于JVM对于给定的类/枚举只能执行一次静态初始化，因此该测试仅在尚未加载和初始化枚举的情况下才有效。
因此，下面将进行更可靠的测试。

@Test
public void whenDatacenterIsNotCORPThenLocalPookStringShouldIncludeEnumName() {
    new MockUp<TestUtils>() {
        @Mock String getEnvironmentName() { return "production"; }
    };

    DatacenterEnum notCORP = DatacenterEnum.LVS;
    String ss = Deencapsulation.invoke(notCORP, "toLocalPookString");

    assertTrue(ss.toUpperCase().endsWith(notCORP.name()));
}

这可行，但是请注意，通常为private方法编写单独的测试被认为是错误的形式。理想情况下，测试应调用将间接行使私有方法的公共方法。但是，在这种特殊情况下，没有这样的公共方法，所以我想这是可以接受的。