Java TenantContext类代码示例

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private void sendEmailForUpComingInvoice(final Account account, final ExtBusEvent killbillEvent, final TenantContext context) throws IOException, InvoiceApiException, EmailException, TenantApiException, EmailNotificationException {

        Preconditions.checkArgument(killbillEvent.getEventType() == ExtBusEventType.INVOICE_NOTIFICATION, String.format("Unexpected event %s", killbillEvent.getEventType()));

        final String dryRunTimePropValue = configProperties.getString(INVOICE_DRY_RUN_TIME_PROPERTY);
        Preconditions.checkArgument(dryRunTimePropValue != null, String.format("Cannot find property %s", INVOICE_DRY_RUN_TIME_PROPERTY));

        final TimeSpan span = new TimeSpan(dryRunTimePropValue);

        final DateTime now = clock.getClock().getUTCNow();
        final DateTime targetDateTime = now.plus(span.getMillis());

        final PluginCallContext callContext = new PluginCallContext(EmailNotificationActivator.PLUGIN_NAME, now, context.getTenantId());
        final Invoice invoice = osgiKillbillAPI.getInvoiceUserApi().triggerInvoiceGeneration(account.getId(), new LocalDate(targetDateTime, account.getTimeZone()), NULL_DRY_RUN_ARGUMENTS, callContext);
        if (invoice != null) {
            final EmailContent emailContent = templateRenderer.generateEmailForUpComingInvoice(account, invoice, context);
            sendEmail(account, emailContent, context);
        }
    }
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
@Test(groups = "fast")
public void defaultAccountCustomField() {
    // given
    Account account = EasyTaxTestUtils.createAccount("NZ",
            DateTimeZone.forID("Pacific/Auckland"));
    CustomFieldUserApi fieldApi = Mockito.mock(CustomFieldUserApi.class);
    when(killbillApi.getCustomFieldUserApi()).thenReturn(fieldApi);
    ArgumentCaptor<TenantContext> contextCaptor = ArgumentCaptor.forClass(TenantContext.class);
    CustomField customField = createCustomField(
            AccountCustomFieldTaxZoneResolver.TAX_ZONE_CUSTOM_FIIELD, TEST_TAX_ZONE);
    when(fieldApi.getCustomFieldsForObject(eq(account.getId()), eq(ObjectType.ACCOUNT),
            contextCaptor.capture())).thenReturn(Collections.singletonList(customField));

    // when
    AccountCustomFieldTaxZoneResolver resolver = createResolver(new Properties());
    String taxZone = resolver.taxZoneForInvoice(tenantId, account, null,
            Collections.emptyList());

    // then
    assertEquals(contextCaptor.getValue().getTenantId(), tenantId, "TenantContext tenant ID");
    assertEquals(taxZone, TEST_TAX_ZONE, "Tax zone resolved from account custom field");
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
@Test(groups = "fast")
public void defaultFallbackToAccountCountry() {
    // given
    Account account = EasyTaxTestUtils.createAccount("NZ",
            DateTimeZone.forID("Pacific/Auckland"));
    CustomFieldUserApi fieldApi = Mockito.mock(CustomFieldUserApi.class);
    when(killbillApi.getCustomFieldUserApi()).thenReturn(fieldApi);
    ArgumentCaptor<TenantContext> contextCaptor = ArgumentCaptor.forClass(TenantContext.class);
    when(fieldApi.getCustomFieldsForObject(eq(account.getId()), eq(ObjectType.ACCOUNT),
            contextCaptor.capture())).thenReturn(Collections.emptyList());

    // when
    AccountCustomFieldTaxZoneResolver resolver = createResolver(new Properties());
    String taxZone = resolver.taxZoneForInvoice(tenantId, account, null,
            Collections.emptyList());

    // then
    assertEquals(contextCaptor.getValue().getTenantId(), tenantId, "TenantContext tenant ID");
    assertEquals(taxZone, account.getCountry(),
            "Fallback to account country when no custom field present");
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private void givenPermissions(String username, String password, Set<Permission> permissions) {
    String credentials = username + ":" + password;
    try {
        String authHeaderValue = "Basic "
                + Base64.getEncoder().encodeToString(credentials.getBytes("UTF-8"));
        given(req.getHeader("Authorization")).willReturn(authHeaderValue);
    } catch (UnsupportedEncodingException e) {
        throw new RuntimeException(e);
    }
    given(securityApi.isSubjectAuthenticated()).willReturn(true);
    given(securityApi
            .getCurrentUserPermissions(Matchers.argThat(new ArgumentMatcher<TenantContext>() {

                @Override
                public boolean matches(Object arg) {
                    return (arg instanceof TenantContext
                            && tenantId.equals(((TenantContext) arg).getTenantId()));
                }

            }))).willReturn(permissions);
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private Map<String, String> getTranslationMap(final String accountLocale, final ResourceBundleFactory.ResourceBundleType bundleType, final TenantContext context) throws TenantApiException, EmailNotificationException {
    final ResourceBundle translationBundle = accountLocale != null ?
            bundleFactory.createBundle(LocaleUtils.toLocale(accountLocale), bundleType, context) : null;

    if (translationBundle == null){
        throw new EmailNotificationException(TRANSLATION_INVALID, accountLocale);
    }

    final Map<String, String> text = new HashMap<String, String>();
    Enumeration<String> keys = translationBundle.getKeys();
    while (keys.hasMoreElements()) {
        final String key = keys.nextElement();
        text.put(key, translationBundle.getString(key));
    }
    return text;
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private String getTemplateText(final Locale locale, final TemplateType templateType, final TenantContext context) throws TenantApiException {

        final String defaultTemplateName = DEFAULT_TEMPLATE_PATH_PREFIX + templateType.getDefaultTemplateName();
        if (context.getTenantId() == null) {
            return getDefaultTemplate(defaultTemplateName);
        }

        // TODO Caching strategy
        final String templateTenantKey = LocaleUtils.localeString(locale, templateType.getTemplateKey());
        final List<String> result = tenantApi.getTenantValuesForKey(templateTenantKey, context);
        if (result.size() == 1) {

            return result.get(0);
        }
        return getDefaultTemplate(defaultTemplateName);

    }
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private String getTenantBundleForType(final Locale locale, final ResourceBundleType type, final TenantContext tenantContext) throws TenantApiException {
    String tenantKey;
    switch (type) {
        case CATALOG_TRANSLATION:
            tenantKey = LocaleUtils.localeString(locale, TenantKV.TenantKey.CATALOG_TRANSLATION_.name());
            break;
        case TEMPLATE_TRANSLATION:
            tenantKey = LocaleUtils.localeString(locale, type.getTranslationKey());
            break;
        default:
            return null;
    }
    if (tenantKey !=  null) {
        final List<String> result = tenantApi.getTenantValuesForKey(tenantKey, tenantContext);
        if (result.size() == 1) {
            return result.get(0);
        }
    }
    return null;
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
@Test(groups = "fast")
public void testPaymentRefund() throws Exception {
    final AccountData account = createAccount();
    final PaymentTransaction paymentTransaction = createPaymentTransaction(new BigDecimal("937.070000000"), Currency.USD);

    final TenantContext tenantContext = createTenantContext();
    final EmailContent email = renderer.generateEmailForPaymentRefund(account, paymentTransaction, tenantContext);

    final String expectedBody = "*** Your payment has been refunded ***\n" +
            "\n" +
            "We have processed a refund in the amount of $937.07.\n" +
            "\n" +
            "This refund will appear on your next credit card statement in approximately 3-5 business days.\n" +
            "\n" +
            "If you have any questions about your account, please reply to this email or contact MERCHANT_NAME Support at: (888) 555-1234";

    // System.err.println(email.getBody());
    Assert.assertEquals(email.getSubject(), "MERCHANT_NAME: Refund Receipt");
    Assert.assertEquals(email.getBody(), expectedBody);
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
@Test(groups = "fast")
public void testSubscriptionCancellationEffective() throws Exception {

    final AccountData account = createAccount();
    final Subscription cancelledSubscription = createFutureCancelledSubscription(new LocalDate("2015-04-06"), "myPlanName");

    final TenantContext tenantContext = createTenantContext();
    final EmailContent email = renderer.generateEmailForSubscriptionCancellationEffective(account, cancelledSubscription, tenantContext);

    final String expectedBody = "Your Personal subscription to MERCHANT_NAME was canceled earlier and has now ended. Your access to MERCHANT_NAME on your iPad will end shortly.\n" +
            "\n" +
            "We're sorry to see you go.  To reactivate your MERCHANT_NAME service, contact the Support Team at (888) 555-1234.\n" +
            "\n" +
            "If you have any questions about your account, please reply to this email or contact MERCHANT_NAME Support at: (888) 555-1234";

    //System.err.println(email.getBody());
    Assert.assertEquals(email.getSubject(), "MERCHANT_NAME: Subscription Ended");
    Assert.assertEquals(email.getBody(), expectedBody);
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private void withThreePagesOfSearchResults(TenantContext tenantContext) {
    when(customFieldApi.searchCustomFields(anyString(), eq(0L), eq(PAGE_SIZE), eq(tenantContext)))//
            .thenReturn(pageStart);
    when(customFieldApi.searchCustomFields(anyString(), eq(PAGE_SIZE), eq(PAGE_SIZE), eq(tenantContext)))//
            .thenReturn(pageMiddle);
    when(customFieldApi.searchCustomFields(anyString(), eq(2 * PAGE_SIZE), eq(PAGE_SIZE), eq(tenantContext)))//
            .thenReturn(pageEnd);

    when(pageStart.getNextOffset()).thenReturn(PAGE_SIZE);
    when(pageMiddle.getNextOffset()).thenReturn(2 * PAGE_SIZE);
    when(pageEnd.getNextOffset()).thenReturn(null);

    CustomFieldBuilder builder = new CustomFieldBuilder().withObjectType(ACCOUNT).withFieldName("toto");
    when(pageStart.iterator()).thenReturn(forArray(//
            copy(builder).withFieldValue("page0-invoice").withObjectType(INVOICE).build(),//
            copy(builder).withFieldValue("page0-account").build()));
    when(pageMiddle.iterator()).thenReturn(forArray(//
            copy(builder).withFieldValue("page1-toto").build(),//
            copy(builder).withFieldValue("page1-plop").withFieldName("plop").build()));
    when(pageEnd.iterator()).thenReturn(forArray(//
            copy(builder).withFieldValue("page2").build()));
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
@Override
public List<PaymentTransactionInfoPlugin> getPaymentInfo(final UUID kbAccountId, final UUID kbPaymentId, final Iterable<PluginProperty> properties, final TenantContext context) throws PaymentPluginApiException {
    final List<PaymentTransactionInfoPlugin> transactions = super.getPaymentInfo(kbAccountId, kbPaymentId, properties, context);
    if (transactions.isEmpty()) {
        // We don't know about this payment (maybe it was aborted in a control plugin)
        return transactions;
    }

    final ExpiredPaymentPolicy expiredPaymentPolicy = expiredPaymentPolicy(context);
    if (expiredPaymentPolicy.isExpired(transactions)) {
        cancelExpiredPayment(expiredPaymentPolicy.latestTransaction(transactions), context);
        // reload payment
        return super.getPaymentInfo(kbAccountId, kbPaymentId, properties, context);
    }
    return transactions;
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private void cancelExpiredPayment(PaymentTransactionInfoPlugin expiredTransaction, final TenantContext context) {
    final List<PluginProperty> updatedStatusProperties = PluginProperties.buildPluginProperties(
            ImmutableMap.builder()
                        .put(PROPERTY_FROM_HPP_TRANSACTION_STATUS,
                             PaymentPluginStatus.CANCELED.toString())
                        .put("message",
                             "Payment Expired - Cancelled by Janitor")
                        .build());

    try {
        dao.updateResponse(expiredTransaction.getKbTransactionPaymentId(),
                           PluginProperties.merge(expiredTransaction.getProperties(), updatedStatusProperties),
                           context.getTenantId());
    } catch (final SQLException e) {
        logService.log(LogService.LOG_ERROR, "Unable to update canceled payment", e);
    }
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private PaymentTransactionInfoPlugin getPaymentTransactionInfoPluginForHPP(final TransactionType transactionType, final UUID kbAccountId, final UUID kbPaymentId, final UUID kbTransactionId, final BigDecimal amount, final Currency currency, final Iterable<PluginProperty> properties, final TenantContext context) throws PaymentPluginApiException {
    final AdyenPaymentServiceProviderHostedPaymentPagePort hostedPaymentPagePort = adyenHppConfigurationHandler.getConfigurable(context.getTenantId());
    final Map<String, String> requestParameterMap = Maps.transformValues(PluginProperties.toStringMap(properties), new Function<String, String>() {
        @Override
        public String apply(final String input) {
            // Adyen will encode parameters like merchantSig
            return decode(input);
        }
    });
    final HppCompletedResult hppCompletedResult = hostedPaymentPagePort.parseAndVerifyRequestIntegrity(requestParameterMap);
    final PurchaseResult purchaseResult = new PurchaseResult(hppCompletedResult);

    final DateTime utcNow = clock.getUTCNow();
    try {
        final AdyenResponsesRecord adyenResponsesRecord = dao.addResponse(kbAccountId, kbPaymentId, kbTransactionId, transactionType, amount, currency, purchaseResult, utcNow, context.getTenantId());
        return buildPaymentTransactionInfoPlugin(adyenResponsesRecord);
    } catch (final SQLException e) {
        throw new PaymentPluginApiException("HPP payment came through, but we encountered a database error", e);
    }
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
private PaymentData buildPaymentData(final String merchantAccount, final String countryCode, final AccountData account, final UUID kbPaymentId, final UUID kbTransactionId, final AdyenPaymentMethodsRecord paymentMethodsRecord, final BigDecimal amount, final Currency currency, final Iterable<PluginProperty> properties, final TenantContext context) throws PaymentPluginApiException {
    final Payment payment;
    try {
        payment = killbillAPI.getPaymentApi().getPayment(kbPaymentId, false, false, properties, context);
    } catch (final PaymentApiException e) {
        throw new PaymentPluginApiException(String.format("Unable to retrieve kbPaymentId='%s'", kbPaymentId), e);
    }

    final PaymentTransaction paymentTransaction = Iterables.<PaymentTransaction>find(payment.getTransactions(),
                                                                                     new Predicate<PaymentTransaction>() {
                                                                                         @Override
                                                                                         public boolean apply(final PaymentTransaction input) {
                                                                                             return kbTransactionId.equals(input.getId());
                                                                                         }
                                                                                     });

    final PaymentInfo paymentInfo = buildPaymentInfo(merchantAccount, countryCode, account, paymentMethodsRecord, properties, context);

    return new PaymentData<PaymentInfo>(amount, currency, paymentTransaction.getExternalKey(), paymentInfo);
}
 

import org.killbill.billing.util.callcontext.TenantContext; //导入依赖的package包/类
public Collection<BusinessInvoice> getInvoicesForAccount(final UUID accountId, final TenantContext context) {
    final Long accountRecordId = getAccountRecordId(accountId, context);
    final Long tenantRecordId = getTenantRecordId(context);

    final List<BusinessInvoiceItemBaseModelDao> businessInvoiceItemModelDaos = new ArrayList<BusinessInvoiceItemBaseModelDao>();
    businessInvoiceItemModelDaos.addAll(sqlDao.getInvoiceAdjustmentsByAccountRecordId(accountRecordId, tenantRecordId, context));
    businessInvoiceItemModelDaos.addAll(sqlDao.getInvoiceItemsByAccountRecordId(accountRecordId, tenantRecordId, context));
    businessInvoiceItemModelDaos.addAll(sqlDao.getInvoiceItemAdjustmentsByAccountRecordId(accountRecordId, tenantRecordId, context));
    businessInvoiceItemModelDaos.addAll(sqlDao.getInvoiceItemCreditsByAccountRecordId(accountRecordId, tenantRecordId, context));

    final Map<UUID, List<BusinessInvoiceItemBaseModelDao>> itemsPerInvoice = new LinkedHashMap<UUID, List<BusinessInvoiceItemBaseModelDao>>();
    for (final BusinessInvoiceItemBaseModelDao businessInvoiceModelDao : businessInvoiceItemModelDaos) {
        if (itemsPerInvoice.get(businessInvoiceModelDao.getInvoiceId()) == null) {
            itemsPerInvoice.put(businessInvoiceModelDao.getInvoiceId(), new LinkedList<BusinessInvoiceItemBaseModelDao>());
        }
        itemsPerInvoice.get(businessInvoiceModelDao.getInvoiceId()).add(businessInvoiceModelDao);
    }

    final List<BusinessInvoiceModelDao> businessInvoiceModelDaos = sqlDao.getInvoicesByAccountRecordId(accountRecordId, tenantRecordId, context);
    return Lists.transform(businessInvoiceModelDaos, new Function<BusinessInvoiceModelDao, BusinessInvoice>() {
        @Override
        public BusinessInvoice apply(final BusinessInvoiceModelDao input) {
            return new BusinessInvoice(input, MoreObjects.firstNonNull(itemsPerInvoice.get(input.getInvoiceId()), ImmutableList.<BusinessInvoiceItemBaseModelDao>of()));
        }
    });
}