Support for testing by providing access to DownloadsList.

chrome/browser/ui/webui/active_downloads_ui.cc

 // Copyright (c) 2011 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.
 
 #include "chrome/browser/ui/webui/active_downloads_ui.h"
 
 #include <algorithm>
 #include <string>
 #include <vector>
 
@@ skipping 65 matching lines @@
   virtual std::string GetMimeType(const std::string&) const {
     return "text/html";
   }
 
  private:
   ~ActiveDownloadsUIHTMLSource() {}
 
   DISALLOW_COPY_AND_ASSIGN(ActiveDownloadsUIHTMLSource);
 };
 
-// The handler for Javascript messages related to the "active_downloads" view.

[achuithb, 2011/06/23 02:56:38]

This code just moved out of the anonymous namespace. No code change.

-class ActiveDownloadsHandler
-    : public WebUIMessageHandler,
-      public DownloadManager::Observer,
-      public DownloadItem::Observer {
- public:
-  ActiveDownloadsHandler();
-  virtual ~ActiveDownloadsHandler();
-
-  // Initialization after Attach.
-  void Init();
-
-  // WebUIMessageHandler implementation.
-  virtual WebUIMessageHandler* Attach(WebUI* web_ui);
-  virtual void RegisterMessages();
-
-  // DownloadItem::Observer interface.
-  virtual void OnDownloadUpdated(DownloadItem* item);
-  virtual void OnDownloadOpened(DownloadItem* item) { }
-
-  // DownloadManager::Observer interface.
-  virtual void ModelChanged();
-
-  // WebUI Callbacks.
-  void HandleGetDownloads(const ListValue* args);
-  void HandlePauseToggleDownload(const ListValue* args);
-  void HandleCancelDownload(const ListValue* args);
-  void HandleAllowDownload(const ListValue* args);
-  void OpenNewFullWindow(const ListValue* args);
-  void PlayMediaFile(const ListValue* args);
-
- private:
-  // Downloads helpers.
-  DownloadItem* GetDownloadById(const ListValue* args);
-  void UpdateDownloadList();
-  void SendDownloads();
-  void AddDownload(DownloadItem* item);
-  bool SelectTab(const GURL& url);
-
-  Profile* profile_;
-  TabContents* tab_contents_;
-  DownloadManager* download_manager_;
-
-  typedef std::vector<DownloadItem*> DownloadList;
-  DownloadList active_downloads_;
-  DownloadList downloads_;
-
-  DISALLOW_COPY_AND_ASSIGN(ActiveDownloadsHandler);
-};
-
 ////////////////////////////////////////////////////////////////////////////////
 //
 // ActiveDownloadsUIHTMLSource
 //
 ////////////////////////////////////////////////////////////////////////////////
 
 ActiveDownloadsUIHTMLSource::ActiveDownloadsUIHTMLSource()
     : DataSource(chrome::kChromeUIActiveDownloadsHost, MessageLoop::current()) {
 }
 
@@ skipping 36 matching lines @@
   const std::string full_html = jstemplate_builder::GetI18nTemplateHtml(
       active_downloads_html, &localized_strings);
 
   scoped_refptr<RefCountedBytes> html_bytes(new RefCountedBytes);
   html_bytes->data.resize(full_html.size());
   std::copy(full_html.begin(), full_html.end(), html_bytes->data.begin());
 
   SendResponse(request_id, html_bytes);
 }
 
+}  // namespace
+
 ////////////////////////////////////////////////////////////////////////////////
 //
 // ActiveDownloadsHandler
 //
 ////////////////////////////////////////////////////////////////////////////////
 
+// The handler for Javascript messages related to the "active_downloads" view.
+class ActiveDownloadsHandler
+    : public WebUIMessageHandler,
+      public DownloadManager::Observer,
+      public DownloadItem::Observer {
+ public:
+  ActiveDownloadsHandler();
+  virtual ~ActiveDownloadsHandler();
+
+  // Initialization after Attach.
+  void Init();
+
+  // WebUIMessageHandler implementation.
+  virtual WebUIMessageHandler* Attach(WebUI* web_ui);
+  virtual void RegisterMessages();
+
+  // DownloadItem::Observer interface.
+  virtual void OnDownloadUpdated(DownloadItem* item);
+  virtual void OnDownloadOpened(DownloadItem* item) { }
+
+  // DownloadManager::Observer interface.
+  virtual void ModelChanged();
+
+  // WebUI Callbacks.
+  void HandleGetDownloads(const ListValue* args);
+  void HandlePauseToggleDownload(const ListValue* args);
+  void HandleCancelDownload(const ListValue* args);
+  void HandleAllowDownload(const ListValue* args);
+  void OpenNewFullWindow(const ListValue* args);
+  void PlayMediaFile(const ListValue* args);
+
+  // For testing.
+  typedef std::vector<DownloadItem*> DownloadList;
+  const DownloadList& downloads() const { return downloads_; }
+
+ private:
+  // Downloads helpers.
+  DownloadItem* GetDownloadById(const ListValue* args);
+  void UpdateDownloadList();
+  void SendDownloads();
+  void AddDownload(DownloadItem* item);
+  bool SelectTab(const GURL& url);
+
+  Profile* profile_;
+  TabContents* tab_contents_;
+  DownloadManager* download_manager_;
+
+  DownloadList active_downloads_;
+  DownloadList downloads_;
+
+  DISALLOW_COPY_AND_ASSIGN(ActiveDownloadsHandler);
+};
+
 ActiveDownloadsHandler::ActiveDownloadsHandler()
     : profile_(NULL),
       tab_contents_(NULL),
       download_manager_(NULL) {
 }
 
 ActiveDownloadsHandler::~ActiveDownloadsHandler() {
   for (size_t i = 0; i < downloads_.size(); ++i) {
     downloads_[i]->RemoveObserver(this);
   }
@@ skipping 150 matching lines @@
       active_downloads_.erase(ita);
     SendDownloads();
   } else {
     const size_t id = it - downloads_.begin();
     scoped_ptr<DictionaryValue> result(
         download_util::CreateDownloadItemValue(item, id));
     web_ui_->CallJavascriptFunction("downloadUpdated", *result);
   }
 }
 
-}  // namespace
-
 ////////////////////////////////////////////////////////////////////////////////
 //
 // ActiveDownloadsUI
 //
 ////////////////////////////////////////////////////////////////////////////////
 
 
 ActiveDownloadsUI::ActiveDownloadsUI(TabContents* contents)
-    : HtmlDialogUI(contents) {
-  ActiveDownloadsHandler* handler = new ActiveDownloadsHandler();
-  AddMessageHandler(handler->Attach(this));
-  handler->Init();
+    : HtmlDialogUI(contents),
+      handler_(new ActiveDownloadsHandler()) {
+  AddMessageHandler(handler_->Attach(this));
+  handler_->Init();
   ActiveDownloadsUIHTMLSource* html_source = new ActiveDownloadsUIHTMLSource();
 
   // Set up the chrome://active-downloads/ source.
   contents->profile()->GetChromeURLDataManager()->AddDataSource(html_source);
 }
 
 // static
 Browser* ActiveDownloadsUI::OpenPopup(Profile* profile) {
   Browser* browser = GetPopup(profile);
 
@@ skipping 34 matching lines @@
 
       if (url.SchemeIs(chrome::kChromeUIScheme) &&
           url.host() == chrome::kChromeUIActiveDownloadsHost) {
         return (*it);
       }
     }
   }
   return NULL;
 }
 
+const ActiveDownloadsUI::DownloadList& ActiveDownloadsUI::GetDownloads() const {
+  return handler_->downloads();
+}