Junit测试随机数的方法

时间:2018-02-10 10:09:02

标签: java junit mockito

我试图为一个方法创建一个Junit测试,该方法使用一个简单的随机数来调用另一个简单的方法,该方法根据参数的值返回rock,paper或scissor。

我不知道该怎么做。

public class Play {

    public String player2Choice() {

        // Create a Random object.
        Random random = new Random();

        // Ramdon number between 1-3
        int numberChoice = random.nextInt(3) + 1;

        // Return the player random choice
        return getItem(numberChoice);
    }

    public String getItem(int numberChoice) {

        String item;

        switch (numberChoice) {
        case 1:
            item = "rock";
            break;
        case 2:
            item = "scissors";
            break;
        case 3:
            item = "paper";
            break;
        default:
            item = null;
        }

        return item;
    }
}

现在我有了这个测试:

Play play;

@Before
public void setup() {

    play = new Play();
}

@Test
public void testPlayer2Choice() {

    int param = 1;
    Play play2 = Mockito.mock(Play.class);


    play2.getItem(param);
    Mockito.verify(play2).getItem(param);

    String result = play.player2Choice();

    assertEquals("rock", result);


}

1 个答案:

答案 0 :(得分:2)

PlayRandom紧密耦合,它实际上无法控制。这使得测试特定数字选择变得困难。

如果目标是测试选择1player2Choice返回"rock",则需要更多控制所谓的随机性。

public interface RandomInteger {
    int nextInt(int max);
}

接口的实现将封装实际的Random类并代理所需的行为/功能

public class RandomIntegerImplementation : RandomInteger {
    private Random random ;

    public RandomIntegerImplementation() {
        // Create a Random object.
        random  = new Random();
    }

    public int nextInt(int max) {
        return random.nextInt(max);
    }
}

然后需要重构Play以明确依赖于现在的抽象随机提供程序。

public class Play {
    private RandomInteger random;

    public Play(RandomInteger random) {
        this.random = random;
    }

    public String player2Choice() {
        // Ramdon number between 1-3
        int numberChoice = random.nextInt(3) + 1;

        // Return the player random choice
        return getItem(numberChoice);
    }

    //...code removed for brevity
}

所以现在如果要满足上述测试用例,可以模拟依赖关系以使其按预期运行。

RandomInteger random;

Play play;

@Before
public void setup() {
    random = Mockito.mock(RandomInteger.class);
    play = new Play(random);
}

@Test
public void testPlayer2Choice_Given_Zero() {
    //Arrange
    String expected = "rock";
    int numberChoice = 0;

    when(random.nextInt(3)).thenReturn(numberChoice);

    //Act
    String actual = play.player2Choice();

    //Assert
    Mockito.verify(play).getItem(numberChoice + 1);
    assertEquals(expected, actual);
}

这应该足够一个开始。对于其他测试用例也可以这样做。

最终getItem也可以重构为自己的服务,但这超出了原始问题的范围。