How to use

DefaultOverdueStateSet

in

com.ning.billing.overdue.config

@Override
public DefaultOverdueState calculateOverdueState(final BillingState billingState, final LocalDate now) throws OverdueApiException {
  for (final DefaultOverdueState overdueState : getStates()) {
    if (overdueState.getCondition() != null && overdueState.getCondition().evaluate(billingState, now)) {
      return overdueState;
    }
  }
  return getClearState();
}

final InputStream is = new ByteArrayInputStream(xml.getBytes());
final OverdueConfig c = XMLLoader.getObjectFromStreamNoValidation(is, OverdueConfig.class);
Assert.assertEquals(c.getStateSet().size(), 2);
Assert.assertNull(c.getStateSet().getStates()[0].getEnterStateEmailNotification());
Assert.assertNotNull(c.getStateSet().getInitialReevaluationInterval());
Assert.assertEquals(c.getStateSet().getInitialReevaluationInterval().getDays(), 1);
final EmailNotification secondNotification = c.getStateSet().getStates()[1].getEnterStateEmailNotification();
Assert.assertEquals(secondNotification.getSubject(), "ToTo");
Assert.assertEquals(secondNotification.getTemplateName(), "Titi");

@Override
public int size() {
  return getStates().length;
}

@Test(groups = "slow")
public void testApplicator() throws Exception {
  final InputStream is = new ByteArrayInputStream(testOverdueHelper.getConfigXml().getBytes());
  final OverdueConfig config = XMLLoader.getObjectFromStreamNoValidation(is, OverdueConfig.class);
  overdueWrapperFactory.setOverdueConfig(config);
  final Account account = Mockito.mock(Account.class);
  Mockito.when(account.getId()).thenReturn(UUID.randomUUID());
  final OverdueStateSet overdueStateSet = config.getStateSet();
  final OverdueState clearState = config.getStateSet().findState(DefaultBlockingState.CLEAR_STATE_NAME);
  OverdueState state;
  state = config.getStateSet().findState("OD1");
  applicator.apply(overdueStateSet, null, account, clearState, state, internalCallContext);
  testOverdueHelper.checkStateApplied(state);
  checkBussEvent("OD1");
  state = config.getStateSet().findState("OD2");
  applicator.apply(overdueStateSet, null, account, clearState, state, internalCallContext);
  testOverdueHelper.checkStateApplied(state);
  checkBussEvent("OD2");
  state = config.getStateSet().findState("OD3");
  applicator.apply(overdueStateSet, null, account, clearState, state, internalCallContext);
  testOverdueHelper.checkStateApplied(state);
  checkBussEvent("OD3");
}

  @Override
  public OverdueState getFirstState() {
    return getStates()[0];
  }
}

  @Test(groups = "slow")
  public void testWrapperNoConfig() throws Exception {
    overdueWrapperFactory.setOverdueConfig(null);

    final Account account;
    final OverdueWrapper wrapper;
    final OverdueState state;

    final InputStream is = new ByteArrayInputStream(testOverdueHelper.getConfigXml().getBytes());
    final OverdueConfig config = XMLLoader.getObjectFromStreamNoValidation(is, OverdueConfig.class);
    state = config.getStateSet().findState(DefaultBlockingState.CLEAR_STATE_NAME);
    account = testOverdueHelper.createAccount(clock.getUTCToday().minusDays(31));
    wrapper = overdueWrapperFactory.createOverdueWrapperFor(account);
    final OverdueState result = wrapper.refresh(internalCallContext);

    Assert.assertEquals(result.getName(), state.getName());
    Assert.assertEquals(result.blockChanges(), state.blockChanges());
    Assert.assertEquals(result.disableEntitlementAndChangesBlocked(), state.disableEntitlementAndChangesBlocked());
  }
}

@Override
public ValidationErrors validate(final OverdueConfig root,
                 final ValidationErrors errors) {
  for (final DefaultOverdueState state : getStates()) {
    state.validate(root, errors);
  }
  try {
    getClearState();
  } catch (OverdueApiException e) {
    if (e.getCode() == ErrorCode.CAT_MISSING_CLEAR_STATE.getCode()) {
      errors.add("Overdue state set is missing a clear state.",
            root.getURI(), this.getClass(), "");
    }
  }
  return errors;
}

@Override
public OverdueState findState(final String stateName) throws OverdueApiException {
  if (stateName.equals(DefaultBlockingState.CLEAR_STATE_NAME)) {
    return clearState;
  }
  for (final DefaultOverdueState state : getStates()) {
    if (state.getName().equals(stateName)) {
      return state;
    }
  }
  throw new OverdueApiException(ErrorCode.CAT_NO_SUCH_OVEDUE_STATE, stateName);
}

@Test(groups = "slow")
public void testWrapperBasic() throws Exception {
  final InputStream is = new ByteArrayInputStream(testOverdueHelper.getConfigXml().getBytes());
  final OverdueConfig config = XMLLoader.getObjectFromStreamNoValidation(is, OverdueConfig.class);
  overdueWrapperFactory.setOverdueConfig(config);
  Account account;
  OverdueWrapper wrapper;
  OverdueState state;
  state = config.getStateSet().findState("OD1");
  account = testOverdueHelper.createAccount(clock.getUTCToday().minusDays(31));
  wrapper = overdueWrapperFactory.createOverdueWrapperFor(account);
  wrapper.refresh(internalCallContext);
  testOverdueHelper.checkStateApplied(state);
  state = config.getStateSet().findState("OD2");
  account = testOverdueHelper.createAccount(clock.getUTCToday().minusDays(41));
  wrapper = overdueWrapperFactory.createOverdueWrapperFor(account);
  wrapper.refresh(internalCallContext);
  testOverdueHelper.checkStateApplied(state);
  state = config.getStateSet().findState("OD3");
  account = testOverdueHelper.createAccount(clock.getUTCToday().minusDays(51));
  wrapper = overdueWrapperFactory.createOverdueWrapperFor(account);
  wrapper.refresh(internalCallContext);
  testOverdueHelper.checkStateApplied(state);
}