Reshow the notification after validating dangerous 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_prefs.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"
+#include "content/public/test/download_test_observer.h"
 #include "grit/theme_resources.h"
 #include "net/test/url_request/url_request_slow_download_job.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/message_center/message_center.h"
 #include "ui/message_center/message_center_observer.h"
 #include "url/gurl.h"
 
 namespace {
 
 enum {
@@ skipping 258 matching lines @@
 
   TestChromeDownloadManagerDelegate* GetIncognitoDownloadManagerDelegate()
       const {
     Profile* incognito_profile = incognito_browser()->profile();
     return static_cast<TestChromeDownloadManagerDelegate*>(
         DownloadServiceFactory::GetForBrowserContext(incognito_profile)->
         GetDownloadManagerDelegate());
   }
 
   void CreateDownload() {
-    return CreateDownloadForBrowser(browser());
+    return CreateDownloadForBrowserAndURL(
+        browser(),
+        GURL(net::URLRequestSlowDownloadJob::kKnownSizeUrl));
   }
 
-  void CreateDownloadForBrowser(Browser* browser) {
-    GURL url(net::URLRequestSlowDownloadJob::kKnownSizeUrl);
-
+  void CreateDownloadForBrowserAndURL(Browser* browser, GURL url) {
     // Starts a download.
     NotificationAddObserver download_start_notification_observer;
     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());
 
@@ skipping 10 matching lines @@
     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_; }
+  base::FilePath GetDownloadPath() {
+    return DownloadPrefs::FromDownloadManager(GetDownloadManager(browser()))->
+      DownloadPath();
+  }
 
  private:
   content::DownloadItem* download_item_ = nullptr;
   Browser* incognito_browser_ = nullptr;
   std::string notification_id_;
 };
 
 IN_PROC_BROWSER_TEST_F(DownloadNotificationTest, DownloadFile) {
   CreateDownload();
 
@@ skipping 25 matching lines @@
   GetMessageCenter()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
 
   // Try to open the downloaded item by clicking the notification.
   EXPECT_FALSE(GetDownloadManagerDelegate()->opened());
   GetMessageCenter()->ClickOnNotification(notification_id());
   EXPECT_TRUE(GetDownloadManagerDelegate()->opened());
 
   EXPECT_FALSE(GetNotification(notification_id()));
 }
 
+IN_PROC_BROWSER_TEST_F(DownloadNotificationTest, DownloadDangerousFile) {
+  ASSERT_TRUE(test_server()->Start());
+  GURL download_url(
+      test_server()->GetURL("files/downloads/dangerous/dangerous.swf"));
+
+  content::DownloadTestObserverTerminal download_terminal_observer(
+      GetDownloadManager(browser()),
+      1u,  /* wait_count */
+      content::DownloadTestObserver::ON_DANGEROUS_DOWNLOAD_IGNORE);
+
+  CreateDownloadForBrowserAndURL(browser(), download_url);
+
+  base::FilePath filename = download_item()->GetFileNameToReportUser();
+
+  // Checks the download status.
+  EXPECT_EQ(content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE,
+            download_item()->GetDangerType());
+  EXPECT_TRUE(download_item()->IsDangerous());
+
+  // Opens the message center.
+  GetMessageCenter()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
+
+  NotificationRemoveObserver notification_close_observer;
+  NotificationAddObserver notification_add_observer;
+
+  // Cicks the "keep" button.
+  notification()->ButtonClick(1);  // 2nd button: "Keep"
+  // Clicking makes the message center closed.
+  GetMessageCenter()->SetVisibility(message_center::VISIBILITY_TRANSIENT);
+
+  // Confirms that the notification is closed and re-shown.
+  EXPECT_EQ(notification_id(), notification_close_observer.Wait());
+  notification_add_observer.Wait();
+  EXPECT_EQ(notification_id(), notification_add_observer.notification_id());
+  EXPECT_EQ(1u, GetMessageCenter()->GetVisibleNotifications().size());
+
+  // Checks the download status.
+  EXPECT_EQ(content::DOWNLOAD_DANGER_TYPE_USER_VALIDATED,
+            download_item()->GetDangerType());
+  EXPECT_FALSE(download_item()->IsDangerous());
+
+  // Wait for the download completion.
+  download_terminal_observer.WaitForFinished();
+
+  // Checks the download status.
+  EXPECT_FALSE(download_item()->IsDangerous());
+  EXPECT_EQ(content::DownloadItem::COMPLETE, download_item()->GetState());
+
+  // Checks the downloaded file.
+  EXPECT_TRUE(base::PathExists(GetDownloadPath().Append(filename.BaseName())));
+}
+
+IN_PROC_BROWSER_TEST_F(DownloadNotificationTest, DiscardDangerousFile) {
+  ASSERT_TRUE(test_server()->Start());
+  GURL download_url(
+      test_server()->GetURL("files/downloads/dangerous/dangerous.swf"));
+
+  content::DownloadTestObserverTerminal download_terminal_observer(
+      GetDownloadManager(browser()),
+      1u,  /* wait_count */
+      content::DownloadTestObserver::ON_DANGEROUS_DOWNLOAD_IGNORE);
+
+  CreateDownloadForBrowserAndURL(browser(), download_url);
+
+  base::FilePath filename = download_item()->GetFileNameToReportUser();
+
+  // Checks the download status.
+  EXPECT_EQ(content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE,
+            download_item()->GetDangerType());
+  EXPECT_TRUE(download_item()->IsDangerous());
+
+  // Opens the message center.
+  GetMessageCenter()->SetVisibility(message_center::VISIBILITY_MESSAGE_CENTER);
+
+  NotificationRemoveObserver notification_close_observer;
+
+  // Clicks the "Discard" button.
+  notification()->ButtonClick(0);  // 1st button: "Discard"
+  // Clicking makes the message center closed.
+  GetMessageCenter()->SetVisibility(message_center::VISIBILITY_TRANSIENT);
+
+  // Confirms that the notification is closed.
+  EXPECT_EQ(notification_id(), notification_close_observer.Wait());
+
+  // Wait for the download completion.
+  download_terminal_observer.WaitForFinished();
+
+  // Checks there is neither any download nor any notification.
+  EXPECT_EQ(0u, GetMessageCenter()->GetVisibleNotifications().size());
+  std::vector<content::DownloadItem*> downloads;
+  GetDownloadManager(browser())->GetAllDownloads(&downloads);
+  EXPECT_EQ(0u, downloads.size());
+
+  // Checks the downloaded file doesn't exist.
+  EXPECT_FALSE(base::PathExists(GetDownloadPath().Append(filename.BaseName())));
+}
+
 IN_PROC_BROWSER_TEST_F(DownloadNotificationTest,
                        CloseNotificationAfterDownload) {
   CreateDownload();
 
   // Requests to complete the download.
   ui_test_utils::NavigateToURL(
       browser(), GURL(net::URLRequestSlowDownloadJob::kFinishDownloadUrl));
 
   // Waits for download completion.
   while (download_item()->GetState() != content::DownloadItem::COMPLETE) {
@@ skipping 164 matching lines @@
   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());
+  CreateDownloadForBrowserAndURL(
+      incognito_browser(),
+      GURL(net::URLRequestSlowDownloadJob::kKnownSizeUrl));
 
   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.
@@ skipping 244 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());
 }