Add support to update_engine to poke Omaha after an update has been applied

update_check_scheduler.cc

Index: update_check_scheduler.cc
diff --git a/update_check_scheduler.cc b/update_check_scheduler.cc
index 16f1929c2d65423cd68e3d3ca89195f5f7a1ad87..f9b45bd0435a73d5cfbbf7d7dbd5670ae3686538 100644
--- a/update_check_scheduler.cc
+++ b/update_check_scheduler.cc
@@ -137,7 +137,11 @@ void UpdateCheckScheduler::ScheduleNextCheck() {
 }
 
 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();
   }
 }