// Copyright 2021 The Chromium Authors
// 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_web_contents_manager.h"

#include "base/memory/raw_ptr.h"
#include "components/autofill/core/browser/data_manager/test_personal_data_manager.h"
#include "components/payments/content/payment_request.h"
#include "components/payments/content/test_content_payment_request_delegate.h"
#include "content/public/browser/web_contents.h"
#include "content/public/test/browser_task_environment.h"
#include "content/public/test/navigation_simulator.h"
#include "content/public/test/test_browser_context.h"
#include "content/public/test/test_web_contents_factory.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace payments {

class PaymentRequestWebContentsManagerTest : public testing::Test {
 public:
  PaymentRequestWebContentsManagerTest()
      : web_contents_(web_contents_factory_.CreateWebContents(&context_)) {
    manager_ = PaymentRequestWebContentsManager::GetOrCreateForWebContents(
        web_contents_);
  }

  ~PaymentRequestWebContentsManagerTest() override { manager_ = nullptr; }

  content::WebContents* web_contents() { return web_contents_; }

  PaymentRequest* CreateAndReturnPaymentRequest(SPCTransactionMode mode) {
    manager_->SetSPCTransactionMode(mode);

    std::unique_ptr<TestContentPaymentRequestDelegate> delegate =
        std::make_unique<TestContentPaymentRequestDelegate>(
            /*task_executor=*/nullptr, &test_personal_data_manager_);
    delegate->set_frame_routing_id(
        web_contents()->GetPrimaryMainFrame()->GetGlobalId());

    mojo::PendingRemote<payments::mojom::PaymentRequest> remote;
    mojo::PendingReceiver<payments::mojom::PaymentRequest> receiver =
        remote.InitWithNewPipeAndPassReceiver();

    // PaymentRequest is a DocumentService, whose lifetime is managed by the
    // RenderFrameHost passed into the delegate.
    return new PaymentRequest(std::move(delegate), std::move(receiver));
  }

  // The PaymentRequestWebContentsManager under test.
  raw_ptr<PaymentRequestWebContentsManager> manager_;

 private:
  content::BrowserTaskEnvironment task_environment_;
  content::TestBrowserContext context_;
  autofill::TestPersonalDataManager test_personal_data_manager_;
  content::TestWebContentsFactory web_contents_factory_;
  raw_ptr<content::WebContents> web_contents_;
};

TEST_F(PaymentRequestWebContentsManagerTest, SPCTransactionMode) {
  // The mode given to the PaymentRequest is exposed on its API.
  PaymentRequest* request1 =
      CreateAndReturnPaymentRequest(SPCTransactionMode::kNone);
  ASSERT_EQ(request1->spc_transaction_mode(), SPCTransactionMode::kNone);
  PaymentRequest* request2 =
      CreateAndReturnPaymentRequest(SPCTransactionMode::kAutoAccept);
  ASSERT_EQ(request2->spc_transaction_mode(), SPCTransactionMode::kAutoAccept);
  PaymentRequest* request3 =
      CreateAndReturnPaymentRequest(SPCTransactionMode::kAutoReject);
  ASSERT_EQ(request3->spc_transaction_mode(), SPCTransactionMode::kAutoReject);

  // Check that already-created PaymentRequests were not altered.
  ASSERT_EQ(request1->spc_transaction_mode(), SPCTransactionMode::kNone);
  ASSERT_EQ(request2->spc_transaction_mode(), SPCTransactionMode::kAutoAccept);
}

TEST_F(PaymentRequestWebContentsManagerTest, HadActivationlessShow) {
  ASSERT_FALSE(manager_->HadActivationlessShow());
  manager_->RecordActivationlessShow();
  ASSERT_TRUE(manager_->HadActivationlessShow());

  // A renderer initiated navigation without a user activation should not
  // reset the activationless show state.
  {
    auto navigation_simulator =
        content::NavigationSimulator::CreateRendererInitiated(
            GURL("http://example1.test"),
            web_contents()->GetPrimaryMainFrame());
    navigation_simulator->SetHasUserGesture(false);
    navigation_simulator->Start();
    navigation_simulator->Commit();
    ASSERT_TRUE(manager_->HadActivationlessShow());
  }

  // A renderer initiated navigation with a user activation should reset the
  // activationless show state.
  {
    auto navigation_simulator =
        content::NavigationSimulator::CreateRendererInitiated(
            GURL("http://example2.test"),
            web_contents()->GetPrimaryMainFrame());
    navigation_simulator->SetHasUserGesture(true);
    navigation_simulator->Start();
    navigation_simulator->Commit();
    ASSERT_FALSE(manager_->HadActivationlessShow());
  }

  manager_->RecordActivationlessShow();
  ASSERT_TRUE(manager_->HadActivationlessShow());

  // A browser reload should not reset the activationless show state.
  {
    auto navigation_simulator =
        content::NavigationSimulator::CreateBrowserInitiated(
            GURL("http://example2.test"), web_contents());
    navigation_simulator->Start();
    navigation_simulator->Commit();
    ASSERT_TRUE(manager_->HadActivationlessShow());
  }

  // A browser initiated navigation should reset the activationless show state.
  {
    auto navigation_simulator =
        content::NavigationSimulator::CreateBrowserInitiated(
            GURL("http://example3.test"), web_contents());
    navigation_simulator->Start();
    navigation_simulator->Commit();
    ASSERT_FALSE(manager_->HadActivationlessShow());
  }
}

}  // namespace payments