boarding_pass_extractor_browsertest.cc

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

#include "chrome/renderer/wallet/boarding_pass_extractor.h"

#include "chrome/test/base/chrome_render_view_test.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace wallet {

class BoardingPassExtractorBrowserTest : public ChromeRenderViewTest {
 public:
};

TEST_F(BoardingPassExtractorBrowserTest, ExtractBoardingPassWithScript) {
  auto extractor_ = std::make_unique<BoardingPassExtractor>(
      GetMainRenderFrame(), registry_.get());

  const char* html = R"HTML(
      <html>
        <head>
          <title>Google Boarding Pass Page</title>
        </head>
        <body>
        </body>
      </html>
    )HTML";
  LoadHTML(html);

  const std::string script = R"###(
    results = ['M1TEST', 'M2TEST']
  )###";

  extractor_->ExtractBoardingPassWithScript(
      script,
      base::BindOnce([](const std::vector<std::string>& boarding_passes) {
        EXPECT_EQ(boarding_passes.size(), 2u);
        EXPECT_EQ(boarding_passes[0], "M1TEST");
        EXPECT_EQ(boarding_passes[1], "M2TEST");
      }));
  base::RunLoop().RunUntilIdle();
}

TEST_F(BoardingPassExtractorBrowserTest, ExtractBoardingPassTitleNotMatch) {
  auto extractor_ = std::make_unique<BoardingPassExtractor>(
      GetMainRenderFrame(), registry_.get());

  const char* html = R"HTML(
      <html>
        <head>
          <title>Google Testing Page</title>
        </head>
        <body>
        </body>
      </html>
    )HTML";
  LoadHTML(html);

  extractor_->ExtractBoardingPass(
      base::BindOnce([](const std::vector<std::string>& boarding_passes) {
        EXPECT_EQ(boarding_passes.size(), 0u);
      }));
  base::RunLoop().RunUntilIdle();
}

}  // namespace wallet
chromium/chrome/renderer/wallet/boarding_pass_extractor_browsertest.cc
