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..a5e8f79f8b4d69c0d793a76e42f01cd34e29cacb 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,17 @@ 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();
+
+  // Records the fact that we found an update through the update check
+  // we just did.
+  void RecordUpdateCheckResultingInUpdate();
+
+  // 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_;