[Payments] Refactor into PaymentRequestState and Spec

components/payments/content/payment_request_state_unittest.cc

Index: components/payments/content/payment_request_state_unittest.cc
diff --git a/components/payments/content/payment_request_state_unittest.cc b/components/payments/content/payment_request_state_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..2af1af0468ac6ee782686858dca9dab32add59cd
--- /dev/null
+++ b/components/payments/content/payment_request_state_unittest.cc
@@ -0,0 +1,218 @@
+// Copyright 2017 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 "components/payments/content/payment_request_state.h"
+
+#include <utility>
+
+#include "base/memory/weak_ptr.h"
+#include "components/autofill/core/browser/autofill_profile.h"
+#include "components/autofill/core/browser/autofill_test_utils.h"
+#include "components/autofill/core/browser/credit_card.h"
+#include "components/autofill/core/browser/test_personal_data_manager.h"
+#include "components/payments/content/payment_request.mojom.h"
+#include "components/payments/content/payment_request_spec.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace payments {
+
+class PaymentRequestStateTest : public testing::Test,
+                                public PaymentRequestState::Observer,
+                                public PaymentRequestState::Delegate {
+ protected:
+  PaymentRequestStateTest()
+      : num_on_selected_information_changed_called_(0),
+        locale_("en-US"),
+        address_(autofill::test::GetFullProfile()),
+        credit_card_(autofill::test::GetCreditCard()) {
+    test_personal_data_manager_.AddTestingProfile(&address_);
+    credit_card_.set_billing_address_id(address_.guid());
+    test_personal_data_manager_.AddTestingCreditCard(&credit_card_);
+  }
+  ~PaymentRequestStateTest() override {}
+
+  // PaymentRequestState::Observer:
+  void OnSelectedInformationChanged() override {
+    num_on_selected_information_changed_called_++;
+  }
+
+  // PaymentRequestState::Delegate:
+  const std::string& GetApplicationLocale() override { return locale_; };
+  autofill::PersonalDataManager* GetPersonalDataManager() override {
+    return &test_personal_data_manager_;
+  }
+  void OnPaymentResponseAvailable(mojom::PaymentResponsePtr response) override {
+    payment_response_ = std::move(response);
+  };
+
+  void RecreateStateWithOptionsAndDetails(
+      mojom::PaymentOptionsPtr options,
+      mojom::PaymentDetailsPtr details,
+      std::vector<mojom::PaymentMethodDataPtr> method_data) {
+    // The spec will be based on the |options| and |details| passed in.
+    spec_ = base::MakeUnique<PaymentRequestSpec>(
+        std::move(options), std::move(details), std::move(method_data),
+        nullptr);
+    state_ = base::MakeUnique<PaymentRequestState>(spec_.get(), this);
+    state_->AddObserver(this);
+  }
+
+  // Convenience method to create a PaymentRequestState with default details
+  // (one shipping option) and method data (only supports visa).
+  void RecreateStateWithOptions(mojom::PaymentOptionsPtr options) {
+    // Create dummy PaymentDetails with a single shipping option.
+    std::vector<mojom::PaymentShippingOptionPtr> shipping_options;
+    mojom::PaymentShippingOptionPtr option =
+        mojom::PaymentShippingOption::New();
+    option->id = "option:1";
+    shipping_options.push_back(std::move(option));
+    mojom::PaymentDetailsPtr details = mojom::PaymentDetails::New();
+    details->shipping_options = std::move(shipping_options);
+
+    RecreateStateWithOptionsAndDetails(std::move(options), std::move(details),
+                                       GetMethodDataForVisa());
+  }
+
+  // Convenience method that returns MethodData that supports Visa.
+  std::vector<mojom::PaymentMethodDataPtr> GetMethodDataForVisa() {
+    std::vector<mojom::PaymentMethodDataPtr> method_data;
+    mojom::PaymentMethodDataPtr entry = mojom::PaymentMethodData::New();
+    entry->supported_methods.push_back("visa");
+    method_data.push_back(std::move(entry));
+    return method_data;
+  }
+
+  PaymentRequestState* state() { return state_.get(); }
+  const mojom::PaymentResponsePtr& response() { return payment_response_; }
+  int num_on_selected_information_changed_called() {
+    return num_on_selected_information_changed_called_;
+  }
+
+  autofill::AutofillProfile* test_address() { return &address_; }
+  autofill::CreditCard* test_credit_card() { return &credit_card_; }
+
+ private:
+  std::unique_ptr<PaymentRequestState> state_;
+  std::unique_ptr<PaymentRequestSpec> spec_;
+  int num_on_selected_information_changed_called_;
+  std::string locale_;
+  mojom::PaymentResponsePtr payment_response_;
+  autofill::TestPersonalDataManager test_personal_data_manager_;
+
+  // Test data.
+  autofill::AutofillProfile address_;
+  autofill::CreditCard credit_card_;
+};
+
+// Test that the last shipping option is selected.
+TEST_F(PaymentRequestStateTest, ShippingOptionsSelection) {
+  std::vector<mojom::PaymentShippingOptionPtr> shipping_options;
+  mojom::PaymentShippingOptionPtr option = mojom::PaymentShippingOption::New();
+  option->id = "option:1";
+  option->selected = false;
+  shipping_options.push_back(std::move(option));
+  mojom::PaymentShippingOptionPtr option2 = mojom::PaymentShippingOption::New();
+  option2->id = "option:2";
+  option2->selected = true;
+  shipping_options.push_back(std::move(option2));
+  mojom::PaymentDetailsPtr details = mojom::PaymentDetails::New();
+  details->shipping_options = std::move(shipping_options);
+
+  mojom::PaymentOptionsPtr options = mojom::PaymentOptions::New();
+  options->request_shipping = true;
+  RecreateStateWithOptionsAndDetails(std::move(options), std::move(details),
+                                     GetMethodDataForVisa());
+
+  EXPECT_EQ("option:2", state()->selected_shipping_option()->id);
+}
+
+TEST_F(PaymentRequestStateTest, ReadyToPay_DefaultSelections) {
+  mojom::PaymentOptionsPtr options = mojom::PaymentOptions::New();
+  options->request_shipping = true;
+  options->request_payer_name = true;
+  options->request_payer_phone = true;
+  options->request_payer_email = true;
+  RecreateStateWithOptions(std::move(options));
+
+  EXPECT_TRUE(state()->is_ready_to_pay());
+}
+
+// Testing that the card is supported when determining "is ready to pay". In
+// this test the merchant only supports Visa.
+TEST_F(PaymentRequestStateTest, ReadyToPay_SelectUnsupportedCard) {
+  // Default options.
+  RecreateStateWithOptions(mojom::PaymentOptions::New());
+
+  // Ready to pay because the default card is selected and supported.
+  EXPECT_TRUE(state()->is_ready_to_pay());
+
+  autofill::CreditCard amex_card = autofill::test::GetCreditCard2();  // Amex.
+  state()->SetSelectedCreditCard(&amex_card);
+  EXPECT_EQ(1, num_on_selected_information_changed_called());
+
+  // Not ready to pay because the card is not supported.
+  EXPECT_FALSE(state()->is_ready_to_pay());
+
+  // Go back to the Visa card.
+  state()->SetSelectedCreditCard(test_credit_card());  // Visa card.
+  EXPECT_EQ(2, num_on_selected_information_changed_called());
+
+  // Visa card is supported by the merchant.
+  EXPECT_TRUE(state()->is_ready_to_pay());
+}
+
+// Test selecting a contact info profile will make the user ready to pay.
+TEST_F(PaymentRequestStateTest, ReadyToPay_ContactInfo) {
+  mojom::PaymentOptionsPtr options = mojom::PaymentOptions::New();
+  options->request_payer_name = true;
+  options->request_payer_phone = true;
+  options->request_payer_email = true;
+  RecreateStateWithOptions(std::move(options));
+
+  // Ready to pay because of default selections.
+  EXPECT_TRUE(state()->is_ready_to_pay());
+
+  // Unselecting contact profile.
+  state()->SetSelectedContactProfile(nullptr);
+  EXPECT_EQ(1, num_on_selected_information_changed_called());
+
+  EXPECT_FALSE(state()->is_ready_to_pay());
+
+  state()->SetSelectedContactProfile(test_address());
+  EXPECT_EQ(2, num_on_selected_information_changed_called());
+
+  // Ready to pay!
+  EXPECT_TRUE(state()->is_ready_to_pay());
+}
+
+// Test generating a PaymentResponse.
+TEST_F(PaymentRequestStateTest, GeneratePaymentResponse) {
+  // Default options (no shipping, no contact info).
+  RecreateStateWithOptions(mojom::PaymentOptions::New());
+  state()->SetSelectedCreditCard(test_credit_card());
+  EXPECT_EQ(1, num_on_selected_information_changed_called());
+  EXPECT_TRUE(state()->is_ready_to_pay());
+
+  // TODO(mathp): Currently synchronous, when async will need a RunLoop.
+  state()->GeneratePaymentResponse();
+  EXPECT_EQ("basic-card", response()->method_name);
+  EXPECT_EQ(
+      "{\"billingAddress\":"
+      "{\"addressLine\":[\"666 Erebus St.\",\"Apt 8\"],"
+      "\"city\":\"Elysium\","
+      "\"country\":\"US\","
+      "\"organization\":\"Underworld\","
+      "\"phone\":\"16502111111\","
+      "\"postalCode\":\"91111\","
+      "\"recipient\":\"John H. Doe\","
+      "\"region\":\"CA\"},"
+      "\"cardNumber\":\"4111111111111111\","
+      "\"cardSecurityCode\":\"123\","
+      "\"cardholderName\":\"Test User\","
+      "\"expiryMonth\":\"11\","
+      "\"expiryYear\":\"2017\"}",
+      response()->stringified_details);
+}
+
+}  // namespace payments