java - Junit 测试用例

Question

public static String generateSaltString() {
    SecureRandom random = new SecureRandom();
    byte[] salt = random.generateSeed(12);

    return byteToBase64(salt);
}

想知道如何使用 Junit 为该方法编写测试用例。有什么建议吗？

Answer 1

首先，new SecureRandom() 可能会非常慢，因此您可能需要缓存它。看this回答和this还有一个。

我会重构你的代码如下

public class DoRandomStuff {

private RandomUtil randomUtil;

public DoRandomStuff(RandomUtil randomUtil) {
    this.randomUtil = randomUtil;
}

public String generateSaltString() {
    byte[] salt = randomUtil.generateSeed(12);
    return byteToBase64(salt);
}

public String byteToBase64(byte[] salt) {
    // Do salt generation logic here
    return null;
}

generateSeed 看起来像一个实用程序方法调用，因此它可以进入类 RandomUtil，如下所示

class RandomUtil {
    private SecureRandom random = new SecureRandom();
    public byte[] generateSeed(int seed) {
        return random.generateSeed(seed);
    }
}

现在您的 DoRandomStuff 测试类将很容易编写。使用 Mockito 等测试框架注入(inject)模拟的 randomUtil。使 mockedRandomUtil.generateSeed(int) 方法返回一个固定数字。您的测试实际上是为了检查 byteToBase64() 方法是否正在执行它应该执行的操作。现在你的测试用例有了一个确定的数字。您可以在测试类中将 SecureRandom 生成的各种数字作为单独的测试用例来检查 byteToBase64() 结果。另外，您的随机数生成代码现在已与您的 base64 代码分离。

希望对您有所帮助。