PaymentHandler: Implement GetAllPaymentApps().

content/browser/payments/payment_app_provider_impl_unittest.cc

 // 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 <cstddef>
 #include <string>
 #include <utility>
 
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "components/payments/mojom/payment_app.mojom.h"
 #include "content/browser/payments/payment_app_content_unittest_base.h"
 #include "content/browser/payments/payment_app_provider_impl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 using payments::mojom::PaymentAppManifestError;
 using payments::mojom::PaymentAppManifestPtr;
 
 namespace content {
 
 class PaymentManager;
 
 namespace {
 
+using ::payments::mojom::PaymentHandlerStatus;
+using ::payments::mojom::PaymentInstrumentPtr;
+
 void SetManifestCallback(bool* called,
                          PaymentAppManifestError* out_error,
                          PaymentAppManifestError error) {
   *called = true;
   *out_error = error;
 }
 
 void GetAllManifestsCallback(bool* called,
                              PaymentAppProvider::Manifests* out_manifests,
                              PaymentAppProvider::Manifests manifests) {
   *called = true;
   *out_manifests = std::move(manifests);
 }
 
+void SetPaymentInstrumentCallback(PaymentHandlerStatus* out_status,
+                                  PaymentHandlerStatus status) {
+  *out_status = status;
+}
+
+void GetAllPaymentAppsCallback(PaymentAppProvider::PaymentApps* out_apps,
+                               PaymentAppProvider::PaymentApps apps) {
+  *out_apps = std::move(apps);
+}
+
 void InvokePaymentAppCallback(bool* called,
                               payments::mojom::PaymentAppResponsePtr response) {
   *called = true;
 }
 
 }  // namespace
 
 class PaymentAppProviderTest : public PaymentAppContentUnitTestBase {
  public:
   PaymentAppProviderTest() {}
   ~PaymentAppProviderTest() override {}
 
   void GetAllManifests(PaymentAppProvider::GetAllManifestsCallback callback) {
     PaymentAppProviderImpl::GetInstance()->GetAllManifests(browser_context(),
                                                            callback);
     base::RunLoop().RunUntilIdle();
   }
 
+  void SetPaymentInstrument(
+      PaymentManager* manager,
+      const std::string& instrument_key,
+      PaymentInstrumentPtr instrument,
+      PaymentManager::SetPaymentInstrumentCallback callback) {
+    ASSERT_NE(nullptr, manager);
+    manager->SetPaymentInstrument(instrument_key, std::move(instrument),
+                                  std::move(callback));
+    base::RunLoop().RunUntilIdle();
+  }
+
+  void GetAllPaymentApps(
+      PaymentAppProvider::GetAllPaymentAppsCallback callback) {
+    PaymentAppProviderImpl::GetInstance()->GetAllPaymentApps(
+        browser_context(), std::move(callback));
+    base::RunLoop().RunUntilIdle();
+  }
+
   void InvokePaymentApp(int64_t registration_id,
                         payments::mojom::PaymentAppRequestPtr app_request,
                         PaymentAppProvider::InvokePaymentAppCallback callback) {
     PaymentAppProviderImpl::GetInstance()->InvokePaymentApp(
         browser_context(), registration_id, std::move(app_request), callback);
     base::RunLoop().RunUntilIdle();
   }
 
   void CreatePaymentApp(const GURL& scope_url, const GURL& sw_script_url) {
     PaymentManager* manager = CreatePaymentManager(scope_url, sw_script_url);
@@ skipping 73 matching lines @@
 
   called = false;
   InvokePaymentApp(manifests[1].first, std::move(app_request),
                    base::Bind(&InvokePaymentAppCallback, &called));
   ASSERT_TRUE(called);
 
   EXPECT_EQ(manifests[1].first, last_sw_registration_id());
   EXPECT_EQ(GURL(kPaymentAppInfo[1].scopeUrl), last_sw_scope_url());
 }
 
+TEST_F(PaymentAppProviderTest, GetAllPaymentAppsTest) {
+  PaymentManager* manager1 = CreatePaymentManager(
+      GURL("https://hellopay.com/a"), GURL("https://hellopay.com/a/script.js"));
+  PaymentManager* manager2 = CreatePaymentManager(
+      GURL("https://bobpay.com/b"), GURL("https://bobpay.com/b/script.js"));
+
+  PaymentHandlerStatus status;
+  SetPaymentInstrument(manager1, "test_key1",
+                       payments::mojom::PaymentInstrument::New(),
+                       base::Bind(&SetPaymentInstrumentCallback, &status));
+  SetPaymentInstrument(manager2, "test_key2",
+                       payments::mojom::PaymentInstrument::New(),
+                       base::Bind(&SetPaymentInstrumentCallback, &status));
+  SetPaymentInstrument(manager2, "test_key3",
+                       payments::mojom::PaymentInstrument::New(),
+                       base::Bind(&SetPaymentInstrumentCallback, &status));
+
+  PaymentAppProvider::PaymentApps apps;
+  GetAllPaymentApps(base::Bind(&GetAllPaymentAppsCallback, &apps));
+
+  ASSERT_EQ(2U, apps.size());
+  ASSERT_EQ(1U, apps[GURL("https://hellopay.com/")].size());
+  ASSERT_EQ(2U, apps[GURL("https://bobpay.com/")].size());
+}
+
 }  // namespace content