diff options
Diffstat (limited to 'jdisc_core/src/main/java/com/yahoo/jdisc/test/TestTimer.java')
-rw-r--r-- | jdisc_core/src/main/java/com/yahoo/jdisc/test/TestTimer.java | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/jdisc_core/src/main/java/com/yahoo/jdisc/test/TestTimer.java b/jdisc_core/src/main/java/com/yahoo/jdisc/test/TestTimer.java new file mode 100644 index 00000000000..4d274a33a1f --- /dev/null +++ b/jdisc_core/src/main/java/com/yahoo/jdisc/test/TestTimer.java @@ -0,0 +1,31 @@ +// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root. +package com.yahoo.jdisc.test; + +import com.yahoo.jdisc.Timer; + +import java.time.Duration; + +/** + * A {@link Timer} to be used in tests when the advancement of time needs to be controlled. + * + * @author hakonhall + */ +public class TestTimer implements Timer { + private Duration durationSinceEpoch = Duration.ZERO; + + public void setMillis(long millisSinceEpoch) { + durationSinceEpoch = Duration.ofMillis(millisSinceEpoch); + } + + public void advanceMillis(long millis) { advance(Duration.ofMillis(millis)); } + public void advanceSeconds(long seconds) { advance(Duration.ofSeconds(seconds)); } + public void advanceMinutes(long minutes) { advance(Duration.ofMinutes(minutes)); } + public void advance(Duration duration) { + durationSinceEpoch = durationSinceEpoch.plus(duration); + } + + @Override + public long currentTimeMillis() { + return durationSinceEpoch.toMillis(); + } +} |