Add support to update_engine to poke Omaha after an update has been applied
successfully and is awaiting reboot to help ensure the number of actives
remains accurate.

BUG=chromium-os:12026
TEST=Manual test, unit tests

Change-Id: Ie3397264b0b34e8d423fb9748970f7d330122180

Review URL: http://codereview.chromium.org/6836025
diff --git a/update_check_scheduler.cc b/update_check_scheduler.cc
index 16f1929..f9b45bd 100644
--- a/update_check_scheduler.cc
+++ b/update_check_scheduler.cc
@@ -137,7 +137,11 @@
 }
 
 void UpdateCheckScheduler::SetUpdateStatus(UpdateStatus status) {
-  if (status == UPDATE_STATUS_IDLE) {
+  // We want to schedule the update checks for when we're idle as well as
+  // after we've successfully applied an update and waiting for the user
+  // to reboot to ensure our active count is accurate.
+  if (status == UPDATE_STATUS_IDLE ||
+      status == UPDATE_STATUS_UPDATED_NEED_REBOOT) {
     ScheduleNextCheck();
   }
 }