Better JerseyTest

BetterJerseyTest.java

public class BetterJerseyTest extends JerseyTest {

  //Fire up jersey with Guice
  private static final AppDescriptor APP_DESCRIPTOR = new WebAppDescriptor.Builder("com.some.package.name")
    .filterClass(GuiceFilter.class)
    .contextPath("jersey-ctx-path")
    .servletPath("/")
    .clientConfig(new DefaultClientConfig(JacksonJaxbJsonProvider.class))
    .build();

  public JerseyTestNG() {
    super(APP_DESCRIPTOR);
  }

  @Override
  protected TestContainerFactory getTestContainerFactory() throws TestContainerException {
    return new OnePerAppDescriptorTestContainerFactory(super.getTestContainerFactory());
  }

  @BeforeSuite
  public void initTestContainer() throws Exception {
    setUp();
  }

  @AfterSuite
  public void tearDownTestContainer() throws Exception {
    tearDown();
  }
}

OnePerAppDescriptorTestContainerFactory.java

class OnePerAppDescriptorTestContainerFactory implements TestContainerFactory {
  private static final ConcurrentMap<AppDescriptor, TestContainer> cache = new ConcurrentHashMap<AppDescriptor, TestContainer>();

  private final TestContainerFactory tcf;

  public OnePerAppDescriptorTestContainerFactory(TestContainerFactory tcf) {
    this.tcf = tcf;
  }

  @Override
  public <T extends AppDescriptor> Class<T> supports() {
    return tcf.supports();
  }

  @Override
  public TestContainer create(URI baseUri, AppDescriptor ad) {
    if (cache.get(ad) == null) {
      cache.putIfAbsent(ad, tcf.create(baseUri, ad));
    }
    return cache.get(ad);
  }
}