Fixed channel switch when user decides to switch to another channel in

chrome/browser/ui/webui/help/version_updater_chromeos.h

Index: chrome/browser/ui/webui/help/version_updater_chromeos.h
diff --git a/chrome/browser/ui/webui/help/version_updater_chromeos.h b/chrome/browser/ui/webui/help/version_updater_chromeos.h
index bfdee79df091d2ee082631bfd251bbf22c2a8099..ae4b9dfa87b5fd5ff129e91dd0befbac550e7552 100644
--- a/chrome/browser/ui/webui/help/version_updater_chromeos.h
+++ b/chrome/browser/ui/webui/help/version_updater_chromeos.h
@@ -24,6 +24,10 @@ class VersionUpdaterCros : public VersionUpdater,
   // Gets the last update status, without triggering a new check or download.
   void GetUpdateStatus(const StatusCallback& callback);
 
+  void set_can_update_for_testing(bool can_update_for_testing) {
+    can_update_for_testing_ = can_update_for_testing;
+  }
+
  protected:
   friend class VersionUpdater;
 
@@ -39,12 +43,23 @@ class VersionUpdaterCros : public VersionUpdater,
   // Callback from UpdateEngineClient::RequestUpdateCheck().
   void OnUpdateCheck(chromeos::UpdateEngineClient::UpdateCheckResult result);
 
+  // Returns whether an update is allowed. If not, it calls the callback with
+  // the appropriate status.
+  bool EnsureCanUpdate(const StatusCallback& callback);
+
   // Callback used to communicate update status to the client.
   StatusCallback callback_;
 
   // Last state received via UpdateStatusChanged().
   chromeos::UpdateEngineClient::UpdateStatusOperation last_operation_;
 
+  // True if an update check should be scheduled when the update engine will

[James Hawkins, 2014/09/22 16:48:31]

Optional nit: This is easier to grok for me if you use the phrasing, "when the
update engine is idle."

[ygorshenin1, 2014/09/22 19:01:54]

Done.

+  // be idle.
+  bool check_for_update_when_idle_;
+
+  // True if an update check can be performed in tests.
+  bool can_update_for_testing_;

[James Hawkins, 2014/09/22 16:48:31]

I took a high-level look at what you're trying to accomplish, and I think,
barring other design changes, the most appropriate thing for you to do is to use
object mocking to mock out the CanUpdate method to return true under test.

A better design would be to create an interface with a method on it that
performs this check.  Then you'd implement the real implementation in the
version_updater_chromeos.cc file and have a test implementation in the
_unittest.cc file.

[ygorshenin1, 2014/09/22 19:01:54]

I've made a virtual EnsureCanUpdate() method and overridden it in the test code.

On 2014/09/22 16:48:31, James Hawkins wrote:

+
   base::WeakPtrFactory<VersionUpdaterCros> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(VersionUpdaterCros);