mdsalutil/mdsalutil-testutils/src/main/java/org/opendaylight/genius/datastoreutils/testutils/AbstractTestableJobCoordinatorEventsWaiter.java

   1 /*
   2  * Copyright (c) 2018 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.genius.datastoreutils.testutils;
   9
  10 import static java.util.concurrent.TimeUnit.MILLISECONDS;
  11 import static java.util.concurrent.TimeUnit.SECONDS;
  12 import static org.hamcrest.Matchers.is;
  13
  14 import java.util.function.Supplier;
  15 import javax.inject.Inject;
  16 import org.awaitility.Awaitility;
  17 import org.awaitility.core.ConditionTimeoutException;
  18 import org.opendaylight.infrautils.jobcoordinator.JobCoordinatorMonitor;
  19 import org.slf4j.Logger;
  20 import org.slf4j.LoggerFactory;
  21
  22 public abstract class AbstractTestableJobCoordinatorEventsWaiter {
  23
  24     private static final Logger LOG = LoggerFactory.getLogger(AbstractTestableJobCoordinatorEventsWaiter.class);
  25
  26     protected final JobCoordinatorMonitor jobCoordinatorMonitor;
  27
  28     @Inject
  29     public AbstractTestableJobCoordinatorEventsWaiter(JobCoordinatorMonitor jobCoordinatorMonitor) {
  30         this.jobCoordinatorMonitor = jobCoordinatorMonitor;
  31     }
  32
  33     protected boolean awaitJobsConsumption(Supplier<Long> countSupplier, long expectedCount)
  34             throws ConditionTimeoutException {
  35         try {
  36             Awaitility.await("TestableJobCoordinatorEventsWaiter")
  37                     .atMost(120, SECONDS)
  38                     .pollDelay(0, MILLISECONDS)
  39                     .conditionEvaluationListener(condition -> LOG.info(
  40                             "awaitEventsConsumption: Elapsed time {}s, remaining time {}s; current count: {}"
  41                                     + " expected event count: {}",
  42                             condition.getElapsedTimeInMS() / 1000, condition.getRemainingTimeInMS() / 1000,
  43                             condition.getValue(), expectedCount))
  44                     .until(countSupplier::get, is(expectedCount));
  45         } catch (ConditionTimeoutException e) {
  46             LOG.error("Details about stuck JobCoordinator: {}", jobCoordinatorMonitor.toString());
  47             throw e;
  48         }
  49         return true;
  50     }
  51 }