Skip to content

Commit

Permalink
TestEnvironment: Fix timeout arithmetic
Browse files Browse the repository at this point in the history
Motivation:
The timeout calculation decrementing `totalTimeoutRemainingNs` is incorrect
which results in premature timeouts and test failures.

Signed-off-by: Scott Mitchell <scott_mitchell@apple.com>
  • Loading branch information
Scottmitch committed May 7, 2023
1 parent 944163a commit 11b1a04
Show file tree
Hide file tree
Showing 2 changed files with 176 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,175 @@
package org.reactivestreams.example.unicast;

import org.reactivestreams.Publisher;
import org.reactivestreams.Subscriber;
import org.reactivestreams.Subscription;
import org.reactivestreams.tck.PublisherVerification;
import org.reactivestreams.tck.TestEnvironment;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.Executor;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.LinkedBlockingQueue;

@Test // Must be here for TestNG to find and run this, do not remove
public class AsyncRangePublisherTest extends PublisherVerification<Integer> {
private static final int TERMINAL_DELAY_MS = 20;
private static final int DEFAULT_TIMEOUT_MS = 5000;
private static final int DEFAULT_POLL_INTERVAL_MS = TERMINAL_DELAY_MS / 2;
private ExecutorService e;
@BeforeClass
void before() { e = Executors.newCachedThreadPool(); }
@AfterClass
void after() { if (e != null) e.shutdown(); }

public AsyncRangePublisherTest() {
super(new TestEnvironment(DEFAULT_TIMEOUT_MS, 50, DEFAULT_POLL_INTERVAL_MS));
}

@Override
public Publisher<Integer> createPublisher(long elements) {
return new AsyncPublisher<Integer>(new RangePublisher(1, (int)elements), e);
}

@Override
public Publisher<Integer> createFailedPublisher() {
return null;
}

private static final class AsyncPublisher<T> implements Publisher<T> {
private final Publisher<T> original;
private final Executor executor;

private AsyncPublisher(Publisher<T> original, Executor executor) {
this.original = requireNonNull(original);
this.executor = requireNonNull(executor);
}

@Override
public void subscribe(Subscriber<? super T> s) {
original.subscribe(new AsyncSubscriber<T>(requireNonNull(s), executor));
}

private static final class AsyncSubscriber<T> implements Subscriber<T> {
private final BlockingQueue<Object> signalQueue = new LinkedBlockingQueue<Object>();

private AsyncSubscriber(final Subscriber<? super T> original, final Executor executor) {
try {
executor.execute(new Runnable() {
@Override
public void run() {
for (; ; ) {
try {
final Object signal = signalQueue.take();
if (signal instanceof Cancelled) {
return;
} else if (signal instanceof TerminalSignal) {
Thread.sleep(TERMINAL_DELAY_MS);
TerminalSignal terminalSignal = (TerminalSignal) signal;
if (terminalSignal.cause == null) {
original.onComplete();
} else {
original.onError(terminalSignal.cause);
}
return;
} else if (signal instanceof OnSubscribeSignal) {
original.onSubscribe(((OnSubscribeSignal) signal).subscription);
} else {
@SuppressWarnings("unchecked")
final T onNextSignal = ((OnNextSignal<T>) signal).onNext;
original.onNext(onNextSignal);
}
} catch (InterruptedException ex) {
throw new RuntimeException(ex);
}
}
}
});
} catch (Throwable cause) {
original.onSubscribe(new Subscription() {
@Override
public void request(long n) {
}

@Override
public void cancel() {
}
});
original.onError(new IllegalStateException("Executor rejected", cause));
}
}

@Override
public void onSubscribe(final Subscription s) {
signalQueue.add(new OnSubscribeSignal(new Subscription() {
@Override
public void request(long n) {
s.request(n);
}

@Override
public void cancel() {
try {
s.cancel();
} finally {
signalQueue.add(new Cancelled());
}
}
}));
}

@Override
public void onNext(T t) {
signalQueue.add(new OnNextSignal<T>(t));
}

@Override
public void onError(Throwable t) {
signalQueue.add(new TerminalSignal(requireNonNull(t)));
}

@Override
public void onComplete() {
signalQueue.add(new TerminalSignal(null));
}
}

private static final class TerminalSignal {
private final Throwable cause;

private TerminalSignal(Throwable cause) {
this.cause = cause;
}
}

private static final class OnSubscribeSignal {
private final Subscription subscription;

private OnSubscribeSignal(Subscription subscription) {
this.subscription = subscription;
}
}

private static final class OnNextSignal<T> {
private final T onNext;

private OnNextSignal(T onNext) {
this.onNext = onNext;
}
}

private static final class Cancelled {
}

private static <T> T requireNonNull(T o) {
if (o == null) {
throw new NullPointerException();
}
return o;
}
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -1112,7 +1112,7 @@ final <E extends Throwable> E expectError(Class<E> clazz, final long totalTimeou

if (env.asyncErrors.isEmpty()) {
timeStampBNs = System.nanoTime();
totalTimeoutRemainingNs =- timeStampBNs - timeStampANs;
totalTimeoutRemainingNs -= (timeStampBNs - timeStampANs);
timeStampANs = timeStampBNs;

if (totalTimeoutRemainingNs <= 0) {
Expand Down

0 comments on commit 11b1a04

Please sign in to comment.