TaskScheduler [5/9] Task Tracker

base/task_scheduler/task_traits.cc

Index: base/task_scheduler/task_traits.cc
diff --git a/base/task_scheduler/task_traits.cc b/base/task_scheduler/task_traits.cc
index 9e5be327383c20b760b7034e9240c3656dae64e3..5a0906402ecb251a2a8f3d83f00ae599c64d5bb5 100644
--- a/base/task_scheduler/task_traits.cc
+++ b/base/task_scheduler/task_traits.cc
@@ -34,18 +34,35 @@ TaskTraits& TaskTraits::WithShutdownBehavior(
   return *this;
 }
 
-void PrintTo(const TaskPriority& task_priority, std::ostream* os) {
+std::ostream& operator<<(std::ostream& os, const TaskPriority& task_priority) {
   switch (task_priority) {
     case TaskPriority::BACKGROUND:
-      *os << "BACKGROUND";
+      os << "BACKGROUND";
       break;
     case TaskPriority::USER_VISIBLE:
-      *os << "USER_VISIBLE";
+      os << "USER_VISIBLE";
       break;
     case TaskPriority::USER_BLOCKING:
-      *os << "USER_BLOCKING";
+      os << "USER_BLOCKING";
       break;
   }
+  return os;
+}
+
+std::ostream& operator<<(std::ostream& os,
+                         const TaskShutdownBehavior& shutdown_behavior) {
+  switch (shutdown_behavior) {
+    case TaskShutdownBehavior::CONTINUE_ON_SHUTDOWN:
+      os << "CONTINUE_ON_SHUTDOWN";
+      break;
+    case TaskShutdownBehavior::SKIP_ON_SHUTDOWN:
+      os << "SKIP_ON_SHUTDOWN";
+      break;
+    case TaskShutdownBehavior::BLOCK_SHUTDOWN:
+      os << "BLOCK_SHUTDOWN";
+      break;
+  }
+  return os;
 }
 
 }  // namespace base