@Test
  public void testWaitUnitAgentConnected() {
    SingleConsole singleConsole =
        new SingleConsole(11113) {
          @Override
          public boolean isCanceled() {
            return cancelded;
          }
        };
    ProcessReports report = mock(ProcessReports.class);
    WorkerProcessReport workerProcessReport = mock(WorkerProcessReport.class);
    WorkerProcessReport workerProcessReport2 = mock(WorkerProcessReport.class);
    WorkerProcessReport[] workerProcessReports =
        new WorkerProcessReport[] {workerProcessReport, workerProcessReport2};
    when(workerProcessReport.getNumberOfRunningThreads()).thenReturn(new Short((short) 3));
    when(workerProcessReport2.getNumberOfRunningThreads()).thenReturn(new Short((short) 2));

    when(report.getWorkerProcessReports()).thenReturn(workerProcessReports);

    ProcessReports[] processReports = new ProcessReports[] {report};
    singleConsole.update(processReports);
    assertThat(singleConsole.getRunningProcess(), is(2));
    assertThat(singleConsole.getRunningThread(), is(5));

    processReports = new ProcessReports[] {};
    singleConsole.waitUntilAgentConnected(1);
    singleConsole.update(processReports);
    try {
      singleConsole.waitUntilAgentConnected(1);
      fail("Shoule throw Exception");
    } catch (NGrinderRuntimeException e) {
      //
    }

    singleConsole.waitUntilAllAgentDisconnected();

    processReports = new ProcessReports[] {report};
    singleConsole.update(processReports);
    try {
      singleConsole.waitUntilAllAgentDisconnected();
      fail("Shoule throw Exception");
    } catch (NGrinderRuntimeException e) {
      //
    }
  }
Exemple #2
0
  /**
   * Update the count of current processes and threads.
   *
   * @param processReports ProcessReports array.
   */
  private void updateCurrentProcessAndThread(ProcessReports[] processReports) {
    int notFinishedWorkerCount = 0;
    int processCount = 0;
    int threadCount = 0;
    // Per agents
    for (ProcessReports agentReport : processReports) {
      // Per process
      for (WorkerProcessReport processReport : agentReport.getWorkerProcessReports()) {
        // There might be the processes which is not finished but no
        // running thread in it.
        if (processReport.getState() != 3) {
          notFinishedWorkerCount++;
        }
        processCount++;
        threadCount += processReport.getNumberOfRunningThreads();
      }
    }

    synchronized (this) {
      this.runningProcess = processCount;
      this.runningThread = threadCount;
      this.currentNotFinishedProcessCount = notFinishedWorkerCount;
    }
  }