com.ocadotechnology.scenario.AfterAtLeastStep Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of OcavaScenarioTest Show documentation
There is a newer version: 16.6.21
/*
 * Copyright © 2017-2023 Ocado (Ocava)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.ocadotechnology.scenario;

import java.util.concurrent.atomic.AtomicBoolean;
import java.util.function.Supplier;

import javax.annotation.CheckForNull;
import javax.annotation.ParametersAreNonnullByDefault;

import org.junit.jupiter.api.Assertions;

import com.google.common.base.Preconditions;
import com.google.common.util.concurrent.AtomicDouble;
import com.ocadotechnology.event.EventUtil;
import com.ocadotechnology.event.scheduling.EventScheduler;

/**
 * A variant of CheckStep which enforces that the step is completed after at least a specified time has passed.
 */
@ParametersAreNonnullByDefault
class AfterAtLeastStep extends CheckStep {
    private final Supplier schedulerSupplier;
    private final StepFuture duration;

    private final AtomicBoolean startedTimer = new AtomicBoolean(false);
    private final AtomicDouble earliestPermittedTime = new AtomicDouble(Double.POSITIVE_INFINITY);

    AfterAtLeastStep(CheckStep checkStep, Supplier schedulerSupplier, StepFuture duration) {
        super(checkStep.type, checkStep.notificationCache, checkStep.predicate);
        this.schedulerSupplier = schedulerSupplier;
        this.duration = duration;
    }

    @Override
    public void execute(@CheckForNull Object notification) {
        Preconditions.checkState(startedTimer.get(), "Called execute on AfterAtLeastStep before setActive");
        super.execute(notification);

        if (isFinished()) {
            Assertions.assertTrue(schedulerSupplier.get().getTimeProvider().getTime() >= earliestPermittedTime.get(),
                    String.format("After At Least step failed - finished before %s", EventUtil.eventTimeToString(earliestPermittedTime.get())));
        }
    }

    @Override
    public void setActive() {
        Preconditions.checkState(!startedTimer.get(), "Started AfterAtLeastStep twice");
        earliestPermittedTime.set(schedulerSupplier.get().getTimeProvider().getTime() + duration.get());
        startedTimer.set(true);
    }
}