blob: 590fb8729f1b3358e2e183940a3e82ef66cf978b [file] [log] [blame]
// 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.
#ifndef THIRD_PARTY_BLINK_RENDERER_CORE_WORKERS_WORKER_THREAD_TEST_HELPER_H_
#define THIRD_PARTY_BLINK_RENDERER_CORE_WORKERS_WORKER_THREAD_TEST_HELPER_H_
#include <memory>
#include "base/macros.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "third_party/blink/public/mojom/net/ip_address_space.mojom-blink.h"
#include "third_party/blink/renderer/bindings/core/v8/source_location.h"
#include "third_party/blink/renderer/bindings/core/v8/v8_cache_options.h"
#include "third_party/blink/renderer/bindings/core/v8/v8_gc_controller.h"
#include "third_party/blink/renderer/core/frame/csp/content_security_policy.h"
#include "third_party/blink/renderer/core/frame/settings.h"
#include "third_party/blink/renderer/core/inspector/console_message.h"
#include "third_party/blink/renderer/core/inspector/worker_devtools_params.h"
#include "third_party/blink/renderer/core/script/script.h"
#include "third_party/blink/renderer/core/workers/global_scope_creation_params.h"
#include "third_party/blink/renderer/core/workers/parent_execution_context_task_runners.h"
#include "third_party/blink/renderer/core/workers/worker_backing_thread.h"
#include "third_party/blink/renderer/core/workers/worker_backing_thread_startup_data.h"
#include "third_party/blink/renderer/core/workers/worker_clients.h"
#include "third_party/blink/renderer/core/workers/worker_global_scope.h"
#include "third_party/blink/renderer/core/workers/worker_reporting_proxy.h"
#include "third_party/blink/renderer/core/workers/worker_thread.h"
#include "third_party/blink/renderer/platform/cross_thread_functional.h"
#include "third_party/blink/renderer/platform/heap/handle.h"
#include "third_party/blink/renderer/platform/network/content_security_policy_parsers.h"
#include "third_party/blink/renderer/platform/waitable_event.h"
#include "third_party/blink/renderer/platform/web_thread_supporting_gc.h"
#include "third_party/blink/renderer/platform/weborigin/kurl.h"
#include "third_party/blink/renderer/platform/weborigin/security_origin.h"
#include "third_party/blink/renderer/platform/wtf/forward.h"
#include "third_party/blink/renderer/platform/wtf/time.h"
#include "third_party/blink/renderer/platform/wtf/vector.h"
#include "v8/include/v8.h"
namespace blink {
class FakeWorkerGlobalScope : public WorkerGlobalScope {
public:
FakeWorkerGlobalScope(
std::unique_ptr<GlobalScopeCreationParams> creation_params,
WorkerThread* thread)
: WorkerGlobalScope(std::move(creation_params),
thread,
CurrentTimeTicks()) {}
~FakeWorkerGlobalScope() override = default;
// EventTarget
const AtomicString& InterfaceName() const override {
return event_target_names::kDedicatedWorkerGlobalScope;
}
// WorkerGlobalScope
void ImportModuleScript(
const KURL& module_url_record,
FetchClientSettingsObjectSnapshot* outside_settings_object,
network::mojom::FetchCredentialsMode) override {
NOTREACHED();
}
void ExceptionThrown(ErrorEvent*) override {}
};
class WorkerThreadForTest : public WorkerThread {
public:
explicit WorkerThreadForTest(
WorkerReportingProxy& mock_worker_reporting_proxy)
: WorkerThread(mock_worker_reporting_proxy),
worker_backing_thread_(WorkerBackingThread::Create(
ThreadCreationParams(WebThreadType::kTestThread))) {}
~WorkerThreadForTest() override = default;
WorkerBackingThread& GetWorkerBackingThread() override {
return *worker_backing_thread_;
}
void ClearWorkerBackingThread() override { worker_backing_thread_.reset(); }
void StartWithSourceCode(
const SecurityOrigin* security_origin,
const String& source,
ParentExecutionContextTaskRunners* parent_execution_context_task_runners,
const KURL& script_url = KURL("http://fake.url/"),
WorkerClients* worker_clients = nullptr) {
Vector<CSPHeaderAndType> headers{
{"contentSecurityPolicy", kContentSecurityPolicyHeaderTypeReport}};
auto creation_params = std::make_unique<GlobalScopeCreationParams>(
script_url, mojom::ScriptType::kClassic, "fake user agent",
nullptr /* web_worker_fetch_context */, headers,
network::mojom::ReferrerPolicy::kDefault, security_origin,
false /* starter_secure_context */,
CalculateHttpsState(security_origin), worker_clients,
mojom::IPAddressSpace::kLocal, nullptr,
base::UnguessableToken::Create(),
std::make_unique<WorkerSettings>(Settings::Create().get()),
kV8CacheOptionsDefault, nullptr /* worklet_module_responses_map */);
Start(std::move(creation_params),
WorkerBackingThreadStartupData::CreateDefault(),
std::make_unique<WorkerDevToolsParams>(),
parent_execution_context_task_runners);
EvaluateClassicScript(script_url, source, nullptr /* cached_meta_data */,
v8_inspector::V8StackTraceId());
}
void WaitForInit() {
WaitableEvent completion_event;
GetWorkerBackingThread().BackingThread().PostTask(
FROM_HERE, CrossThreadBind(&WaitableEvent::Signal,
CrossThreadUnretained(&completion_event)));
completion_event.Wait();
}
protected:
WorkerOrWorkletGlobalScope* CreateWorkerGlobalScope(
std::unique_ptr<GlobalScopeCreationParams> creation_params) override {
return MakeGarbageCollected<FakeWorkerGlobalScope>(
std::move(creation_params), this);
}
private:
WebThreadType GetThreadType() const override {
return WebThreadType::kUnspecifiedWorkerThread;
}
std::unique_ptr<WorkerBackingThread> worker_backing_thread_;
};
} // namespace blink
#endif // THIRD_PARTY_BLINK_RENDERER_CORE_WORKERS_WORKER_THREAD_TEST_HELPER_H_