Support FileDescriptorWatcher in TaskScheduler.

base/task_scheduler/task_tracker_posix_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/task_scheduler/task_tracker_posix.h"
+
+#include <unistd.h>
+
+#include <utility>
+
+#include "base/bind.h"
+#include "base/files/file_descriptor_watcher_posix.h"
+#include "base/memory/ptr_util.h"
+#include "base/message_loop/message_loop.h"
+#include "base/posix/eintr_wrapper.h"
+#include "base/run_loop.h"
+#include "base/sequence_token.h"
+#include "base/task_scheduler/task.h"
+#include "base/task_scheduler/task_traits.h"
+#include "base/time/time.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace base {
+namespace internal {
+
+// Verify that TaskTrackerPosix runs a Task it receives.
+TEST(TaskSchedulerTaskTrackerPosixTest, RunTask) {
+  MessageLoopForIO message_loop;
+  bool did_run = false;
+  auto task = MakeUnique<Task>(
+      FROM_HERE,
+      Bind([](bool* did_run) { *did_run = true; }, Unretained(&did_run)),
+      TaskTraits(), TimeDelta());
+  TaskTrackerPosix tracker(&message_loop);
+
+  EXPECT_TRUE(tracker.WillPostTask(task.get()));
+  EXPECT_TRUE(tracker.RunTask(std::move(task), SequenceToken::Create()));
+  EXPECT_TRUE(did_run);
+}
+
+// Verify that FileDescriptorWatcher::WatchReadable() can be called from a task
+// running in TaskTrackerPosix without a crash.
+TEST(TaskSchedulerTaskTrackerPosixTest, FileDescriptorWatcher) {
+  MessageLoopForIO message_loop;
+  int fds[2];
+  ASSERT_EQ(0, pipe(fds));
+  auto task = MakeUnique<Task>(
+      FROM_HERE, Bind(IgnoreResult(&FileDescriptorWatcher::WatchReadable),
+                      fds[0], Bind(&DoNothing)),
+      TaskTraits(), TimeDelta());
+  TaskTrackerPosix tracker(&message_loop);
+
+  EXPECT_TRUE(tracker.WillPostTask(task.get()));
+  EXPECT_TRUE(tracker.RunTask(std::move(task), SequenceToken::Create()));
+
+  // Run the MessageLoop to allow the read watch to be registered and
+  // unregistered. This prevents a memory leak.
+  RunLoop().RunUntilIdle();
+
+  EXPECT_EQ(0, IGNORE_EINTR(close(fds[0])));
+  EXPECT_EQ(0, IGNORE_EINTR(close(fds[1])));
+}
+
+}  // namespace internal
+}  // namespace base