ScopeTest.java

/*
 * Copyright [2014] [Christian Loehnert, krampenschiesser@gmail.com]
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package de.ks.activity.context;

import de.ks.LauncherRunner;
import org.junit.After;
import org.junit.Before;
import org.junit.Ignore;
import org.junit.Test;
import org.junit.runner.RunWith;

import javax.enterprise.inject.spi.CDI;
import javax.inject.Inject;
import java.util.Collections;
import java.util.concurrent.*;

import static org.junit.Assert.*;

@RunWith(LauncherRunner.class)
public class ScopeTest {
  @Inject
  ActivityScopedBean1 bean1;
  @Inject
  ActivityScopedBean1 bean2;
  @Inject
  ActivityContext context;
  ExecutorService service;

  @Before
  public void setUp() throws Exception {
    service = Executors.newCachedThreadPool();
  }

  @After
  public void tearDown() throws Exception {
    ActivityContext.stopAll();
    service.shutdown();
  }

  @Test
  public void testActivityContextActive() throws Exception {
    ActivityContext context = (ActivityContext) CDI.current().getBeanManager().getContext(ActivityScoped.class);
    assertTrue(context.isActive());

    context.startActivity("1").getCount();

    bean1.getName();

    context.startActivity("2");
    bean2.getName();
    assertEquals(2, context.activities.size());


    context.stopActivity("1");
    assertEquals(1, context.activities.size());
    assertNull(context.activities.get("1"));
    assertFalse(context.activities.get("2").getObjectStore().isEmpty());

    context.cleanupAllActivities();
    assertEquals(0, context.activities.size());
  }

  @Test
  public void testScopePropagation() throws Exception {
    ActivityContext.start("2");

    bean1.setValue("Hello Sauerland!");

    Callable<Object> callable = () -> {
      ActivityScopedBean1 bean = CDI.current().select(ActivityScopedBean1.class).get();
      assertNotNull(bean.getValue());
      assertEquals("Hello Sauerland!", bean.getValue());
      return null;
    };
    service.invokeAny(Collections.singletonList(callable));
    assertEquals("Hello Sauerland!", bean1.getValue());
  }

  @Ignore
  @Test
  public void testScopeCleanup() throws Exception {
    final CyclicBarrier barrier = new CyclicBarrier(2);
    ActivityContext.start("2");

    bean1.setValue("Hello Sauerland!");

    service.execute(() -> {
      ActivityScopedBean1 bean = CDI.current().select(ActivityScopedBean1.class).get();
      assertNotNull(bean.getValue());
      assertEquals("Hello Sauerland!", bean.getValue());
      try {
        barrier.await();
      } catch (InterruptedException | BrokenBarrierException e) {
        //brz
      }
    });
    assertEquals(1, context.activities.size());
    assertEquals("Hello Sauerland!", bean1.getValue());
    ActivityContext.stop("2");

    assertEquals("Activity got removed although one thread still holds it", 1, context.activities.size());
    barrier.await();
    Thread.sleep(100);

    assertEquals(0, context.activities.size());
  }
}