package rx.schedulers;

import static org.junit.Assert.*;

import java.util.*;
import java.util.concurrent.*;

import org.junit.Test;

import rx.Scheduler.Worker;
import rx.functions.Action0;
import rx.internal.schedulers.GenericScheduledExecutorService;
import rx.internal.util.RxRingBuffer;
import rx.schedulers.Schedulers;
import rx.subscriptions.CompositeSubscription;

public class SchedulerLifecycleTest {
    @Test
    public void testShutdown() throws InterruptedException {
        tryOutSchedulers();
        
        System.out.println("testShutdown >> Giving time threads to spin-up");
        Thread.sleep(500);
        
        Set<Thread> rxThreads = new HashSet<Thread>();
        for (Thread t : Thread.getAllStackTraces().keySet()) {
            if (t.getName().startsWith("Rx")) {
                rxThreads.add(t);
            }
        }
        Schedulers.shutdown();
        System.out.println("testShutdown >> Giving time to threads to stop");
        Thread.sleep(500);
        
        StringBuilder b = new StringBuilder();
        for (Thread t : rxThreads) {
            if (t.isAlive()) {
                b.append("Thread " + t + " failed to shutdown\r\n");
                for (StackTraceElement ste : t.getStackTrace()) {
                    b.append("  ").append(ste).append("\r\n");
                }
            }
        }
        if (b.length() > 0) {
            System.out.print(b);
            System.out.println("testShutdown >> Restarting schedulers...");
            Schedulers.start(); // restart them anyways
            fail("Rx Threads were still alive:\r\n" + b);
        }
        
        System.out.println("testShutdown >> Restarting schedulers...");
        Schedulers.start();
        
        tryOutSchedulers();
    }

    private void tryOutSchedulers() throws InterruptedException {
        final CountDownLatch cdl = new CountDownLatch(4);
        
        final Action0 countAction = new Action0() {
            @Override
            public void call() {
                cdl.countDown();
            }
        };
        
        CompositeSubscription csub = new CompositeSubscription();
        
        try {
            Worker w1 = Schedulers.computation().createWorker();
            csub.add(w1);
            w1.schedule(countAction);
            
            Worker w2 = Schedulers.io().createWorker();
            csub.add(w2);
            w2.schedule(countAction);
            
            Worker w3 = Schedulers.newThread().createWorker();
            csub.add(w3);
            w3.schedule(countAction);
            
            GenericScheduledExecutorService.getInstance().execute(new Runnable() {
                @Override
                public void run() {
                    countAction.call();
                }
            });
            
            RxRingBuffer.getSpscInstance().release();
            
            if (!cdl.await(3, TimeUnit.SECONDS)) {
                fail("countAction was not run by every worker");
            }
        } finally {
            csub.unsubscribe();
        }
    }
    
    @Test
    public void testStartIdempotence() throws InterruptedException {
        tryOutSchedulers();
        
        System.out.println("testStartIdempotence >> giving some time");
        Thread.sleep(500);
        
        Set<Thread> rxThreads = new HashSet<Thread>();
        for (Thread t : Thread.getAllStackTraces().keySet()) {
            if (t.getName().startsWith("Rx")) {
                rxThreads.add(t);
                System.out.println("testStartIdempotence >> " + t);
            }
        }
        System.out.println("testStartIdempotence >> trying to start again");
        Schedulers.start();
        System.out.println("testStartIdempotence >> giving some time again");
        Thread.sleep(500);
        
        Set<Thread> rxThreads2 = new HashSet<Thread>();
        for (Thread t : Thread.getAllStackTraces().keySet()) {
            if (t.getName().startsWith("Rx")) {
                rxThreads2.add(t);
                System.out.println("testStartIdempotence >>>> " + t);
            }
        }
        
        assertEquals(rxThreads, rxThreads2);
    }
}