ViewVC Help

View File | Revision Log | Show Annotations | Download File | Root Listing

root/jsr166/jsr166/src/test/tck/ScheduledExecutorTest.java

Comparing jsr166/src/test/tck/ScheduledExecutorTest.java (file contents):
Revision 1.86 by jsr166, Sat Mar 25 21:41:10 2017 UTC vs.
Revision 1.92 by jsr166, Wed Mar 29 17:47:16 2017 UTC

#	Line 24 | Line 24 | import java.util.concurrent.RejectedExec
24		import java.util.concurrent.ScheduledFuture;
25		import java.util.concurrent.ScheduledThreadPoolExecutor;
26		import java.util.concurrent.ThreadFactory;
27	+	import java.util.concurrent.ThreadLocalRandom;
28		import java.util.concurrent.ThreadPoolExecutor;
29		import java.util.concurrent.atomic.AtomicBoolean;
30		import java.util.concurrent.atomic.AtomicInteger;
31		import java.util.concurrent.atomic.AtomicLong;
32	+	import java.util.stream.Stream;
33
34		import junit.framework.Test;
35		import junit.framework.TestSuite;
#	Line 744 | Line 746 | public class ScheduledExecutorTest exten
746		* - setContinueExistingPeriodicTasksAfterShutdownPolicy
747		*/
748		public void testShutdown_cancellation() throws Exception {
749	<	Boolean[] allBooleans = { null, Boolean.FALSE, Boolean.TRUE };
748	<	for (Boolean policy : allBooleans)
749	<	{
750	<	final int poolSize = 2;
749	>	final int poolSize = 4;
750		final ScheduledThreadPoolExecutor p
751		= new ScheduledThreadPoolExecutor(poolSize);
752	<	final boolean effectiveDelayedPolicy = (policy != Boolean.FALSE);
753	<	final boolean effectivePeriodicPolicy = (policy == Boolean.TRUE);
754	<	final boolean effectiveRemovePolicy = (policy == Boolean.TRUE);
755	<	if (policy != null) {
756	<	p.setExecuteExistingDelayedTasksAfterShutdownPolicy(policy);
757	<	p.setContinueExistingPeriodicTasksAfterShutdownPolicy(policy);
758	<	p.setRemoveOnCancelPolicy(policy);
759	<	}
752	>	final BlockingQueue<Runnable> q = p.getQueue();
753	>	final ThreadLocalRandom rnd = ThreadLocalRandom.current();
754	>	final long delay = rnd.nextInt(2);
755	>	final int rounds = rnd.nextInt(1, 3);
756	>	final boolean effectiveDelayedPolicy;
757	>	final boolean effectivePeriodicPolicy;
758	>	final boolean effectiveRemovePolicy;
759	>
760	>	if (rnd.nextBoolean())
761	>	p.setExecuteExistingDelayedTasksAfterShutdownPolicy(
762	>	effectiveDelayedPolicy = rnd.nextBoolean());
763	>	else
764	>	effectiveDelayedPolicy = true;
765		assertEquals(effectiveDelayedPolicy,
766		p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
767	+
768	+	if (rnd.nextBoolean())
769	+	p.setContinueExistingPeriodicTasksAfterShutdownPolicy(
770	+	effectivePeriodicPolicy = rnd.nextBoolean());
771	+	else
772	+	effectivePeriodicPolicy = false;
773		assertEquals(effectivePeriodicPolicy,
774		p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
775	+
776	+	if (rnd.nextBoolean())
777	+	p.setRemoveOnCancelPolicy(
778	+	effectiveRemovePolicy = rnd.nextBoolean());
779	+	else
780	+	effectiveRemovePolicy = false;
781		assertEquals(effectiveRemovePolicy,
782		p.getRemoveOnCancelPolicy());
783	<	// Strategy: Wedge the pool with poolSize "blocker" threads
783	>
784	>	final boolean periodicTasksContinue = effectivePeriodicPolicy && rnd.nextBoolean();
785	>
786	>	// Strategy: Wedge the pool with one wave of "blocker" tasks,
787	>	// then add a second wave that waits in the queue until unblocked.
788		final AtomicInteger ran = new AtomicInteger(0);
789		final CountDownLatch poolBlocked = new CountDownLatch(poolSize);
790		final CountDownLatch unblock = new CountDownLatch(1);
791	<	final CountDownLatch periodicLatch1 = new CountDownLatch(2);
792	<	final CountDownLatch periodicLatch2 = new CountDownLatch(2);
793	<	Runnable task = new CheckedRunnable() { public void realRun()
794	<	throws InterruptedException {
795	<	poolBlocked.countDown();
796	<	await(unblock);
797	<	ran.getAndIncrement();
798	<	}};
799	<	List<Future<?>> blockers = new ArrayList<>();
800	<	List<Future<?>> periodics = new ArrayList<>();
801	<	List<Future<?>> delayeds = new ArrayList<>();
802	<	for (int i = 0; i < poolSize; i++)
803	<	blockers.add(p.submit(task));
791	>	final RuntimeException exception = new RuntimeException();
792	>
793	>	class Task implements Runnable {
794	>	public void run() {
795	>	try {
796	>	ran.getAndIncrement();
797	>	poolBlocked.countDown();
798	>	await(unblock);
799	>	} catch (Throwable fail) { threadUnexpectedException(fail); }
800	>	}
801	>	}
802	>
803	>	class PeriodicTask extends Task {
804	>	PeriodicTask(int rounds) { this.rounds = rounds; }
805	>	int rounds;
806	>	public void run() {
807	>	if (--rounds == 0) super.run();
808	>	// throw exception to surely terminate this periodic task,
809	>	// but in a separate execution and in a detectable way.
810	>	if (rounds == -1) throw exception;
811	>	}
812	>	}
813	>
814	>	Runnable task = new Task();
815	>
816	>	List<Future<?>> immediates = new ArrayList<>();
817	>	List<Future<?>> delayeds   = new ArrayList<>();
818	>	List<Future<?>> periodics  = new ArrayList<>();
819	>
820	>	immediates.add(p.submit(task));
821	>	delayeds.add(p.schedule(task, delay, MILLISECONDS));
822	>	periodics.add(p.scheduleAtFixedRate(
823	>	new PeriodicTask(rounds), delay, 1, MILLISECONDS));
824	>	periodics.add(p.scheduleWithFixedDelay(
825	>	new PeriodicTask(rounds), delay, 1, MILLISECONDS));
826	>
827		await(poolBlocked);
828
829	<	periodics.add(p.scheduleAtFixedRate(countDowner(periodicLatch1),
830	<	1, 1, MILLISECONDS));
831	<	periodics.add(p.scheduleWithFixedDelay(countDowner(periodicLatch2),
832	<	1, 1, MILLISECONDS));
833	<	delayeds.add(p.schedule(task, 1, MILLISECONDS));
829	>	assertEquals(poolSize, ran.get());
830	>	assertEquals(poolSize, p.getActiveCount());
831	>	assertTrue(q.isEmpty());
832	>
833	>	// Add second wave of tasks.
834	>	immediates.add(p.submit(task));
835	>	delayeds.add(p.schedule(task, effectiveDelayedPolicy ? delay : LONG_DELAY_MS, MILLISECONDS));
836	>	periodics.add(p.scheduleAtFixedRate(
837	>	new PeriodicTask(rounds), delay, 1, MILLISECONDS));
838	>	periodics.add(p.scheduleWithFixedDelay(
839	>	new PeriodicTask(rounds), delay, 1, MILLISECONDS));
840	>
841	>	assertEquals(poolSize, q.size());
842	>	assertEquals(poolSize, ran.get());
843	>
844	>	immediates.forEach(
845	>	f -> assertTrue(((ScheduledFuture)f).getDelay(NANOSECONDS) <= 0L));
846	>
847	>	Stream.of(immediates, delayeds, periodics).flatMap(c -> c.stream())
848	>	.forEach(f -> assertFalse(f.isDone()));
849
792	–	assertTrue(p.getQueue().containsAll(periodics));
793	–	assertTrue(p.getQueue().containsAll(delayeds));
850		try { p.shutdown(); } catch (SecurityException ok) { return; }
851		assertTrue(p.isShutdown());
852	+	assertTrue(p.isTerminating());
853		assertFalse(p.isTerminated());
854	<	for (Future<?> periodic : periodics) {
855	<	assertTrue(effectivePeriodicPolicy ^ periodic.isCancelled());
856	<	assertTrue(effectivePeriodicPolicy ^ periodic.isDone());
857	<	}
858	<	for (Future<?> delayed : delayeds) {
859	<	assertTrue(effectiveDelayedPolicy ^ delayed.isCancelled());
860	<	assertTrue(effectiveDelayedPolicy ^ delayed.isDone());
861	<	}
862	<	if (testImplementationDetails) {
863	<	assertEquals(effectivePeriodicPolicy,
864	<	p.getQueue().containsAll(periodics));
865	<	assertEquals(effectiveDelayedPolicy,
866	<	p.getQueue().containsAll(delayeds));
867	<	}
868	<	// Release all pool threads
869	<	unblock.countDown();
870	<
871	<	for (Future<?> delayed : delayeds) {
872	<	if (effectiveDelayedPolicy) {
873	<	assertNull(delayed.get());
874	<	}
875	<	}
876	<	if (effectivePeriodicPolicy) {
877	<	await(periodicLatch1);
878	<	await(periodicLatch2);
879	<	for (Future<?> periodic : periodics) {
880	<	assertTrue(periodic.cancel(false));
881	<	assertTrue(periodic.isCancelled());
882	<	assertTrue(periodic.isDone());
883	<	}
854	>
855	>	if (rnd.nextBoolean())
856	>	assertThrows(
857	>	RejectedExecutionException.class,
858	>	() -> p.submit(task),
859	>	() -> p.schedule(task, 1, SECONDS),
860	>	() -> p.scheduleAtFixedRate(
861	>	new PeriodicTask(1), 1, 1, SECONDS),
862	>	() -> p.scheduleWithFixedDelay(
863	>	new PeriodicTask(2), 1, 1, SECONDS));
864	>
865	>	assertTrue(q.contains(immediates.get(1)));
866	>	assertTrue(!effectiveDelayedPolicy
867	>	^ q.contains(delayeds.get(1)));
868	>	assertTrue(!effectivePeriodicPolicy
869	>	^ q.containsAll(periodics.subList(2, 4)));
870	>
871	>	immediates.forEach(f -> assertFalse(f.isDone()));
872	>
873	>	assertFalse(delayeds.get(0).isDone());
874	>	if (effectiveDelayedPolicy)
875	>	assertFalse(delayeds.get(1).isDone());
876	>	else
877	>	assertTrue(delayeds.get(1).isCancelled());
878	>
879	>	if (effectivePeriodicPolicy)
880	>	periodics.forEach(
881	>	f -> {
882	>	assertFalse(f.isDone());
883	>	if (!periodicTasksContinue) {
884	>	assertTrue(f.cancel(false));
885	>	assertTrue(f.isCancelled());
886	>	}
887	>	});
888	>	else {
889	>	periodics.subList(0, 2).forEach(f -> assertFalse(f.isDone()));
890	>	periodics.subList(2, 4).forEach(f -> assertTrue(f.isCancelled()));
891		}
892	<	for (Future<?> blocker : blockers) assertNull(blocker.get());
892	>
893	>	unblock.countDown();    // Release all pool threads
894	>
895		assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
896	+	assertFalse(p.isTerminating());
897		assertTrue(p.isTerminated());
898	<	assertEquals(2 + (effectiveDelayedPolicy ? 1 : 0), ran.get());
899	<	}}
898	>
899	>	assertTrue(q.isEmpty());
900	>
901	>	Stream.of(immediates, delayeds, periodics).flatMap(c -> c.stream())
902	>	.forEach(f -> assertTrue(f.isDone()));
903	>
904	>	for (Future<?> f : immediates) assertNull(f.get());
905	>
906	>	assertNull(delayeds.get(0).get());
907	>	if (effectiveDelayedPolicy)
908	>	assertNull(delayeds.get(1).get());
909	>	else
910	>	assertTrue(delayeds.get(1).isCancelled());
911	>
912	>	if (periodicTasksContinue)
913	>	periodics.forEach(
914	>	f -> {
915	>	try { f.get(); }
916	>	catch (ExecutionException success) {
917	>	assertSame(exception, success.getCause());
918	>	}
919	>	catch (Throwable fail) { threadUnexpectedException(fail); }
920	>	});
921	>	else
922	>	periodics.forEach(f -> assertTrue(f.isCancelled()));
923	>
924	>	assertEquals(poolSize + 1
925	>	+ (effectiveDelayedPolicy ? 1 : 0)
926	>	+ (periodicTasksContinue ? 2 : 0),
927	>	ran.get());
928	>	}
929
930		/**
931		* completed submit of callable returns result

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines