components/drive/file_system/search_operation_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "components/drive/file_system/search_operation.h"
   6
   7 #include "base/callback_helpers.h"
   8 #include "components/drive/change_list_loader.h"
   9 #include "components/drive/file_system/operation_test_base.h"
  10 #include "components/drive/service/fake_drive_service.h"
  11 #include "content/public/test/test_utils.h"
  12 #include "google_apis/drive/drive_api_parser.h"
  13 #include "google_apis/drive/test_util.h"
  14 #include "testing/gtest/include/gtest/gtest.h"
  15
  16 namespace drive {
  17 namespace file_system {
  18
  19 typedef OperationTestBase SearchOperationTest;
  20
  21 TEST_F(SearchOperationTest, ContentSearch) {
  22   SearchOperation operation(blocking_task_runner(), scheduler(), metadata(),
  23                             loader_controller());
  24
  25   std::set<std::string> expected_results;
  26   expected_results.insert(
  27       "drive/root/Directory 1/Sub Directory Folder/Sub Sub Directory Folder");
  28   expected_results.insert("drive/root/Directory 1/Sub Directory Folder");
  29   expected_results.insert("drive/root/Directory 1/SubDirectory File 1.txt");
  30   expected_results.insert("drive/root/Directory 1");
  31   expected_results.insert("drive/root/Directory 2 excludeDir-test");
  32
  33   FileError error = FILE_ERROR_FAILED;
  34   GURL next_link;
  35   scoped_ptr<std::vector<SearchResultInfo> > results;
  36
  37   operation.Search("Directory", GURL(),
  38                    google_apis::test_util::CreateCopyResultCallback(
  39                        &error, &next_link, &results));
  40   content::RunAllBlockingPoolTasksUntilIdle();
  41
  42   EXPECT_EQ(FILE_ERROR_OK, error);
  43   EXPECT_TRUE(next_link.is_empty());
  44   EXPECT_EQ(expected_results.size(), results->size());
  45   for (size_t i = 0; i < results->size(); i++) {
  46     EXPECT_TRUE(expected_results.count(results->at(i).path.AsUTF8Unsafe()))
  47         << results->at(i).path.AsUTF8Unsafe();
  48   }
  49 }
  50
  51 TEST_F(SearchOperationTest, ContentSearchWithNewEntry) {
  52   SearchOperation operation(blocking_task_runner(), scheduler(), metadata(),
  53                             loader_controller());
  54
  55   // Create a new directory in the drive service.
  56   google_apis::DriveApiErrorCode status = google_apis::DRIVE_OTHER_ERROR;
  57   scoped_ptr<google_apis::FileResource> server_entry;
  58   fake_service()->AddNewDirectory(
  59       fake_service()->GetRootResourceId(), "New Directory 1!",
  60       AddNewDirectoryOptions(),
  61       google_apis::test_util::CreateCopyResultCallback(&status, &server_entry));
  62   content::RunAllBlockingPoolTasksUntilIdle();
  63   ASSERT_EQ(google_apis::HTTP_CREATED, status);
  64
  65   // As the result of the first Search(), only entries in the current file
  66   // system snapshot are expected to be returned in the "right" path. New
  67   // entries like "New Directory 1!" is temporarily added to "drive/other".
  68   std::set<std::string> expected_results;
  69   expected_results.insert("drive/root/Directory 1");
  70   expected_results.insert("drive/other/New Directory 1!");
  71
  72   FileError error = FILE_ERROR_FAILED;
  73   GURL next_link;
  74   scoped_ptr<std::vector<SearchResultInfo> > results;
  75
  76   operation.Search("\"Directory 1\"", GURL(),
  77                    google_apis::test_util::CreateCopyResultCallback(
  78                        &error, &next_link, &results));
  79   content::RunAllBlockingPoolTasksUntilIdle();
  80
  81   EXPECT_EQ(FILE_ERROR_OK, error);
  82   EXPECT_TRUE(next_link.is_empty());
  83   ASSERT_EQ(expected_results.size(), results->size());
  84   for (size_t i = 0; i < results->size(); i++) {
  85     EXPECT_TRUE(expected_results.count(results->at(i).path.AsUTF8Unsafe()))
  86         << results->at(i).path.AsUTF8Unsafe();
  87   }
  88
  89   // Load the change from FakeDriveService.
  90   ASSERT_EQ(FILE_ERROR_OK, CheckForUpdates());
  91
  92   // Now the new entry must be reported to be in the right directory.
  93   expected_results.clear();
  94   expected_results.insert("drive/root/Directory 1");
  95   expected_results.insert("drive/root/New Directory 1!");
  96   error = FILE_ERROR_FAILED;
  97   operation.Search("\"Directory 1\"", GURL(),
  98                    google_apis::test_util::CreateCopyResultCallback(
  99                        &error, &next_link, &results));
 100   content::RunAllBlockingPoolTasksUntilIdle();
 101
 102   EXPECT_EQ(FILE_ERROR_OK, error);
 103   EXPECT_TRUE(next_link.is_empty());
 104   ASSERT_EQ(expected_results.size(), results->size());
 105   for (size_t i = 0; i < results->size(); i++) {
 106     EXPECT_TRUE(expected_results.count(results->at(i).path.AsUTF8Unsafe()))
 107         << results->at(i).path.AsUTF8Unsafe();
 108   }
 109 }
 110
 111 TEST_F(SearchOperationTest, ContentSearchEmptyResult) {
 112   SearchOperation operation(blocking_task_runner(), scheduler(), metadata(),
 113                             loader_controller());
 114
 115   FileError error = FILE_ERROR_FAILED;
 116   GURL next_link;
 117   scoped_ptr<std::vector<SearchResultInfo> > results;
 118
 119   operation.Search("\"no-match query\"", GURL(),
 120                    google_apis::test_util::CreateCopyResultCallback(
 121                        &error, &next_link, &results));
 122   content::RunAllBlockingPoolTasksUntilIdle();
 123
 124   EXPECT_EQ(FILE_ERROR_OK, error);
 125   EXPECT_TRUE(next_link.is_empty());
 126   EXPECT_EQ(0U, results->size());
 127 }
 128
 129 TEST_F(SearchOperationTest, Lock) {
 130   SearchOperation operation(blocking_task_runner(), scheduler(), metadata(),
 131                             loader_controller());
 132
 133   // Lock.
 134   scoped_ptr<base::ScopedClosureRunner> lock = loader_controller()->GetLock();
 135
 136   // Search does not return the result as long as lock is alive.
 137   FileError error = FILE_ERROR_FAILED;
 138   GURL next_link;
 139   scoped_ptr<std::vector<SearchResultInfo> > results;
 140
 141   operation.Search("\"Directory 1\"", GURL(),
 142                    google_apis::test_util::CreateCopyResultCallback(
 143                        &error, &next_link, &results));
 144   content::RunAllBlockingPoolTasksUntilIdle();
 145   EXPECT_EQ(FILE_ERROR_FAILED, error);
 146   EXPECT_FALSE(results);
 147
 148   // Unlock, this should resume the pending search.
 149   lock.reset();
 150   content::RunAllBlockingPoolTasksUntilIdle();
 151   EXPECT_EQ(FILE_ERROR_OK, error);
 152   ASSERT_TRUE(results);
 153   EXPECT_EQ(1u, results->size());
 154 }
 155
 156 }  // namespace file_system
 157 }  // namespace drive