/*
* Copyright 2010-2011 Ning, Inc.
*
* Ning licenses this file to you 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 com.ning.billing.junction.blocking;
import java.util.List;
import java.util.UUID;
import org.mockito.Mockito;
import org.testng.Assert;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import com.ning.billing.account.api.Account;
import com.ning.billing.entitlement.api.user.EntitlementUserApi;
import com.ning.billing.entitlement.api.user.EntitlementUserApiException;
import com.ning.billing.entitlement.api.user.Subscription;
import com.ning.billing.entitlement.api.user.SubscriptionBundle;
import com.ning.billing.junction.JunctionTestSuite;
import com.ning.billing.junction.api.Blockable;
import com.ning.billing.junction.api.BlockingApiException;
import com.ning.billing.junction.api.BlockingState;
import com.ning.billing.junction.block.BlockingChecker;
import com.ning.billing.junction.block.DefaultBlockingChecker;
import com.ning.billing.junction.dao.BlockingStateDao;
import com.ning.billing.util.callcontext.InternalCallContext;
import com.ning.billing.util.callcontext.InternalTenantContext;
import com.ning.billing.util.callcontext.TenantContext;
import com.ning.billing.util.clock.Clock;
import com.ning.billing.util.svcapi.entitlement.EntitlementInternalApi;
import com.ning.billing.util.svcapi.junction.DefaultBlockingState;
import com.google.inject.AbstractModule;
import com.google.inject.Guice;
import com.google.inject.Injector;
public class TestBlockingChecker extends JunctionTestSuite {
private BlockingState bundleState;
private BlockingState subscriptionState;
private BlockingState accountState;
private final BlockingStateDao dao = new BlockingStateDao() {
@Override
public BlockingState getBlockingStateFor(final UUID blockableId, final InternalTenantContext context) {
if (blockableId == account.getId()) {
return accountState;
} else if (blockableId == subscription.getId()) {
return subscriptionState;
} else {
return bundleState;
}
}
@Override
public List<BlockingState> getBlockingHistoryFor(final UUID overdueableId, final InternalTenantContext context) {
throw new UnsupportedOperationException();
}
@Override
public <T extends Blockable> void setBlockingState(final BlockingState state, final Clock clock, final InternalCallContext context) {
throw new UnsupportedOperationException();
}
};
private BlockingChecker checker;
private Subscription subscription;
private Account account;
private SubscriptionBundle bundle;
@BeforeClass(groups = "fast")
public void setup() {
final UUID accountId = UUID.randomUUID();
account = Mockito.mock(Account.class);
Mockito.when(account.getId()).thenReturn(accountId);
bundle = Mockito.mock(SubscriptionBundle.class);
Mockito.when(bundle.getAccountId()).thenReturn(accountId);
final UUID bundleId = UUID.randomUUID();
Mockito.when(bundle.getId()).thenReturn(bundleId);
Mockito.when(bundle.getExternalKey()).thenReturn("key");
subscription = Mockito.mock(Subscription.class);
Mockito.when(subscription.getId()).thenReturn(UUID.randomUUID());
Mockito.when(subscription.getBundleId()).thenReturn(bundleId);
final Injector i = Guice.createInjector(new AbstractModule() {
@Override
protected void configure() {
bind(BlockingChecker.class).to(DefaultBlockingChecker.class).asEagerSingleton();
bind(BlockingStateDao.class).toInstance(dao);
// Since we re-enabled EntitlementUserApi for Checher we need a binding, that should go eventually
final EntitlementUserApi entitlementUserApi = Mockito.mock(EntitlementUserApi.class);
bind(EntitlementUserApi.class).toInstance(entitlementUserApi);
final EntitlementInternalApi entitlementInternalApi = Mockito.mock(EntitlementInternalApi.class);
bind(EntitlementInternalApi.class).toInstance(entitlementInternalApi);
try {
Mockito.when(entitlementUserApi.getBundleFromId(Mockito.<UUID>any(), Mockito.<TenantContext>any())).thenReturn(bundle);
Mockito.when(entitlementInternalApi.getBundleFromId(Mockito.<UUID>any(), Mockito.<InternalTenantContext>any())).thenReturn(bundle);
} catch (EntitlementUserApiException e) {
Assert.fail(e.toString());
}
}
});
checker = i.getInstance(BlockingChecker.class);
}
private void setStateBundle(final boolean bC, final boolean bE, final boolean bB) {
bundleState = new DefaultBlockingState(UUID.randomUUID(), "state", Blockable.Type.SUBSCRIPTION_BUNDLE, "test-service", bC, bE, bB);
Mockito.when(bundle.getBlockingState()).thenReturn(bundleState);
}
private void setStateAccount(final boolean bC, final boolean bE, final boolean bB) {
accountState = new DefaultBlockingState(UUID.randomUUID(), "state", Blockable.Type.SUBSCRIPTION_BUNDLE, "test-service", bC, bE, bB);
}
private void setStateSubscription(final boolean bC, final boolean bE, final boolean bB) {
subscriptionState = new DefaultBlockingState(UUID.randomUUID(), "state", Blockable.Type.SUBSCRIPTION_BUNDLE, "test-service", bC, bE, bB);
Mockito.when(subscription.getBlockingState()).thenReturn(subscriptionState);
}
@Test(groups = "fast")
public void testSubscriptionChecker() throws Exception {
setStateAccount(false, false, false);
setStateBundle(false, false, false);
setStateSubscription(false, false, false);
checker.checkBlockedChange(subscription, internalCallContext);
checker.checkBlockedEntitlement(subscription, internalCallContext);
checker.checkBlockedBilling(subscription, internalCallContext);
//BLOCKED SUBSCRIPTION
setStateSubscription(true, false, false);
checker.checkBlockedEntitlement(subscription, internalCallContext);
checker.checkBlockedBilling(subscription, internalCallContext);
try {
checker.checkBlockedChange(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateSubscription(false, true, false);
checker.checkBlockedChange(subscription, internalCallContext);
checker.checkBlockedBilling(subscription, internalCallContext);
try {
checker.checkBlockedEntitlement(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateSubscription(false, false, true);
checker.checkBlockedChange(subscription, internalCallContext);
checker.checkBlockedEntitlement(subscription, internalCallContext);
try {
checker.checkBlockedBilling(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
//BLOCKED BUNDLE
setStateSubscription(false, false, false);
setStateBundle(true, false, false);
checker.checkBlockedEntitlement(subscription, internalCallContext);
checker.checkBlockedBilling(subscription, internalCallContext);
try {
checker.checkBlockedChange(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateBundle(false, true, false);
checker.checkBlockedChange(subscription, internalCallContext);
checker.checkBlockedBilling(subscription, internalCallContext);
try {
checker.checkBlockedEntitlement(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateBundle(false, false, true);
checker.checkBlockedChange(subscription, internalCallContext);
checker.checkBlockedEntitlement(subscription, internalCallContext);
try {
checker.checkBlockedBilling(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
//BLOCKED ACCOUNT
setStateSubscription(false, false, false);
setStateBundle(false, false, false);
setStateAccount(true, false, false);
checker.checkBlockedEntitlement(subscription, internalCallContext);
checker.checkBlockedBilling(subscription, internalCallContext);
try {
checker.checkBlockedChange(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateAccount(false, true, false);
checker.checkBlockedChange(subscription, internalCallContext);
checker.checkBlockedBilling(subscription, internalCallContext);
try {
checker.checkBlockedEntitlement(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateAccount(false, false, true);
checker.checkBlockedChange(subscription, internalCallContext);
checker.checkBlockedEntitlement(subscription, internalCallContext);
try {
checker.checkBlockedBilling(subscription, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
}
@Test(groups = "fast")
public void testBundleChecker() throws Exception {
setStateAccount(false, false, false);
setStateBundle(false, false, false);
setStateSubscription(false, false, false);
checker.checkBlockedChange(bundle, internalCallContext);
checker.checkBlockedEntitlement(bundle, internalCallContext);
checker.checkBlockedBilling(bundle, internalCallContext);
//BLOCKED BUNDLE
setStateSubscription(false, false, false);
setStateBundle(true, false, false);
checker.checkBlockedEntitlement(bundle, internalCallContext);
checker.checkBlockedBilling(bundle, internalCallContext);
try {
checker.checkBlockedChange(bundle, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateBundle(false, true, false);
checker.checkBlockedChange(bundle, internalCallContext);
checker.checkBlockedBilling(bundle, internalCallContext);
try {
checker.checkBlockedEntitlement(bundle, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateBundle(false, false, true);
checker.checkBlockedChange(bundle, internalCallContext);
checker.checkBlockedEntitlement(bundle, internalCallContext);
try {
checker.checkBlockedBilling(bundle, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
//BLOCKED ACCOUNT
setStateSubscription(false, false, false);
setStateBundle(false, false, false);
setStateAccount(true, false, false);
checker.checkBlockedEntitlement(bundle, internalCallContext);
checker.checkBlockedBilling(bundle, internalCallContext);
try {
checker.checkBlockedChange(bundle, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateAccount(false, true, false);
checker.checkBlockedChange(bundle, internalCallContext);
checker.checkBlockedBilling(bundle, internalCallContext);
try {
checker.checkBlockedEntitlement(bundle, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateAccount(false, false, true);
checker.checkBlockedChange(bundle, internalCallContext);
checker.checkBlockedEntitlement(bundle, internalCallContext);
try {
checker.checkBlockedBilling(bundle, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
}
@Test(groups = "fast")
public void testAccountChecker() throws Exception {
setStateAccount(false, false, false);
setStateBundle(false, false, false);
setStateSubscription(false, false, false);
checker.checkBlockedChange(account, internalCallContext);
checker.checkBlockedEntitlement(account, internalCallContext);
checker.checkBlockedBilling(account, internalCallContext);
//BLOCKED ACCOUNT
setStateSubscription(false, false, false);
setStateBundle(false, false, false);
setStateAccount(true, false, false);
checker.checkBlockedEntitlement(account, internalCallContext);
checker.checkBlockedBilling(account, internalCallContext);
try {
checker.checkBlockedChange(account, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateAccount(false, true, false);
checker.checkBlockedChange(account, internalCallContext);
checker.checkBlockedBilling(account, internalCallContext);
try {
checker.checkBlockedEntitlement(account, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
setStateAccount(false, false, true);
checker.checkBlockedChange(account, internalCallContext);
checker.checkBlockedEntitlement(account, internalCallContext);
try {
checker.checkBlockedBilling(account, internalCallContext);
Assert.fail("The call should have been blocked!");
} catch (BlockingApiException e) {
//Expected behavior
}
}
}