如何模拟 ENUM 类中的方法？

Question

我正在为下面的 ENum 类编写 JUNIT 测试用例。我下面的课程只会给我运行我的代码的当前机器的主机名。在我编写 JUNIT 测试时，如何模拟下面的类，以便我可以随时更改 getHostName() 方法，以便每当我调用 getDatacenter() 时，它可以通过模拟返回我传递的任何主机名.我不想让它参数化。

我只是想在模拟主机名的同时测试某些情况。

public enum DatacenterEnum {
    DEV, DC1, DC2, DC3;


    public static String forCode(int code) {
    return (code >= 0 && code < values().length) ? values()[code].name() : null;
    }
    private static final String getHostName() {
        try {
            return InetAddress.getLocalHost().getCanonicalHostName().toLowerCase();
        } catch (UnknownHostException e) {
            s_logger.logError("error = ", e);
        }

        return null;
    }

    public static String getDatacenter() {
        return getHostName();
    }
}

Answer 1

有可能，但不推荐，最好重构代码。

使用 Mockito/PowerMock 的工作示例

@RunWith(PowerMockRunner.class)
@PrepareForTest(DatacenterEnum.class)
public class DatacenterEnumTest {

    @Mock
    InetAddress inetAddress;

    @Test
    public void shouldReturnDatacenter() throws UnknownHostException {
        //given
        mockStatic(InetAddress.class);
        given(inetAddress.getCanonicalHostName()).willReturn("foo");
        given(InetAddress.getLocalHost()).willReturn(inetAddress);

        //when
        String datacenter = DatacenterEnum.getDatacenter();

        //then
        assertThat(datacenter).isEqualTo("foo");
    }
}

依赖关系

• org.powermock:powermock-module-junit4:1.5.2
• org.powermock:powermock-api-mockito:1.5.2
• org.assertj:assertj-core:1.5.0
• junit:junit:4.11

Answer 2

使用 JMockit 很容易：

@Test
public void mockInetAddress(@Cascading final InetAddress inetAddress)
{
    new NonStrictExpectations() {{
        inetAddress.getCanonicalHostName(); result = "foo";
    }};

    String datacenter = DatacenterEnum.getDatacenter();

    assertEquals("foo", datacenter);
}

当然，您也可以在枚举中模拟 getHostName() 方法，但最好避免模拟 private 方法。

Answer 3

您可以创建一个数据中心接口并让枚举实现该接口。这将使模拟更容易。

首先，我不会将配置信息放在枚举中。如果您必须添加其他数据中心（或数据中心的配置更改），则必须重新编译代码。考虑将配置放入普通类读取中，例如 java 属性文件或 XML 文件。 （这个函数可能已经在你的框架中实现了。）

如果这不可能，您可以使用“darkes reflation”魔法将 Enum 中的字段更改为所需的值。

Answer 4

这是您可以使用 Mockito/Powermock 实现的一种方式。您需要 Powermock，因为 Mockito 无法模拟静态方法：

import org.junit.Test;
import org.junit.runner.RunWith;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import static org.junit.Assert.assertEquals;
import static org.powermock.api.mockito.PowerMockito.mockStatic;
import static org.powermock.api.mockito.PowerMockito.when;

@RunWith(PowerMockRunner.class)
@PrepareForTest({DatacenterEnum.class})
public class DatacenterEnumTest {

    @Test
    public void testGetDatacenter() {
        mockStatic(DatacenterEnum.class);
        when(DatacenterEnum.getDatacenter()).thenReturn("YourHostname");

        String datacenter = DatacenterEnum.getDatacenter();

        assertEquals("YourHostname", datacenter);
    }
}

Maven 依赖项

<dependencies>
    <dependency>
        <groupId>junit</groupId>
        <artifactId>junit</artifactId>
        <version>4.11</version>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-api-mockito</artifactId>
        <version>1.5.2</version>
    </dependency>
    <dependency>
        <groupId>org.powermock</groupId>
        <artifactId>powermock-module-junit4</artifactId>
        <version>1.5.2</version>
    </dependency>
</dependencies>

Answer 5

我可能是老派，但我真的会重构被测代码，而不是使用类加载器黑客。比如：

public enum DatacenterEnum {
    DEV, DC1, DC2, DC3;


    static String hostName = InetAddress.getLocalHost().getCanonicalHostName().toLowerCase();

    public static String getHostName() {
        return hostName;
    }
}

在您的测试代码中，在运行测试之前：

DataCenterEnum.hostName = "foo";