// This Source Code is in the Public Domain per: http://unlicense.org
package org.litesoft.delayed;

import java.util.*;

import junit.framework.*;

import org.litesoft.core.delayed.*;

public class TimedRunnableManagerTest extends TestCase
{
    public static TestSuite suite()
    {
        return new TestSuite( TimedRunnableManagerTest.class );
    }

    public void testTimer()
            throws Exception
    {
        CallCollector zCC = new CallCollector();

        TimedRunnableManager zTRM = TimedRunnableManager.INSTANCE;

        TimedRunnable zA = zCC.createCallBack( "A", null );
        TimedRunnable zB = zCC.createCallBack( "B", -1 );
        TimedRunnable zC = zCC.createCallBack( "C", 40 );
        TimedRunnable zD = zCC.createCallBack( "D", null );

        zTRM.runIn( zA, 200 );
        zTRM.runIn( zB, 20 );
        zTRM.runIn( zC, 80 );
        zTRM.runIn( zD, 2 );

        zCC.waitFor( 6 );

        assertEquals( "[D, B, B, C, C, A]", zCC.mCalls.toString() );
    }

    private static class CallCollector
    {
        public List<String> mCalls = new ArrayList<String>();

        private synchronized void addCall( String pCallID )
        {
            mCalls.add( pCallID );
            notify();
        }

        public synchronized void waitFor( int pCalls )
                throws InterruptedException
        {
            while ( mCalls.size() < pCalls )
            {
                wait();
            }
        }

        public TimedRunnable createCallBack( String pCallID, Integer pAgain )
        {
            return new MyTR( pCallID, pAgain );
        }

        private class MyTR implements TimedRunnable
        {
            private String mCallID;
            private Integer mAgain;

            public MyTR( String pCallID, Integer pAgain )
            {
                mCallID = pCallID;
                mAgain = pAgain;
            }

            @Override
            public Again runOnce()
            {
                addCall( mCallID );
                if ( mAgain == null )
                {
                    return null;
                }
                long zAgain = mAgain;
                mAgain = null;
                return new RunAgainIn( zAgain );
            }
        }
    }
}