com.mercateo.test.clock.TestClock Maven / Gradle / Ivy

Go to download
package com.mercateo.test.clock;

import java.time.Clock;
import java.time.Instant;
import java.time.OffsetDateTime;
import java.time.ZoneId;
import java.time.temporal.TemporalAmount;

import lombok.AccessLevel;
import lombok.AllArgsConstructor;
import lombok.NonNull;

@AllArgsConstructor(access = AccessLevel.PRIVATE)
public class TestClock extends Clock {

    private Instant instant;

    private final ZoneId zone;

    @Override
    public ZoneId getZone() {
        return zone;
    }

    @Override
    public Clock withZone(@NonNull ZoneId zone) {
        return new TestClock(instant, zone);
    }

    @Override
    public Instant instant() {
        return instant;
    }

    public void fastForward(@NonNull TemporalAmount temporalAmount) {
        set(instant().plus(temporalAmount));
    }

    public void rewind(@NonNull TemporalAmount temporalAmount) {
        set(instant().minus(temporalAmount));
    }

    public void set(@NonNull Instant instant) {
        this.instant = instant;
    }

    public static TestClock fixed(@NonNull Instant instant, @NonNull ZoneId zone) {
        return new TestClock(instant, zone);
    }

    public static TestClock fixed(@NonNull OffsetDateTime odt) {
        return fixed(odt.toInstant(), odt.getOffset());
    }

}