如何对需要时间执行操作的线程进行单元测试

时间:2015-10-29 23:36:37

标签: java multithreading unit-testing junit mockito

我有Thread在程序运行时运行并轮询队列并检查它是否有对象,如果是,则调用对象上的方法

以下是代码:

while(isRunning){
        synchronized (loginQueue) {
            if(loginQueue.peek() != null) {
                Object[] loginObjectWithConnection = loginQueue.poll();
                tryLogin(loginObjectWithConnection);
            }
        }
        try {
            Thread.sleep(10);
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
}

这是tryLogin方法

private void tryLogin(Object[] loginObjectWithConnection) {
        LoginPacket packet = (LoginPacket)loginObjectWithConnection[0];
        Connection connection = (Connection)loginObjectWithConnection[1];

        try {
            if(playerDataService.arevalidCredentials(packet.getUserName(), packet.getPassword())) {

                if(!playerDataService.isPlayerBanned(packet.getUserName())){ //Player exists in the system

                    communicationService.sendTCP(connection, packetFactory.makeLoginFailurePacket(StringConstants.PLAYER_BANNED));

                } else{ //Player is not banned

                }
            } else { // Player does not exist
                communicationService.sendTCP(connection, packetFactory.makeLoginFailurePacket(StringConstants.INVALID_USER));
            }
        } catch (SQLException e) {
            communicationService.sendTCP(connection, packetFactory.makeLoginFailurePacket(StringConstants.SERVER_ERROR));
            e.printStackTrace();
        }
}

现在我的问题是我想测试这些服务方法的调用,但是当我运行单元测试时,它们将无法工作,因为它需要时间才能达到tryLogin,直到那时JUnit失败。我尝试使用Thread.sleep(),但我知道这不是正确的方法,因为它有时失败并且有时会通过。

以下是我在单元测试中的内容

@Test
public void theExpectedMessageShouldBeSentIfUserIsBanned() throws InterruptedException, SQLException {
    //Arrange
    when(moqLoginQueue.peek()).thenReturn(object);
    when(moqLoginQueue.poll()).thenReturn(object);
    LoginFailurePacket packet = new LoginFailurePacket(StringConstants.PLAYER_BANNED);
    when(moqPacketFactory.makeLoginFailurePacket(StringConstants.PLAYER_BANNED)).thenReturn(packet);
    when(moqPlayerDataService.arevalidCredentials(anyString(), anyString())).thenReturn(true);
    when(moqPlayerDataService.isPlayerBanned(anyString())).thenReturn(true);

    //Act
    loginManager.start();
    Thread.sleep(10); //Dirty hack -.-

    //Assert
    verify(moqCommunicationService).sendTCP(any(Connection.class), eq(packet));
}

1 个答案:

答案 0 :(得分:1)

系统在当前形式下是不可测试的:在良好的测试质量中有:

  • 很容易让其他程序员去骗局
  • 其他程序员很难打破
  • 快跑

您要测试的逻辑是LoginManager.tryLogin,它在您的代码段中是私有的。如果你想公开记录它(测试是一种文档:它们说明系统应该如何表现),它必须是公开的。

我建议将所有逻辑移动到新类中的方法:Authentication.attempt()(我建议使用不可变对象和不带任何参数的方法 - 有人说OO设计中的最佳参数数量是零)。

既然测试是可行的,我还认为你应该摆脱LoginManager.start()中的所有代码:只需使用ExecutorService并提交身份验证尝试 - 这样你就可以拥有更快的程序和需要测试的代码较少,因为硬件(和棘手的)部分由Java管理。