MD Downloads: "Clear All" should remove dangerous downloads for good

chrome/browser/ui/webui/md_downloads/md_downloads_dom_handler_unittest.cc

 // Copyright 2016 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/md_downloads/md_downloads_dom_handler.h"
 
+#include <vector>
+
+#include "chrome/browser/download/download_item_model.h"
 #include "chrome/test/base/testing_profile.h"
+#include "content/public/test/mock_download_item.h"
 #include "content/public/test/mock_download_manager.h"
 #include "content/public/test/test_browser_thread_bundle.h"
 #include "content/public/test/test_web_ui.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 class TestMdDownloadsDOMHandler : public MdDownloadsDOMHandler {
  public:
   explicit TestMdDownloadsDOMHandler(content::DownloadManager* download_manager,
                                      content::WebUI* web_ui)
       : MdDownloadsDOMHandler(download_manager, web_ui) {}
 
   using MdDownloadsDOMHandler::set_web_ui;
+  using MdDownloadsDOMHandler::FinalizeRemovals;
+  using MdDownloadsDOMHandler::RemoveDownloads;
 };
 
 }  // namespace
 
 // A fixture to test MdDownloadsDOMHandler.
 class MdDownloadsDOMHandlerTest : public testing::Test {
  public:
   // testing::Test:
   void SetUp() override {
     ON_CALL(manager_, GetBrowserContext())
@@ skipping 27 matching lines @@
   TestMdDownloadsDOMHandler handler(manager(), web_ui());
   handler.set_web_ui(web_ui());
 
   base::ListValue empty_search_terms;
   handler.HandleGetDownloads(&empty_search_terms);
 
   EXPECT_EQ(1U, web_ui()->call_data().size());
   EXPECT_EQ("downloads.Manager.insertItems",
             web_ui()->call_data()[0]->function_name());
 }
+
+TEST_F(MdDownloadsDOMHandlerTest, ClearAll) {
+  std::vector<content::DownloadItem*> downloads;
+
+  // Safe, in-progress items should be passed over.
+  testing::StrictMock<content::MockDownloadItem> in_progress;
+  EXPECT_CALL(in_progress, IsDangerous()).WillOnce(testing::Return(false));
+  EXPECT_CALL(in_progress, GetState()).WillOnce(
+      testing::Return(content::DownloadItem::IN_PROGRESS));
+  downloads.push_back(&in_progress);
+
+  // Dangerous items should be removed (regardless of state).
+  testing::StrictMock<content::MockDownloadItem> dangerous;
+  EXPECT_CALL(dangerous, IsDangerous()).WillOnce(testing::Return(true));
+  EXPECT_CALL(dangerous, Remove());
+  downloads.push_back(&dangerous);
+
+  // Completed items should be marked as hidden from the shelf.
+  testing::StrictMock<content::MockDownloadItem> completed;
+  EXPECT_CALL(completed, IsDangerous()).WillOnce(testing::Return(false));
+  EXPECT_CALL(completed, GetState()).WillOnce(
+      testing::Return(content::DownloadItem::COMPLETE));
+  EXPECT_CALL(completed, GetId()).WillOnce(testing::Return(1));
+  EXPECT_CALL(completed, UpdateObservers());
+  downloads.push_back(&completed);
+
+  ASSERT_TRUE(DownloadItemModel(&completed).ShouldShowInShelf());
+
+  TestMdDownloadsDOMHandler handler(manager(), web_ui());
+  handler.RemoveDownloads(downloads);
+
+  // Ensure |completed| has been "soft removed" (i.e. can be revived).
+  EXPECT_FALSE(DownloadItemModel(&completed).ShouldShowInShelf());
+
+  // Make sure |completed| actually get removed when removals are "finalized".
+  EXPECT_CALL(*manager(), GetDownload(1)).WillOnce(testing::Return(&completed));
+  EXPECT_CALL(completed, Remove());
+  handler.FinalizeRemovals();
+}