package com.atlassian.sal.core.executor;
   
   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<Object> manager = new StubThreadLocalContextManager();
          Runnable delegate = new Runnable() {
              public void run() {
                  assertNotNull(manager.getThreadLocalContext());
              }
          };
  
          manager.setThreadLocalContext(new Object());
          final Executor executor = new ThreadLocalDelegateExecutor<Object>(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();
              }
          }
      }
  }