kiosk: Support update url for enterprise.

chrome/browser/chromeos/app_mode/fake_cws.h

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_CHROMEOS_APP_MODE_FAKE_CWS_H_
 #define CHROME_BROWSER_CHROMEOS_APP_MODE_FAKE_CWS_H_
 
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "net/test/embedded_test_server/http_request.h"
 #include "net/test/embedded_test_server/http_response.h"
 #include "url/gurl.h"
 
 namespace chromeos {
 
+// Simple fake CWS update check request handler that returns a fixed update
+// check response. The response is created either from SetUpdateCrx() or
+// SetNoUpdate().
 class FakeCWS {
  public:
-  FakeCWS() {}
-  virtual ~FakeCWS() {}
+  FakeCWS();
+  ~FakeCWS();
 
+  // Initializes as CWS request handler and overrides app gallery command line
+  // switches.
   void Init(net::test_server::EmbeddedTestServer* embedded_test_server);
-  // Sets up the kiosk app update response.
+
+  // Initializes as a private store handler using the given server and URL end
+  // point. Private store does not override app gallery command lines and use a
+  // slightly different template (as documented on
+  // https://developer.chrome.com/extensions/autoupdate).
+  void InitAsPrivateStore(
+      net::test_server::EmbeddedTestServer* embedded_test_server,
+      const std::string& update_check_end_point);
+
+  // Sets up the update check response with has_update template.
   void SetUpdateCrx(const std::string& app_id,
                     const std::string& crx_file,
                     const std::string& version);
+
+  // Sets up the update check response with no_update template.
   void SetNoUpdate(const std::string& app_id);
 
+  // Returns the current |update_check_count_| and resets it.
+  int GetUpdateCheckCountAndReset();
+
  private:
-  void SetupWebStore(const GURL& test_server_url);
-  void SetupWebStoreGalleryUrl();
-  void SetupCrxDownloadAndUpdateUrls(
-      net::test_server::EmbeddedTestServer* embedded_test_server);
-  void SetupCrxDownloadUrl();
-  void SetupCrxUpdateUrl(
-      net::test_server::EmbeddedTestServer* embedded_test_server);
+  void SetupWebStoreURL(const GURL& test_server_url);
+  void OverrideGalleryCommandlineSwitches();
+
   // Sets up |update_check_content_| used in update request response later by
   // kiosk app update server request handler |HandleRequest|.
   void SetUpdateCheckContent(const std::string& update_check_file,
                              const GURL& crx_download_url,
                              const std::string& app_id,
                              const std::string& crx_fp,
                              const std::string& crx_size,
                              const std::string& version,
                              std::string* update_check_content);
+
   // Request handler for kiosk app update server.
   scoped_ptr<net::test_server::HttpResponse> HandleRequest(
       const net::test_server::HttpRequest& request);
 
   GURL web_store_url_;
+
+  std::string has_update_template_;
+  std::string no_update_template_;
+  std::string update_check_end_point_;
+
   std::string update_check_content_;
+  int update_check_count_;
 
   DISALLOW_COPY_AND_ASSIGN(FakeCWS);
 };
 
 }  // namespace chromeos
 
 #endif  // CHROME_BROWSER_CHROMEOS_APP_MODE_FAKE_CWS_H_