Make Hands-Off Zero-Touch Enrollment compatibile with tests

chrome/browser/chromeos/login/screens/update_screen.h

Index: chrome/browser/chromeos/login/screens/update_screen.h
diff --git a/chrome/browser/chromeos/login/screens/update_screen.h b/chrome/browser/chromeos/login/screens/update_screen.h
index 2c60ba4ea9237bda0af6bc0d8f7464c57479027b..a03787dea330ebf3971d040ebea52c881c5a5fa2 100644
--- a/chrome/browser/chromeos/login/screens/update_screen.h
+++ b/chrome/browser/chromeos/login/screens/update_screen.h
@@ -79,6 +79,7 @@ class UpdateScreen : public BaseScreen,
   FRIEND_TEST_ALL_PREFIXES(UpdateScreenTest, TestBasic);
   FRIEND_TEST_ALL_PREFIXES(UpdateScreenTest, TestUpdateAvailable);
   FRIEND_TEST_ALL_PREFIXES(UpdateScreenTest, TestAPReselection);
+  friend class UpdateScreenUnitTest;
 
   enum class State {
     STATE_IDLE = 0,
@@ -125,6 +126,18 @@ class UpdateScreen : public BaseScreen,
   // The user requested an attempt to connect to the network should be made.
   void OnConnectRequested();
 
+  // Records the fact that we performed an update check but do not yet
+  // know if this update check is going to result in an update.
+  void RecordUpdateCheckWithNoUpdateYet();
+
+  // When an update is found, this is called to clear the time of update check
+  // which had saved when we did not yet know whether it would result in an
+  // update.
+  void ClearUpdateCheckNoUpdateTime();
+
+  // Returns true if we should check for an update.
+  bool ShouldCheckForUpdate();
+
   // Timer for the interval to wait for the reboot.
   // If reboot didn't happen - ask user to reboot manually.
   base::OneShotTimer reboot_timer_;