/*
 * Copyright (c) 2018 Ericsson India Global Services Pvt Ltd. and others. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.genius.datastoreutils.testutils;

import static java.util.concurrent.TimeUnit.MILLISECONDS;
import static java.util.concurrent.TimeUnit.SECONDS;
import static org.hamcrest.Matchers.is;

import java.util.function.Supplier;
import javax.inject.Inject;

import org.awaitility.Awaitility;
import org.awaitility.core.ConditionTimeoutException;
import org.opendaylight.infrautils.jobcoordinator.JobCoordinatorMonitor;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public abstract class AbstractTestableJobCoordinatorEventsWaiter {

    private static final Logger LOG = LoggerFactory.getLogger(AbstractTestableJobCoordinatorEventsWaiter.class);

    protected final JobCoordinatorMonitor jobCoordinatorMonitor;

    @Inject
    public AbstractTestableJobCoordinatorEventsWaiter(JobCoordinatorMonitor jobCoordinatorMonitor) {
        this.jobCoordinatorMonitor = jobCoordinatorMonitor;
    }

    protected boolean awaitJobsConsumption(Supplier<Long> countSupplier, long expectedCount)
            throws ConditionTimeoutException {
        try {
            Awaitility.await("TestableJobCoordinatorEventsWaiter")
                    .atMost(120, SECONDS)
                    .pollDelay(0, MILLISECONDS)
                    .conditionEvaluationListener(condition -> LOG.info(
                            "awaitEventsConsumption: Elapsed time {}s, remaining time {}s; current count: {}"
                                    + " expected event count: {}",
                            condition.getElapsedTimeInMS() / 1000, condition.getRemainingTimeInMS() / 1000,
                            condition.getValue(), expectedCount))
                    .until(countSupplier::get, is(expectedCount));
        } catch (ConditionTimeoutException e) {
            LOG.error("Details about stuck JobCoordinator: {}", jobCoordinatorMonitor.toString());
            throw e;
        }
        return true;
    }
}