package com.atlassian.sal.core.executor;
   
   import com.atlassian.sal.api.executor.ThreadLocalContext;
   import com.atlassian.sal.api.executor.ThreadLocalContextManager;
   import junit.framework.TestCase;
   
   import java.util.concurrent.CountDownLatch;
   import java.util.concurrent.Executor;
   import java.util.concurrent.Executors;
  import java.util.concurrent.TimeUnit;
  
  import static com.google.common.base.Preconditions.checkNotNull;
  
  public final class TestThreadLocalDelegateExecutor extends TestCase
  {
      public void testRun() throws InterruptedException
      {
          final ThreadLocalContextManager<ThreadLocalContext> manager = new StubThreadLocalContextManager();
          Runnable delegate = new Runnable()
          {
              public void run()
              {
                  assertNotNull(manager.getThreadLocalContext());
              }
          };
  
          manager.setThreadLocalContext(new StubThreadLocalContextManager.Context());
          final Executor executor = new ThreadLocalDelegateExecutor<ThreadLocalContext>(manager, Executors.newSingleThreadExecutor());
  
          final CountDownLatch latch = new CountDownLatch(1);
  
          executor.execute(new CountingDownRunnable(latch, delegate));
  
          latch.await(2, TimeUnit.SECONDS);
          assertNotNull(manager.getThreadLocalContext());
      }
  
      private static final class CountingDownRunnable implements Runnable
      {
          private final CountDownLatch latch;
          private final Runnable delegate;
  
          private CountingDownRunnable(CountDownLatch latch, Runnable delegate)
          {
              this.latch = checkNotNull(latch);
              this.delegate = checkNotNull(delegate);
          }
  
          @Override
          public void run()
          {
              try
              {
                  delegate.run();
              }
              finally
              {
                  latch.countDown();
              }
          }
      }
  }