Add Support to Get a COM STA Task Runner from the Task Scheduler API

base/test/scoped_task_scheduler_unittest.cc

 // 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.
 
 #include "base/test/scoped_task_scheduler.h"
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "base/macros.h"
 #include "base/run_loop.h"
 #include "base/sequence_checker.h"
 #include "base/task_scheduler/post_task.h"
 #include "base/task_scheduler/task_scheduler.h"
 #include "base/task_scheduler/test_utils.h"
 #include "base/test/scoped_mock_time_message_loop_task_runner.h"
 #include "base/threading/sequenced_task_runner_handle.h"
 #include "base/threading/thread_checker.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "base/time/time.h"
 #include "build/build_config.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
+#if defined(OS_WIN)
+#include <objbase.h>
+#endif  // defined(OS_WIN)
+
 namespace base {
 namespace test {
 
 TEST(ScopedTaskSchedulerTest, PostTask) {
   ScopedTaskScheduler scoped_task_scheduler;
 
   bool first_task_ran = false;
   bool second_task_ran = false;
 
   SequenceCheckerImpl sequence_checker;
@@ skipping 184 matching lines @@
           },
           Unretained(&sequence_checker), Unretained(&thread_checker),
           Unretained(&second_task_ran)));
 
   RunLoop().RunUntilIdle();
 
   EXPECT_TRUE(first_task_ran);
   EXPECT_TRUE(second_task_ran);
 }
 
+#if defined(OS_WIN)
+// Verify that COM STAs work correctly from the ScopedTaskScheduler.
+TEST(ScopedTaskSchedulerTest, COMSTAAvailable) {
+  ScopedTaskScheduler scoped_task_scheduler;
+  auto com_task_runner = CreateCOMSTATaskRunnerWithTraits(TaskTraits());
+
+  bool com_task_ran = false;
+  com_task_runner->PostTask(
+      FROM_HERE,
+      Bind(
+          [](bool* com_task_ran) {
+            *com_task_ran = true;
+            HRESULT hr = CoInitializeEx(nullptr, COINIT_MULTITHREADED);
+            if (SUCCEEDED(hr)) {
+              ADD_FAILURE() << "COM STA was not initialized on this thread";
+              CoUninitialize();
+            }
+          },
+          &com_task_ran));
+
+  RunLoop().RunUntilIdle();
+
+  EXPECT_TRUE(com_task_ran);
+}
+#endif  // defined(OS_WIN)
+
 TEST(ScopedTaskSchedulerTest, NonBlockShutdownTasksPostedAfterShutdownDontRun) {
   ScopedTaskScheduler scoped_task_scheduler;
   TaskScheduler::GetInstance()->Shutdown();
   PostTaskWithTraits(FROM_HERE, TaskTraits().WithShutdownBehavior(
                                     TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN),
                      Bind([]() {
                        ADD_FAILURE()
                            << "CONTINUE_ON_SHUTDOWN task should not run";
                      }));
   PostTaskWithTraits(
@@ skipping 71 matching lines @@
                                 Bind([](bool* task_ran) { *task_ran = true; },
                                      Unretained(task_ran)));
                           },
                           Unretained(&task_ran)));
   RunLoop().RunUntilIdle();
   EXPECT_TRUE(task_ran);
 }
 
 }  // namespace test
 }  // namespace base