web-threads: Create a single instance of WebThread for each blink thread.

content/test/test_blink_web_unit_test_support.cc

 // Copyright 2013 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 "content/test/test_blink_web_unit_test_support.h"
 
 #include "base/command_line.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/path_service.h"
 #include "base/strings/utf_string_conversions.h"
+#include "base/thread_task_runner_handle.h"
+#include "content/child/webthread_impl.h"
 #include "content/public/common/content_switches.h"
 #include "content/renderer/scheduler/renderer_scheduler.h"
 #include "content/renderer/scheduler/web_scheduler_impl.h"
 #include "content/test/mock_webclipboard_impl.h"
 #include "content/test/web_gesture_curve_mock.h"
 #include "content/test/web_layer_tree_view_impl_for_testing.h"
 #include "content/test/weburl_loader_mock_factory.h"
 #include "media/base/media.h"
 #include "net/cookies/cookie_monster.h"
 #include "net/test/spawned_test_server/spawned_test_server.h"
@@ skipping 15 matching lines @@
 #include "base/mac/foundation_util.h"
 #include "base/mac/scoped_nsautorelease_pool.h"
 #endif
 
 #ifdef V8_USE_EXTERNAL_STARTUP_DATA
 #include "gin/public/isolate_holder.h"
 #endif
 
 namespace content {
 
+namespace {
+
+class WebThreadImplForTest : public WebThreadBase {
+ public:
+  WebThreadImplForTest()
+      : owning_thread_task_runner_(base::ThreadTaskRunnerHandle::Get()),
+        thread_id_(base::PlatformThread::CurrentId()) {
+  }
+
+  ~WebThreadImplForTest() override {
+  }
+
+  // blink::WebThread:
+  blink::PlatformThreadId threadId() const override {
+    return thread_id_;
+  }
+
+  // WebThreadBase:
+  base::MessageLoop* MessageLoop() const override {
+    DCHECK(isCurrentThread());
+    return base::MessageLoop::current();
+  }
+
+ private:
+  base::SingleThreadTaskRunner* TaskRunner() const override {
+    return owning_thread_task_runner_.get();
+  }
+
+  scoped_refptr<base::SingleThreadTaskRunner> owning_thread_task_runner_;
+  blink::PlatformThreadId thread_id_;
+};
+
+}  // namespace
+
 TestBlinkWebUnitTestSupport::TestBlinkWebUnitTestSupport() {
 #if defined(OS_MACOSX)
   base::mac::ScopedNSAutoreleasePool autorelease_pool;
 #endif
 
   url_loader_factory_.reset(new WebURLLoaderMockFactory());
   mock_clipboard_.reset(new MockWebClipboardImpl());
 
 #ifdef V8_USE_EXTERNAL_STARTUP_DATA
   gin::IsolateHolder::LoadV8Snapshot();
 #endif
 
   if (base::MessageLoopProxy::current()) {
     renderer_scheduler_ = RendererScheduler::Create();
     web_scheduler_.reset(new WebSchedulerImpl(renderer_scheduler_.get()));
+    web_thread_.reset(new WebThreadImplForTest());

[Sami, 2015/02/27 17:23:08]

Since we're using the real scheduler, it seems like we should also use
WebThreadImplForScheduler here.

[sadrul, 2015/03/01 10:21:38]

Sounds right. Done.

   }
 
   blink::initialize(this);
   blink::setLayoutTestMode(true);
   blink::WebSecurityPolicy::registerURLSchemeAsLocal(
       blink::WebString::fromUTF8("test-shell-resource"));
   blink::WebSecurityPolicy::registerURLSchemeAsNoAccess(
       blink::WebString::fromUTF8("test-shell-resource"));
   blink::WebSecurityPolicy::registerURLSchemeAsDisplayIsolated(
       blink::WebString::fromUTF8("test-shell-resource"));
@@ skipping 239 matching lines @@
 bool TestBlinkWebUnitTestSupport::getBlobItems(
     const blink::WebString& uuid,
     blink::WebVector<blink::WebBlobData::Item*>* items) {
   return blob_registry_.GetBlobItems(uuid, items);
 }
 
 blink::WebScheduler* TestBlinkWebUnitTestSupport::scheduler() {
   return web_scheduler_.get();
 }
 
+blink::WebThread* TestBlinkWebUnitTestSupport::currentThread() {
+  if (web_thread_ && web_thread_->isCurrentThread())
+    return web_thread_.get();
+  return BlinkPlatformImpl::currentThread();
+}
+
 }  // namespace content