0
Files
src/content/test/storage_partition_test_helpers.cc
Avi Drissman 4e1b7bc33d Update copyright headers in content/
The methodology used to generate this CL is documented in
https://crbug.com/1098010#c34.

No-Try: true
No-Presubmit: true
Bug: 1098010
Change-Id: I8c0f009d16350271f07d8e5e561085822cc9dd27
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/3895935
Owners-Override: Avi Drissman <avi@chromium.org>
Reviewed-by: Mark Mentovai <mark@chromium.org>
Commit-Queue: Mark Mentovai <mark@chromium.org>
Auto-Submit: Avi Drissman <avi@chromium.org>
Cr-Commit-Position: refs/heads/main@{#1047456}
2022-09-15 14:03:50 +00:00

57 lines
2.0 KiB
C++

// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/test/storage_partition_test_helpers.h"
#include "content/browser/storage_partition_impl.h"
#include "content/browser/worker_host/shared_worker_service_impl.h"
#include "content/browser/worker_host/test_shared_worker_service_impl.h"
#include "content/public/browser/storage_partition_config.h"
namespace content {
void InjectTestSharedWorkerService(StoragePartition* storage_partition) {
auto* storage_partition_impl =
static_cast<StoragePartitionImpl*>(storage_partition);
storage_partition_impl->OverrideSharedWorkerServiceForTesting(
std::make_unique<TestSharedWorkerServiceImpl>(
storage_partition_impl,
storage_partition_impl->GetServiceWorkerContext()));
}
void TerminateAllSharedWorkers(StoragePartition* storage_partition,
base::OnceClosure callback) {
static_cast<TestSharedWorkerServiceImpl*>(
storage_partition->GetSharedWorkerService())
->TerminateAllWorkers(std::move(callback));
}
StoragePartitionConfig CreateStoragePartitionConfigForTesting(
bool in_memory,
const std::string& partition_domain,
const std::string& partition_name) {
return StoragePartitionConfig(partition_domain, partition_name, in_memory);
}
CustomStoragePartitionForSomeSites::CustomStoragePartitionForSomeSites(
const GURL& site_to_isolate)
: site_to_isolate_(site_to_isolate) {}
StoragePartitionConfig
CustomStoragePartitionForSomeSites::GetStoragePartitionConfigForSite(
BrowserContext* browser_context,
const GURL& site) {
// Override for |site_to_isolate_|.
if (site == site_to_isolate_) {
return StoragePartitionConfig::Create(
browser_context, "blah_isolated_storage", "blah_isolated_storage",
false /* in_memory */);
}
return StoragePartitionConfig::CreateDefault(browser_context);
}
} // namespace content