X-Git-Url: https://git.pterodactylus.net/?p=xudocci.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fxdcc%2Fcore%2FConnectionBackoffTest.java;h=91b1c8264941295408f0840a11a8a910f19570f6;hp=adaae8bdd94edd8efec66136c694b467a011969c;hb=44825f1cf5ddb021cbf9b8416e8649e768905e77;hpb=2769f52b856c2ebb1ec06836ca81591b7e99a4ba

diff --git a/src/test/java/net/pterodactylus/xdcc/core/ConnectionBackoffTest.java b/src/test/java/net/pterodactylus/xdcc/core/ConnectionBackoffTest.java
index adaae8b..91b1c82 100644
--- a/src/test/java/net/pterodactylus/xdcc/core/ConnectionBackoffTest.java
+++ b/src/test/java/net/pterodactylus/xdcc/core/ConnectionBackoffTest.java
@@ -17,7 +17,8 @@ import org.mockito.Mockito;
 public class ConnectionBackoffTest {
 
 	private final Network network = Mockito.mock(Network.class);
-	private final Clock fixedClock = Clock.fixed(Instant.now(), ZoneId.systemDefault());
+	private final Instant now = Instant.now();
+	private final Clock fixedClock = Clock.fixed(now, ZoneId.systemDefault());
 	private final ConnectionBackoff connectionBackoff = new ConnectionBackoff(fixedClock);
 
 	@Test
@@ -28,12 +29,14 @@ public class ConnectionBackoffTest {
 	@Test
 	public void firstConnectionIsImmediatelyPossible() {
 		MatcherAssert.assertThat(connectionBackoff.getBackoff(network), Matchers.is(0L));
+		MatcherAssert.assertThat(connectionBackoff.getConnectionTime(network), Matchers.is(now.toEpochMilli()));
 	}
 
 	@Test
 	public void afterTheFirstFailedConnectionBackoffIsOneMinute() {
 		connectionBackoff.connectionFailed(network);
 		MatcherAssert.assertThat(connectionBackoff.getBackoff(network), Matchers.is(60000L));
+		MatcherAssert.assertThat(connectionBackoff.getConnectionTime(network), Matchers.is(now.toEpochMilli() + 60000));
 	}
 
 	@Test
@@ -41,6 +44,7 @@ public class ConnectionBackoffTest {
 		connectionBackoff.connectionFailed(network);
 		connectionBackoff.connectionFailed(network);
 		MatcherAssert.assertThat(connectionBackoff.getBackoff(network), Matchers.is(72000L));
+		MatcherAssert.assertThat(connectionBackoff.getConnectionTime(network), Matchers.is(now.toEpochMilli() + 72000));
 	}
 
 	@Test
@@ -48,6 +52,7 @@ public class ConnectionBackoffTest {
 		connectionBackoff.connectionFailed(network);
 		connectionBackoff.connectionSuccessful(network);
 		MatcherAssert.assertThat(connectionBackoff.getBackoff(network), Matchers.is(0L));
+		MatcherAssert.assertThat(connectionBackoff.getConnectionTime(network), Matchers.is(now.toEpochMilli()));
 	}
 
 	@Test
@@ -56,6 +61,7 @@ public class ConnectionBackoffTest {
 			connectionBackoff.connectionFailed(network);
 		}
 		MatcherAssert.assertThat(connectionBackoff.getBackoff(network), Matchers.is(3600000L));
+		MatcherAssert.assertThat(connectionBackoff.getConnectionTime(network), Matchers.is(now.toEpochMilli() + 3600000));
 	}
 
 }