Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.yahoo.vespa.hosted.controller.api.integration.billing.BillingDatabaseClientMock Maven / Gradle / Ivy
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.api.integration.billing;
import com.yahoo.config.provision.TenantName;
import java.time.Clock;
import java.time.LocalDate;
import java.time.ZoneId;
import java.time.ZonedDateTime;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.UUID;
import java.util.stream.Collectors;
/**
* @author olaa
*/
public class BillingDatabaseClientMock implements BillingDatabaseClient {
private final Clock clock;
private final PlanRegistry planRegistry;
private final Map tenantPlans = new HashMap<>();
private final Map invoices = new HashMap<>();
private final Map> lineItems = new HashMap<>();
private final Map> uncommittedLineItems = new HashMap<>();
private final Map statuses = new HashMap<>();
private final Map startTimes = new HashMap<>();
private final Map endTimes = new HashMap<>();
private final Map exportedInvoiceIds = new HashMap<>();
private final ZonedDateTime startTime = LocalDate.of(2020, 4, 1).atStartOfDay(ZoneId.of("UTC"));
private final ZonedDateTime endTime = LocalDate.of(2020, 5, 1).atStartOfDay(ZoneId.of("UTC"));
private final List paymentInstruments = new ArrayList<>();
private final Map collectionMethods = new HashMap<>();
public BillingDatabaseClientMock(Clock clock, PlanRegistry planRegistry) {
this.clock = clock;
this.planRegistry = planRegistry;
}
@Override
public boolean setActivePaymentInstrument(InstrumentOwner paymentInstrument) {
return paymentInstruments.add(paymentInstrument);
}
@Override
public Optional getDefaultPaymentInstrumentForTenant(TenantName tenantName) {
return paymentInstruments.stream()
.filter(paymentInstrument -> paymentInstrument.getTenantName().equals(tenantName))
.findFirst();
}
public BillStatus getStatus(Bill.Id invoiceId) {
return statuses.get(invoiceId).current();
}
@Override
public Bill.Id createBill(TenantName tenant, ZonedDateTime startTime, ZonedDateTime endTime, String agent) {
var invoiceId = Bill.Id.generate();
invoices.put(invoiceId, tenant);
statuses.computeIfAbsent(invoiceId, l -> StatusHistory.open(clock));
startTimes.put(invoiceId, startTime);
endTimes.put(invoiceId, endTime);
return invoiceId;
}
@Override
public Optional readBill(Bill.Id billId) {
var invoice = Optional.ofNullable(invoices.get(billId));
var lines = lineItems.getOrDefault(billId, List.of());
var status = statuses.getOrDefault(billId, StatusHistory.open(clock));
var start = startTimes.getOrDefault(billId, startTime);
var end = endTimes.getOrDefault(billId, endTime);
var exportedId = exportedInvoiceId(billId);
return invoice.map(tenant -> new Bill(billId, tenant, status, lines, start, end, exportedId));
}
@Override
public String addLineItem(TenantName tenantName, Bill.LineItem lineItem, Optional invoiceId) {
var lineItemId = UUID.randomUUID().toString();
invoiceId.ifPresentOrElse(
invoice -> lineItems.computeIfAbsent(invoice, l -> new ArrayList<>()).add(lineItem),
() -> uncommittedLineItems.computeIfAbsent(tenantName, l -> new ArrayList<>()).add(lineItem)
);
return lineItemId;
}
@Override
public void setStatus(Bill.Id invoiceId, String agent, BillStatus status) {
statuses.computeIfAbsent(invoiceId, k -> StatusHistory.open(clock))
.getHistory()
.put(ZonedDateTime.now(), status);
}
@Override
public List getUnusedLineItems(TenantName tenantName) {
return uncommittedLineItems.getOrDefault(tenantName, new ArrayList<>());
}
@Override
public void deleteLineItem(String lineItemId) {
uncommittedLineItems.values()
.forEach(list ->
list.removeIf(lineItem -> lineItem.id().equals(lineItemId))
);
}
@Override
public void commitLineItems(TenantName tenantName, Bill.Id invoiceId) {
}
@Override
public Optional getPlan(TenantName tenantName) {
return Optional.ofNullable(tenantPlans.get(tenantName));
}
@Override
public Map> getPlans(List tenants) {
return tenantPlans.entrySet().stream()
.filter(entry -> tenants.contains(entry.getKey()))
.collect(Collectors.toMap(
entry -> entry.getKey(),
entry -> planRegistry.plan(entry.getValue().id())
));
}
@Override
public void setPlan(TenantName tenantName, Plan plan) {
tenantPlans.put(tenantName, plan);
}
@Override
public void deactivateDefaultPaymentInstrument(TenantName tenant) {
paymentInstruments.removeIf(instrumentOwner -> instrumentOwner.getTenantName().equals(tenant));
}
@Override
public Optional getCollectionMethod(TenantName tenantName) {
return Optional.ofNullable(collectionMethods.get(tenantName));
}
@Override
public void setCollectionMethod(TenantName tenantName, CollectionMethod collectionMethod) {
collectionMethods.put(tenantName, collectionMethod);
}
@Override
public List readBillsForTenant(TenantName tenant) {
return invoices.entrySet().stream()
.filter(entry -> entry.getValue().equals(tenant))
.map(Map.Entry::getKey)
.map(invoiceId -> {
var items = lineItems.getOrDefault(invoiceId, List.of());
var status = statuses.get(invoiceId);
var start = startTimes.get(invoiceId);
var end = endTimes.get(invoiceId);
return new Bill(invoiceId, tenant, status, items, start, end, exportedInvoiceId(invoiceId));
})
.toList();
}
@Override
public List readBills() {
return invoices.keySet().stream()
.map(invoiceId -> {
var tenant = invoices.get(invoiceId);
var items = lineItems.getOrDefault(invoiceId, List.of());
var status = statuses.get(invoiceId);
var start = startTimes.get(invoiceId);
var end = endTimes.get(invoiceId);
return new Bill(invoiceId, tenant, status, items, start, end, exportedInvoiceId(invoiceId));
})
.toList();
}
@Override
public void maintain() {}
@Override
public void setExportedInvoiceId(Bill.Id billId, String invoiceId) {
exportedInvoiceIds.put(billId, invoiceId);
}
@Override
public void setExportedInvoiceItemId(String lineItemId, String invoiceItemId) { }
private String exportedInvoiceId(Bill.Id billId) {
return exportedInvoiceIds.getOrDefault(billId, null);
}
}
