[Download Notification] Add tests of incognito download

chrome/browser/download/notification/download_notification_browsertest.cc

 // Copyright 2015 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 "base/command_line.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/chromeos/profiles/profile_helper.h"
 #include "chrome/browser/download/chrome_download_manager_delegate.h"
 #include "chrome/browser/download/download_service.h"
 #include "chrome/browser/download/download_service_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/signin/signin_manager_factory.h"
 #include "chrome/browser/ui/browser.h"
+#include "chrome/browser/ui/browser_commands.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/grit/chromium_strings.h"
 #include "chrome/grit/generated_resources.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "chromeos/chromeos_switches.h"
 #include "components/signin/core/browser/signin_manager_base.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/download_item.h"
 #include "content/public/browser/download_manager.h"
@@ skipping 236 matching lines @@
  public:
   ~DownloadNotificationTest() override {}
 
   void SetUpOnMainThread() override {
     Profile* profile = browser()->profile();
 
     scoped_ptr<TestChromeDownloadManagerDelegate> test_delegate;
     test_delegate.reset(new TestChromeDownloadManagerDelegate(profile));
     test_delegate->GetDownloadIdReceiverCallback().Run(
         content::DownloadItem::kInvalidId + 1);
-
     DownloadServiceFactory::GetForBrowserContext(profile)
         ->SetDownloadManagerDelegateForTesting(test_delegate.Pass());
 
     DownloadNotificationTestBase::SetUpOnMainThread();
   }
 
   TestChromeDownloadManagerDelegate* GetDownloadManagerDelegate() const {
     return static_cast<TestChromeDownloadManagerDelegate*>(
         DownloadServiceFactory::GetForBrowserContext(browser()->profile())
             ->GetDownloadManagerDelegate());
   }
 
+  void PrepareIncognitoBrowser() {
+    incognito_browser_ = CreateIncognitoBrowser();
+    Profile* incognito_profile = incognito_browser_->profile();
+
+    scoped_ptr<TestChromeDownloadManagerDelegate> incognito_test_delegate;
+    incognito_test_delegate.reset(
+        new TestChromeDownloadManagerDelegate(incognito_profile));
+    DownloadServiceFactory::GetForBrowserContext(incognito_profile)
+        ->SetDownloadManagerDelegateForTesting(incognito_test_delegate.Pass());
+  }
+
+  TestChromeDownloadManagerDelegate* GetIncognitoDownloadManagerDelegate()
+      const {
+    Profile* incognito_profile = incognito_browser()->profile();
+    return static_cast<TestChromeDownloadManagerDelegate*>(
+        DownloadServiceFactory::GetForBrowserContext(incognito_profile)->
+        GetDownloadManagerDelegate());
+  }
+
   void CreateDownload() {
+    return CreateDownloadForBrowser(browser());
+  }
+
+  void CreateDownloadForBrowser(Browser* browser) {
     GURL url(net::URLRequestSlowDownloadJob::kKnownSizeUrl);
 
     // Starts a download.
     NotificationAddObserver download_start_notification_observer;
-    ui_test_utils::NavigateToURL(browser(), url);
+    ui_test_utils::NavigateToURL(browser, url);
     EXPECT_TRUE(download_start_notification_observer.Wait());
 
     // Confirms that a notification is created.
     notification_id_ = download_start_notification_observer.notification_id();
     EXPECT_FALSE(notification_id_.empty());
     ASSERT_TRUE(notification());
 
     // Confirms that there is only one notification.
     message_center::NotificationList::Notifications
         visible_notifications = GetMessageCenter()->GetVisibleNotifications();
     EXPECT_EQ(1u, visible_notifications.size());
     EXPECT_TRUE(IsInNotifications(visible_notifications, notification_id_));
 
     // Confirms that a download is also started.
     std::vector<content::DownloadItem*> downloads;
-    GetDownloadManager(browser())->GetAllDownloads(&downloads);
+    GetDownloadManager(browser)->GetAllDownloads(&downloads);
     EXPECT_EQ(1u, downloads.size());
     download_item_ = downloads[0];
     ASSERT_TRUE(download_item_);
   }
 
   content::DownloadItem* download_item() const { return download_item_; }
   std::string notification_id() const { return notification_id_; }
   message_center::Notification* notification() const {
     return GetNotification(notification_id_);
   }
+  Browser* incognito_browser() const { return incognito_browser_; }
 
  private:
   content::DownloadItem* download_item_ = nullptr;
+  Browser* incognito_browser_ = nullptr;
   std::string notification_id_;
 };
 
 IN_PROC_BROWSER_TEST_F(DownloadNotificationTest, DownloadFile) {
   CreateDownload();
 
   EXPECT_EQ(l10n_util::GetStringFUTF16(
                 IDS_DOWNLOAD_STATUS_IN_PROGRESS_TITLE,
                 download_item()->GetFileNameToReportUser().LossyDisplayName()),
             GetNotification(notification_id())->title());
@@ skipping 201 matching lines @@
   EXPECT_EQ(notification_id(), notification_close_observer.Wait());
   EXPECT_EQ(0u, GetMessageCenter()->GetVisibleNotifications().size());
 
   // Confirms that a download is also cancelled.
   std::vector<content::DownloadItem*> downloads;
   GetDownloadManager(browser())->GetAllDownloads(&downloads);
   EXPECT_EQ(1u, downloads.size());
   EXPECT_EQ(content::DownloadItem::CANCELLED, downloads[0]->GetState());
 }
 
+IN_PROC_BROWSER_TEST_F(DownloadNotificationTest, IncognitoDownloadFile) {
+  PrepareIncognitoBrowser();
+
+  // Starts an incognito download.
+  CreateDownloadForBrowser(incognito_browser());
+
+  EXPECT_EQ(l10n_util::GetStringFUTF16(
+                IDS_DOWNLOAD_STATUS_IN_PROGRESS_TITLE,
+                download_item()->GetFileNameToReportUser().LossyDisplayName()),
+            GetNotification(notification_id())->title());
+  EXPECT_EQ(message_center::NOTIFICATION_TYPE_PROGRESS,
+            GetNotification(notification_id())->type());
+  EXPECT_TRUE(download_item()->GetBrowserContext()->IsOffTheRecord());
+
+  // Requests to complete the download.
+  ui_test_utils::NavigateToURL(
+      incognito_browser(),
+      GURL(net::URLRequestSlowDownloadJob::kFinishDownloadUrl));
+
+  // Waits for download completion.
+  while (download_item()->GetState() != content::DownloadItem::COMPLETE) {
+    NotificationUpdateObserver download_change_notification_observer;
+    download_change_notification_observer.Wait();
+  }
+
+  EXPECT_EQ(l10n_util::GetStringFUTF16(
+                IDS_DOWNLOAD_STATUS_DOWNLOADED_TITLE,
+                download_item()->GetFileNameToReportUser().LossyDisplayName()),
+            GetNotification(notification_id())->title());
+  EXPECT_EQ(message_center::NOTIFICATION_TYPE_SIMPLE,
+            GetNotification(notification_id())->type());
+
+  // Opens the message center.
+  GetMessageCenter()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
+
+  // Try to open the downloaded item by clicking the notification.
+  EXPECT_FALSE(GetIncognitoDownloadManagerDelegate()->opened());
+  GetMessageCenter()->ClickOnNotification(notification_id());
+  EXPECT_TRUE(GetIncognitoDownloadManagerDelegate()->opened());
+  EXPECT_FALSE(GetDownloadManagerDelegate()->opened());
+
+  EXPECT_FALSE(GetNotification(notification_id()));
+  chrome::CloseWindow(incognito_browser());
+}
+
+IN_PROC_BROWSER_TEST_F(DownloadNotificationTest,
+                       SimultaneousIncognitoAndNormalDownloads) {
+  PrepareIncognitoBrowser();
+
+  GURL url_incognito(net::URLRequestSlowDownloadJob::kUnknownSizeUrl);
+  GURL url_normal(net::URLRequestSlowDownloadJob::kKnownSizeUrl);
+
+  // Starts the incognito download.
+  NotificationAddObserver download_start_notification_observer1;
+  ui_test_utils::NavigateToURL(incognito_browser(), url_incognito);
+  EXPECT_TRUE(download_start_notification_observer1.Wait());
+  std::string notification_id1 =
+      download_start_notification_observer1.notification_id();
+  EXPECT_FALSE(notification_id1.empty());
+
+  // Confirms that there is a download.
+  std::vector<content::DownloadItem*> downloads;
+  GetDownloadManager(browser())->GetAllDownloads(&downloads);
+  EXPECT_EQ(0u, downloads.size());
+  downloads.clear();
+  GetDownloadManager(incognito_browser())->GetAllDownloads(&downloads);
+  EXPECT_EQ(1u, downloads.size());
+  content::DownloadItem* download_incognito = downloads[0];
+
+  // Starts the normal download.
+  NotificationAddObserver download_start_notification_observer2;
+  ui_test_utils::NavigateToURL(browser(), url_normal);
+  EXPECT_TRUE(download_start_notification_observer2.Wait());
+  std::string notification_id2 =
+      download_start_notification_observer2.notification_id();
+  EXPECT_FALSE(notification_id2.empty());
+
+  // Confirms that there are 2 downloads.
+  downloads.clear();
+  GetDownloadManager(browser())->GetAllDownloads(&downloads);
+  content::DownloadItem* download_normal = downloads[0];
+  EXPECT_EQ(1u, downloads.size());
+  EXPECT_NE(download_normal, download_incognito);
+  downloads.clear();
+  GetDownloadManager(incognito_browser())->GetAllDownloads(&downloads);
+  EXPECT_EQ(1u, downloads.size());
+  EXPECT_EQ(download_incognito, downloads[0]);
+
+  // Confirms the types of download notifications are correct.
+  EXPECT_EQ(message_center::NOTIFICATION_TYPE_PROGRESS,
+            GetNotification(notification_id1)->type());
+  EXPECT_EQ(message_center::NOTIFICATION_TYPE_PROGRESS,
+            GetNotification(notification_id2)->type());
+
+  EXPECT_TRUE(download_incognito->GetBrowserContext()->IsOffTheRecord());
+  EXPECT_FALSE(download_normal->GetBrowserContext()->IsOffTheRecord());
+
+  // Requests to complete the downloads.
+  ui_test_utils::NavigateToURL(
+      browser(), GURL(net::URLRequestSlowDownloadJob::kFinishDownloadUrl));
+
+  // Waits for the completion of downloads.
+  while (download_normal->GetState() != content::DownloadItem::COMPLETE ||
+         download_incognito->GetState() != content::DownloadItem::COMPLETE) {
+    NotificationUpdateObserver download_change_notification_observer;
+    download_change_notification_observer.Wait();
+  }
+
+  // Confirms the types of download notifications are correct.
+  EXPECT_EQ(message_center::NOTIFICATION_TYPE_SIMPLE,
+            GetNotification(notification_id1)->type());
+  EXPECT_EQ(message_center::NOTIFICATION_TYPE_SIMPLE,
+            GetNotification(notification_id2)->type());
+
+  chrome::CloseWindow(incognito_browser());
+}
+
 //////////////////////////////////////////////////
 // Test with multi profiles
 //////////////////////////////////////////////////
 
 class MultiProfileDownloadNotificationTest
     : public DownloadNotificationTestBase {
  public:
   ~MultiProfileDownloadNotificationTest() override {}
 
   void SetUpCommandLine(base::CommandLine* command_line) override {
@@ skipping 132 matching lines @@
   }
 
   // Confirms the types of download notifications are correct.
   EXPECT_EQ(message_center::NOTIFICATION_TYPE_SIMPLE,
             GetNotification(notification_id1)->type());
   EXPECT_EQ(message_center::NOTIFICATION_TYPE_SIMPLE,
             GetNotification(notification_id2)->type());
   EXPECT_EQ(message_center::NOTIFICATION_TYPE_SIMPLE,
             GetNotification(notification_id3)->type());
 }