How to use

org.killbill.billing.util.entity.dao.DBRouterUntyped

constructor

public EntitySqlDaoTransactionalJdbiWrapper(final IDBI dbi, final IDBI roDbi, final Clock clock, final CacheControllerDispatcher cacheControllerDispatcher,
                      final NonEntityDao nonEntityDao, final InternalCallContextFactory internalCallContextFactory) {
  this.clock = clock;
  this.cacheControllerDispatcher = cacheControllerDispatcher;
  this.nonEntityDao = nonEntityDao;
  this.internalCallContextFactory = internalCallContextFactory;
  this.dbRouter = new DBRouterUntyped(dbi, roDbi);
}

public EntitySqlDaoTransactionalJdbiWrapper(final IDBI dbi, final IDBI roDbi, final Clock clock, final CacheControllerDispatcher cacheControllerDispatcher,
                      final NonEntityDao nonEntityDao, final InternalCallContextFactory internalCallContextFactory) {
  this.clock = clock;
  this.cacheControllerDispatcher = cacheControllerDispatcher;
  this.nonEntityDao = nonEntityDao;
  this.internalCallContextFactory = internalCallContextFactory;
  this.dbRouter = new DBRouterUntyped(dbi, roDbi);
}

@Inject
public TestDBRouterAPI() {
  final IDBI dbi = Mockito.mock(IDBI.class);
  Mockito.when(dbi.open()).thenAnswer(new Answer<Handle>() {
    @Override
    public Handle answer(final InvocationOnMock invocation) {
      rwCalls.incrementAndGet();
      return null;
    }
  });
  final IDBI roDbi = Mockito.mock(IDBI.class);
  Mockito.when(roDbi.open()).thenAnswer(new Answer<Handle>() {
    @Override
    public Handle answer(final InvocationOnMock invocation) {
      roCalls.incrementAndGet();
      return null;
    }
  });
  this.dbRouter = new DBRouterUntyped(dbi, roDbi);
}