third_party/WebKit/Source/modules/payments/PaymentDetailsTestHelper.cpp

// Copyright 2016 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "modules/payments/PaymentDetailsTestHelper.h"

#include "modules/payments/CurrencyAmount.h"
#include "modules/payments/PaymentItem.h"
#include "modules/payments/ShippingOption.h"
#include "platform/heap/HeapAllocator.h"

namespace blink {

PaymentDetails buildPaymentDetailsForTest(PaymentTestDetailToChange detail, PaymentTestDataToChange data, PaymentTestModificationType modificationType, const String& valueToUse)
{
    CurrencyAmount itemAmount;
    if (detail == PaymentTestDetailItem && data == PaymentTestDataCurrencyCode) {
        if (modificationType == PaymentTestOverwriteValue)
            itemAmount.setCurrencyCode(valueToUse);
    } else {
        itemAmount.setCurrencyCode("USD");
    }
    if (detail == PaymentTestDetailItem && data == PaymentTestDataAmount) {
        if (modificationType == PaymentTestOverwriteValue)
            itemAmount.setValue(valueToUse);
    } else {
        itemAmount.setValue("9.99");
    }

    PaymentItem item;
    item.setAmount(itemAmount);
    if (detail == PaymentTestDetailItem && data == PaymentTestDataId) {
        if (modificationType == PaymentTestOverwriteValue)
            item.setId(valueToUse);
    } else {
        item.setId("total");
    }
    if (detail == PaymentTestDetailItem && data == PaymentTestDataLabel) {
        if (modificationType == PaymentTestOverwriteValue)
            item.setLabel(valueToUse);
    } else {
        item.setLabel("Total charge");
    }

    CurrencyAmount shippingAmount;
    if (detail == PaymentTestDetailShippingOption && data == PaymentTestDataCurrencyCode) {
        if (modificationType == PaymentTestOverwriteValue)
            shippingAmount.setCurrencyCode(valueToUse);
    } else {
        shippingAmount.setCurrencyCode("USD");
    }
    if (detail == PaymentTestDetailShippingOption && data == PaymentTestDataAmount) {
        if (modificationType == PaymentTestOverwriteValue)
            shippingAmount.setValue(valueToUse);
    } else {
        shippingAmount.setValue("9.99");
    }

    ShippingOption shippingOption;
    shippingOption.setAmount(shippingAmount);
    if (detail == PaymentTestDetailShippingOption && data == PaymentTestDataId) {
        if (modificationType == PaymentTestOverwriteValue)
            shippingOption.setId(valueToUse);
    } else {
        shippingOption.setId("standard");
    }
    if (detail == PaymentTestDetailShippingOption && data == PaymentTestDataLabel) {
        if (modificationType == PaymentTestOverwriteValue)
            shippingOption.setLabel(valueToUse);
    } else {
        shippingOption.setLabel("Standard shipping");
    }

    PaymentDetails result;
    result.setItems(HeapVector<PaymentItem>(1, item));
    result.setShippingOptions(HeapVector<ShippingOption>(2, shippingOption));

    return result;
}

} // namespace blink