chrome.downloads.open()

chrome/browser/extensions/api/downloads/downloads_api_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 <algorithm>
 
 #include "base/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/json/json_reader.h"
 #include "base/json/json_writer.h"
@@ skipping 796 matching lines @@
   scoped_ptr<TestURLRequestContext> url_request_context_;
   fileapi::FileSystemContext* fs_;
 
   DISALLOW_COPY_AND_ASSIGN(HTML5FileWriter);
 };
 
 const char HTML5FileWriter::kHTML5FileWritten[] = "html5_file_written";
 const char HTML5FileWriter::kURLRequestContextToreDown[] =
   "url_request_context_tore_down";
 
+// While an object of this class exists, it will mock out download
+// opening for all downloads created on the specified download manager.
+class MockDownloadOpeningObserver : public DownloadManager::Observer {
+ public:
+  explicit MockDownloadOpeningObserver(DownloadManager* manager)
+      : download_manager_(manager) {
+    download_manager_->AddObserver(this);
+  }
+
+  ~MockDownloadOpeningObserver() {
+    download_manager_->RemoveObserver(this);
+  }
+
+  virtual void OnDownloadCreated(
+      DownloadManager* manager, DownloadItem* item) OVERRIDE {
+    item->MockDownloadOpenForTesting();
+  }
+
+ private:
+  DownloadManager* download_manager_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockDownloadOpeningObserver);
+};
+
 }  // namespace
 
 IN_PROC_BROWSER_TEST_F(DownloadExtensionTest,
+                       DownloadExtensionTest_Open) {
+  EXPECT_STREQ(download_extension_errors::kInvalidOperationError,
+               RunFunctionAndReturnError(
+                   new DownloadsOpenFunction(),
+                   "[-42]").c_str());
+
+  MockDownloadOpeningObserver mock_open_observer(GetCurrentManager());
+  DownloadItem* download_item = CreateSlowTestDownload();
+  ASSERT_TRUE(download_item);
+  EXPECT_FALSE(download_item->GetOpened());
+  EXPECT_FALSE(download_item->GetOpenWhenComplete());
+  ASSERT_TRUE(WaitFor(events::kOnDownloadCreated,
+      base::StringPrintf("[{\"danger\": \"safe\","
+                         "  \"incognito\": false,"
+                         "  \"mime\": \"application/octet-stream\","
+                         "  \"paused\": false,"
+                         "  \"url\": \"%s\"}]",
+                         download_item->GetURL().spec().c_str())));
+  FinishPendingSlowDownloads();
+  EXPECT_FALSE(download_item->GetOpened());
+  EXPECT_TRUE(RunFunction(new DownloadsOpenFunction(),
+                          DownloadItemIdAsArgList(download_item)));
+  EXPECT_TRUE(download_item->GetOpened());
+  ASSERT_TRUE(WaitFor(events::kOnDownloadChanged,
+      base::StringPrintf("[{\"id\": %d,"
+                         "  \"opened\": {"
+                         "    \"previous\": false,"
+                         "    \"current\": true}}]",
+                         download_item->GetId())));
+
+  download_item = CreateSlowTestDownload();
+  ASSERT_TRUE(download_item);
+  EXPECT_FALSE(download_item->GetOpened());
+  EXPECT_FALSE(download_item->GetOpenWhenComplete());
+  ASSERT_TRUE(WaitFor(events::kOnDownloadCreated,
+      base::StringPrintf("[{\"danger\": \"safe\","
+                         "  \"incognito\": false,"
+                         "  \"mime\": \"application/octet-stream\","
+                         "  \"paused\": false,"
+                         "  \"url\": \"%s\"}]",
+                         download_item->GetURL().spec().c_str())));
+  EXPECT_TRUE(RunFunction(new DownloadsOpenFunction(),
+                          DownloadItemIdAsArgList(download_item)));
+  EXPECT_TRUE(download_item->GetOpenWhenComplete());
+  EXPECT_FALSE(download_item->GetOpened());
+  FinishPendingSlowDownloads();
+  EXPECT_TRUE(download_item->GetOpened());
+  ASSERT_TRUE(WaitFor(events::kOnDownloadChanged,
+      base::StringPrintf("[{\"id\": %d,"
+                         "  \"opened\": {"
+                         "    \"previous\": false,"
+                         "    \"current\": true}}]",
+                         download_item->GetId())));
+}
+
+IN_PROC_BROWSER_TEST_F(DownloadExtensionTest,
                        DownloadExtensionTest_PauseResumeCancel) {
   DownloadItem* download_item = CreateSlowTestDownload();
   ASSERT_TRUE(download_item);
 
   // Call pause().  It should succeed and the download should be paused on
   // return.
   EXPECT_TRUE(RunFunction(new DownloadsPauseFunction(),
                           DownloadItemIdAsArgList(download_item)));
   EXPECT_TRUE(download_item->IsPaused());
 
@@ skipping 1243 matching lines @@
                           "  \"state\": {"
                           "    \"previous\": \"in_progress\","
                           "    \"current\": \"complete\"}}]",
                           result_id,
                           GetFilename("on_record.txt.crdownload").c_str(),
                           GetFilename("on_record.txt").c_str())));
   std::string disk_data;
   EXPECT_TRUE(file_util::ReadFileToString(item->GetFullPath(), &disk_data));
   EXPECT_STREQ(kPayloadData, disk_data.c_str());
 }