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

import junit.framework.*;

import org.litesoft.commonfoundation.base.*;
import org.litesoft.core.delayed.*;
import org.litesoft.core.util.*;
import org.litesoft.delayed.nonpublic.*;

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

    @Override
    protected void setUp()
            throws Exception
    {
        super.setUp();
        mMillisecTimeSource = MillisecTimeSource.INSTANCE;
        mAfter = null;
        mRan = false;
        mError = null;
        mTimer = new ThreadBasedDelayedTimedRunnableRunner();
        mTimer.initialize( new TimedRunnable()
                           {
                               @Override
                               public Again runOnce()
                               {
                                   mAfter = mMillisecTimeSource.now();
                                   mRan = true;
                                   return null;
                               }
                           }, new UnexpectedExceptionHandler()
        {
            @Override
            public void onUnexpectedException( String pFrom, Throwable pException )
            {
                mRan = true;
                mError = pException;
            }
        }
        );
    }

    private MillisecTimeSource mMillisecTimeSource;
    private volatile Long mAfter;
    private volatile boolean mRan;
    private volatile Throwable mError;
    private ThreadBasedDelayedTimedRunnableRunner mTimer;

    public void testTimer()
            throws Exception
    {
        long before = mMillisecTimeSource.now();
        long targetTime = before + 100;
        mTimer.scheduleFor( targetTime );
        // Give it upto a full second to run...
        for ( int i = 0; i < 100; i++ )
        {
            Thread.sleep( 10 );
            if ( mRan )
            {
                break;
            }
        }
        assertTrue( mRan );
        assertTrue( mAfter >= targetTime );
        assertNull( mError );
    }

    public void testCancelTimer()
            throws Exception
    {
        long before = mMillisecTimeSource.now();
        mTimer.scheduleFor( before + 1000 );
        Thread.sleep( 10 );
        mTimer.cancel();
        Thread.sleep( 10 );
        assertFalse( "!Canceled", mTimer.isScheduled() );
        assertFalse( "Timed thread was supposed to be canceled", mRan );
        assertNull( mError );
    }
}