如何从 java.util 包中模拟 UUID.randomUUID()？

Question

UUID.randomUUID() 有什么问题 - 它不能被嘲笑

可以模拟吗？或者我的来源有误？

看例子：

1) Class 已测试

package com.grayen;

import java.util.UUID;

public class TestedClass {
    public UUID getUuid() {
        return UUID.randomUUID();
    }
    public UUID getUuidFromWrapper() {
        return UuidWrapper.randomUUID();
    }
}

一种方法使用 UUID 的包装器，我可以模拟该包装器！

2) 真实 UUID 的包装器（所有修饰符相同）

package com.grayen;

import java.util.UUID;

public final class UuidWrapper {
    public static UUID randomUUID() {
        return UUID.randomUUID();
    }
}

3) 测试（最后注释行抛出异常）

package com.grayen;

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

@PrepareForTest({UUID.class, UuidWrapper.class})
@RunWith(PowerMockRunner.class)
public class TestedClassTest {

    @Test
    public void testMethod() {
        UUID uuid = UUID.randomUUID();

        PowerMockito.mockStatic(UUID.class);
        PowerMockito.mockStatic(UuidWrapper.class);

        PowerMockito.when(UUID.randomUUID()).thenReturn(uuid);
        PowerMockito.when(UuidWrapper.randomUUID()).thenReturn(uuid);

        TestedClass testedClass = new TestedClass();

        assertEquals(uuid, testedClass.getUuidFromWrapper());
        //assertEquals(uuid, testedClass.getUuid());
    }
}

Answer 1

模拟静态方法始终是一种脆弱的方法。如果可能，更喜欢使用非静态 UUID 源，这样可以很容易地模拟它。

例如：

/**
 * A source of new {@link UUID} instances.
 */
public interface UuidSource {
    /**
     * Returns a new {@link UuidSource} that generates UUIDs using {@link UUID#randomUUID}.
     */
    public static UuidSource random() {
        return UUID::randomUUID;
    }

    /**
     * Returns a new {@link UUID} instance.
     *
     * <p>The returned value is guaranteed to be unique.
     */
    UUID newUuid();
}

然后可以注入TestedClass，或者让TestedClass有一个私有成员：

public class TestedClass {
    private UuidSource uuidSource = UuidSource.random();

    public UUID getUUID() {
        return uuidSource.newUuid();
    }
    // etc.
}

然后要对其进行测试，您可以拥有一个仅测试构造函数，以允许注入模拟的 UuidSource，或者您可以直接替换 uuidSource 字段的值（通过扩大它的可见性或通过使用反射或其他东西）。

作为奖励：这将您的实际生产代码与 UUID.randomUUID() 分离。如果稍后您决定需要使用版本 2 UUID（基于日期时间）或其他版本而不是随机 UUID，您也可以在生产代码中轻松更改它。这就是人们所说的使您的代码更易于测试通常会带来更好的整体设计的意思。

Answer 2

所以我使用了您的代码并使其运行。您需要做的就是将 TestedClass.class 添加到您的 @PrepareForTest.

package com.grayen;

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

@PrepareForTest({
    UUID.class, 
    UuidWrapper.class, 
    TestedClass.class
})
@RunWith(PowerMockRunner.class)
public class TestedClassTest {

    @Test
    public void testMethod() {
        UUID uuid = UUID.randomUUID();

        PowerMockito.mockStatic(UUID.class);
        PowerMockito.mockStatic(UuidWrapper.class);

        PowerMockito.when(UUID.randomUUID()).thenReturn(uuid);
        PowerMockito.when(UuidWrapper.randomUUID()).thenReturn(uuid);

        TestedClass testedClass = new TestedClass();

        assertEquals(uuid, testedClass.getUuidFromWrapper());
        assertEquals(uuid, testedClass.getUuid());
    }
}

编辑

我没看到评论，但是k5_指出了。

如何从 java.util 包中模拟 UUID.randomUUID()？

How to mock UUID.randomUUID() from java.util package?

java

junit4

powermockito