chrome/browser/net/storage_test_utils.cc

// Copyright (c) 2020 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 "chrome/browser/net/storage_test_utils.h"

#include "content/public/test/browser_test_utils.h"

namespace storage {
namespace test {

const std::vector<std::string> kStorageTypes{
    "Cookie", "LocalStorage", "FileSystem",    "SessionStorage", "IndexedDb",
    "WebSql", "CacheStorage", "ServiceWorker", "CookieStore"};

const std::vector<std::string> kCrossTabCommunicationTypes{
    "SharedWorker",
    "WebLock",
};

constexpr char kRequestStorageAccess[] =
    "document.requestStorageAccess().then("
    "  () => { window.domAutomationController.send(true); },"
    "  () => { window.domAutomationController.send(false); },"
    ");";

constexpr char kHasStorageAccess[] =
    "document.hasStorageAccess().then("
    "  (result) => { window.domAutomationController.send(result); },"
    "  () => { window.domAutomationController.send(false); },"
    ");";

void ExpectFrameContent(content::RenderFrameHost* frame,
                        const std::string& expected) {
  std::string content;
  ASSERT_TRUE(ExecuteScriptAndExtractString(
      frame, "window.domAutomationController.send(document.body.textContent)",
      &content));
  EXPECT_EQ(expected, content);
}

void ExpectCookiesOnHost(content::BrowserContext* context,
                         const GURL& host_url,
                         const std::string& expected) {
  EXPECT_EQ(expected, content::GetCookies(context, host_url));
}

void SetStorageForFrame(content::RenderFrameHost* frame) {
  for (const auto& data_type : kStorageTypes) {
    bool data = false;
    EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
        frame, "set" + data_type + "()", &data));
    EXPECT_TRUE(data) << data_type;
  }
}

void ExpectStorageForFrame(content::RenderFrameHost* frame, bool expected) {
  for (const auto& data_type : kStorageTypes) {
    bool data = false;
    EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
        frame, "has" + data_type + "();", &data));
    EXPECT_EQ(expected, data) << data_type;
  }
}

void SetCrossTabInfoForFrame(content::RenderFrameHost* frame) {
  for (const auto& data_type : kCrossTabCommunicationTypes) {
    bool data = false;
    EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
        frame, "set" + data_type + "()", &data));
    EXPECT_TRUE(data) << data_type;
  }
}

void ExpectCrossTabInfoForFrame(content::RenderFrameHost* frame,
                                bool expected) {
  for (const auto& data_type : kCrossTabCommunicationTypes) {
    bool data = false;
    EXPECT_TRUE(content::ExecuteScriptAndExtractBool(
        frame, "has" + data_type + "();", &data));
    EXPECT_EQ(expected, data) << data_type;
  }
}

void RequestStorageAccessForFrame(content::RenderFrameHost* frame,
                                  bool should_resolve) {
  bool data = false;
  EXPECT_TRUE(content::ExecuteScriptAndExtractBool(frame, kRequestStorageAccess,
                                                   &data));
  EXPECT_EQ(should_resolve, data) << "document.requestStorageAccess()";
}

void CheckStorageAccessForFrame(content::RenderFrameHost* frame,
                                bool access_expected) {
  bool data = false;
  EXPECT_TRUE(
      content::ExecuteScriptAndExtractBool(frame, kHasStorageAccess, &data));
  EXPECT_EQ(access_expected, data) << "document.hasStorageAccess()";
}

}  // namespace test
}  // namespace storage