Prompt user for file access permission before download starts

chrome/browser/download/download_target_determiner_unittest.cc

 // Copyright (c) 2012 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/at_exit.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/location.h"
 #include "base/observer_list.h"
 #include "base/prefs/pref_service.h"
@@ skipping 29 matching lines @@
 #if defined(ENABLE_PLUGINS)
 #include "content/public/browser/plugin_service.h"
 #include "content/public/browser/plugin_service_filter.h"
 #include "content/public/common/webplugininfo.h"
 #endif
 
 #if defined(ENABLE_EXTENSIONS)
 #include "extensions/common/extension.h"
 #endif
 
+#if defined(OS_ANDROID)
+#include "chrome/browser/android/download/mock_download_controller_android.h"
+#endif
+
 using ::testing::AnyNumber;
 using ::testing::Invoke;
 using ::testing::Ref;
 using ::testing::Return;
 using ::testing::ReturnRef;
 using ::testing::ReturnRefOfCopy;
 using ::testing::Truly;
 using ::testing::WithArg;
 using ::testing::_;
 using content::DownloadItem;
@@ skipping 190 matching lines @@
   }
 
   MockDownloadTargetDeterminerDelegate* delegate() {
     return &delegate_;
   }
 
   DownloadPrefs* download_prefs() {
     return download_prefs_.get();
   }
 
+#if defined(OS_ANDROID)
+  chrome::android::MockDownloadControllerAndroid* download_controller() {
+    return &download_controller_;
+  }
+#endif
+
  private:
   scoped_ptr<DownloadPrefs> download_prefs_;
   ::testing::NiceMock<MockDownloadTargetDeterminerDelegate> delegate_;
   NullWebContentsDelegate web_contents_delegate_;
   base::ScopedTempDir test_download_dir_;
   base::FilePath test_virtual_dir_;
+#if defined(OS_ANDROID)
+  chrome::android::MockDownloadControllerAndroid download_controller_;
+#endif
 };
 
 void DownloadTargetDeterminerTest::SetUp() {
   ChromeRenderViewHostTestHarness::SetUp();
   CHECK(profile());
   download_prefs_.reset(new DownloadPrefs(profile()));
   web_contents()->SetDelegate(&web_contents_delegate_);
   ASSERT_TRUE(test_download_dir_.CreateUniqueTempDir());
   test_virtual_dir_ = test_download_dir().Append(FILE_PATH_LITERAL("virtual"));
   download_prefs_->SetDownloadPath(test_download_dir());
   delegate_.SetupDefaults();
+#if defined(OS_ANDROID)
+  content::DownloadControllerAndroid::SetDownloadControllerAndroid(
+     &download_controller_);
+#endif
 }
 
 void DownloadTargetDeterminerTest::TearDown() {
   download_prefs_.reset();
+#if defined(OS_ANDROID)
+  content::DownloadControllerAndroid::SetDownloadControllerAndroid(nullptr);
+#endif
   ChromeRenderViewHostTestHarness::TearDown();
 }
 
 content::MockDownloadItem*
 DownloadTargetDeterminerTest::CreateActiveDownloadItem(
     int32 id,
     const DownloadTestCase& test_case) {
   content::MockDownloadItem* item =
       new ::testing::NiceMock<content::MockDownloadItem>();
   GURL download_url(test_case.url);
@@ skipping 856 matching lines @@
     },
   };
 
   SetPromptForDownload(true);
   EnableAutoOpenBasedOnExtension(
       base::FilePath(FILE_PATH_LITERAL("dummy.dummy")));
   RunTestCasesWithActiveItem(kPromptingTestCases,
                              arraysize(kPromptingTestCases));
 }
 
+#if defined(OS_ANDROID)
+TEST_F(DownloadTargetDeterminerTest,
+       TargetDeterminer_DisapprovePromptForUserPermission) {
+  const DownloadTestCase kUserPermissionTestCases[] = {
+    {
+      // 0: Automatic Safe
+      AUTOMATIC,
+      content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS,
+      "http://example.com/foo.txt", "text/plain",
+      FILE_PATH_LITERAL(""),
+
+      FILE_PATH_LITERAL(""),
+      DownloadItem::TARGET_DISPOSITION_OVERWRITE,
+
+      EXPECT_LOCAL_PATH
+    },
+  };
+  content::DownloadControllerAndroid::Get()->
+      SetApproveFileAccessRequestForTesting(false);
+  RunTestCasesWithActiveItem(kUserPermissionTestCases,
+                             arraysize(kUserPermissionTestCases));
+}
+#endif
+
 #if defined(ENABLE_EXTENSIONS)
 // These test cases are run with "Prompt for download" user preference set to
 // true. Automatic extension downloads shouldn't cause prompting.
 // Android doesn't support extensions.
 TEST_F(DownloadTargetDeterminerTest, TargetDeterminer_PromptAlways_Extension) {
   const DownloadTestCase kPromptingTestCases[] = {
     {
       // 0: Automatic Browser Extension download. - Shouldn't prompt for browser
       //    extension downloads even if "Prompt for download" preference is set.
       AUTOMATIC,
@@ skipping 1072 matching lines @@
   EXPECT_CALL(mock_plugin_filter_, MockPluginAvailable(npapi_plugin.path()))
       .WillRepeatedly(Return(true));
 
   target_info = RunDownloadTargetDeterminer(
       GetPathInDownloadDir(kInitialPath), item.get());
   EXPECT_FALSE(target_info->is_filetype_handled_safely);
 }
 #endif  // defined(ENABLE_PLUGINS)
 
 }  // namespace