src/chrome/browser/chromeos/file_system_provider/registry_unittest.cc

   1 // Copyright 2014 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 "chrome/browser/chromeos/file_system_provider/registry.h"
   6
   7 #include <string>
   8 #include <vector>
   9
  10 #include "base/files/file_path.h"
  11 #include "base/memory/scoped_ptr.h"
  12 #include "chrome/browser/chromeos/file_system_provider/provided_file_system_info.h"
  13 #include "chrome/common/pref_names.h"
  14 #include "chrome/test/base/testing_browser_process.h"
  15 #include "chrome/test/base/testing_pref_service_syncable.h"
  16 #include "chrome/test/base/testing_profile.h"
  17 #include "chrome/test/base/testing_profile_manager.h"
  18 #include "components/user_prefs/user_prefs.h"
  19 #include "content/public/test/test_browser_thread_bundle.h"
  20 #include "testing/gtest/include/gtest/gtest.h"
  21
  22 namespace chromeos {
  23 namespace file_system_provider {
  24 namespace {
  25
  26 const char kTemporaryOrigin[] =
  27     "chrome-extension://abcabcabcabcabcabcabcabcabcabcabcabca/";
  28 const char kPersistentOrigin[] =
  29     "chrome-extension://efgefgefgefgefgefgefgefgefgefgefgefge/";
  30 const char kExtensionId[] = "mbflcebpggnecokmikipoihdbecnjfoj";
  31 const char kDisplayName[] = "Camera Pictures";
  32
  33 // The dot in the file system ID is there in order to check that saving to
  34 // preferences works correctly. File System ID is used as a key in
  35 // a base::DictionaryValue, so it has to be stored without path expansion.
  36 const char kFileSystemId[] = "camera/pictures/id .!@#$%^&*()_+";
  37
  38 // Stores a provided file system information in preferences together with a
  39 // fake watcher.
  40 void RememberFakeFileSystem(TestingProfile* profile,
  41                             const std::string& extension_id,
  42                             const std::string& file_system_id,
  43                             const std::string& display_name,
  44                             bool writable,
  45                             bool supports_notify_tag,
  46                             const Watcher& watcher) {
  47   // Warning. Updating this code means that backward compatibility may be
  48   // broken, what is unexpected and should be avoided.
  49   TestingPrefServiceSyncable* const pref_service =
  50       profile->GetTestingPrefService();
  51   ASSERT_TRUE(pref_service);
  52
  53   base::DictionaryValue extensions;
  54   base::DictionaryValue* const file_systems = new base::DictionaryValue();
  55   base::DictionaryValue* const file_system = new base::DictionaryValue();
  56   file_system->SetStringWithoutPathExpansion(kPrefKeyFileSystemId,
  57                                              kFileSystemId);
  58   file_system->SetStringWithoutPathExpansion(kPrefKeyDisplayName, kDisplayName);
  59   file_system->SetBooleanWithoutPathExpansion(kPrefKeyWritable, writable);
  60   file_system->SetBooleanWithoutPathExpansion(kPrefKeySupportsNotifyTag,
  61                                               supports_notify_tag);
  62   file_systems->SetWithoutPathExpansion(kFileSystemId, file_system);
  63   extensions.SetWithoutPathExpansion(kExtensionId, file_systems);
  64
  65   // Remember watchers.
  66   base::DictionaryValue* const watchers = new base::DictionaryValue();
  67   file_system->SetWithoutPathExpansion(kPrefKeyWatchers, watchers);
  68   base::DictionaryValue* const watcher_value = new base::DictionaryValue();
  69   watchers->SetWithoutPathExpansion(watcher.entry_path.value(), watcher_value);
  70   watcher_value->SetStringWithoutPathExpansion(kPrefKeyWatcherEntryPath,
  71                                                watcher.entry_path.value());
  72   watcher_value->SetBooleanWithoutPathExpansion(kPrefKeyWatcherRecursive,
  73                                                 watcher.recursive);
  74   watcher_value->SetStringWithoutPathExpansion(kPrefKeyWatcherLastTag,
  75                                                watcher.last_tag);
  76   base::ListValue* const persistent_origins_value = new base::ListValue();
  77   watcher_value->SetWithoutPathExpansion(kPrefKeyWatcherPersistentOrigins,
  78                                          persistent_origins_value);
  79   for (const auto& subscriber_it : watcher.subscribers) {
  80     if (subscriber_it.second.persistent)
  81       persistent_origins_value->AppendString(subscriber_it.first.spec());
  82   }
  83
  84   pref_service->Set(prefs::kFileSystemProviderMounted, extensions);
  85 }
  86
  87 }  // namespace
  88
  89 class FileSystemProviderRegistryTest : public testing::Test {
  90  protected:
  91   FileSystemProviderRegistryTest() : profile_(NULL) {}
  92
  93   virtual ~FileSystemProviderRegistryTest() {}
  94
  95   virtual void SetUp() override {
  96     profile_manager_.reset(
  97         new TestingProfileManager(TestingBrowserProcess::GetGlobal()));
  98     ASSERT_TRUE(profile_manager_->SetUp());
  99     profile_ = profile_manager_->CreateTestingProfile("test-user@example.com");
 100     registry_.reset(new Registry(profile_));
 101     fake_watcher_.entry_path = base::FilePath(FILE_PATH_LITERAL("/a/b/c"));
 102     fake_watcher_.recursive = true;
 103     fake_watcher_.subscribers[GURL(kTemporaryOrigin)].origin =
 104         GURL(kTemporaryOrigin);
 105     fake_watcher_.subscribers[GURL(kTemporaryOrigin)].persistent = false;
 106     fake_watcher_.subscribers[GURL(kPersistentOrigin)].origin =
 107         GURL(kPersistentOrigin);
 108     fake_watcher_.subscribers[GURL(kPersistentOrigin)].persistent = true;
 109     fake_watcher_.last_tag = "hello-world";
 110   }
 111
 112   content::TestBrowserThreadBundle thread_bundle_;
 113   scoped_ptr<TestingProfileManager> profile_manager_;
 114   TestingProfile* profile_;
 115   scoped_ptr<RegistryInterface> registry_;
 116   Watcher fake_watcher_;
 117 };
 118
 119 TEST_F(FileSystemProviderRegistryTest, RestoreFileSystems) {
 120   // Create a fake entry in the preferences.
 121   RememberFakeFileSystem(profile_,
 122                          kExtensionId,
 123                          kFileSystemId,
 124                          kDisplayName,
 125                          true /* writable */,
 126                          true /* supports_notify_tag */,
 127                          fake_watcher_);
 128
 129   scoped_ptr<RegistryInterface::RestoredFileSystems> restored_file_systems =
 130       registry_->RestoreFileSystems(kExtensionId);
 131
 132   ASSERT_EQ(1u, restored_file_systems->size());
 133   const RegistryInterface::RestoredFileSystem& restored_file_system =
 134       restored_file_systems->at(0);
 135   EXPECT_EQ(kExtensionId, restored_file_system.extension_id);
 136   EXPECT_EQ(kFileSystemId, restored_file_system.options.file_system_id);
 137   EXPECT_EQ(kDisplayName, restored_file_system.options.display_name);
 138   EXPECT_TRUE(restored_file_system.options.writable);
 139   EXPECT_TRUE(restored_file_system.options.supports_notify_tag);
 140
 141   ASSERT_EQ(1u, restored_file_system.watchers.size());
 142   const auto& restored_watcher_it = restored_file_system.watchers.find(
 143       WatcherKey(fake_watcher_.entry_path, fake_watcher_.recursive));
 144   ASSERT_NE(restored_file_system.watchers.end(), restored_watcher_it);
 145
 146   EXPECT_EQ(fake_watcher_.entry_path, restored_watcher_it->second.entry_path);
 147   EXPECT_EQ(fake_watcher_.recursive, restored_watcher_it->second.recursive);
 148   EXPECT_EQ(fake_watcher_.last_tag, restored_watcher_it->second.last_tag);
 149 }
 150
 151 TEST_F(FileSystemProviderRegistryTest, RememberFileSystem) {
 152   MountOptions options(kFileSystemId, kDisplayName);
 153   options.writable = true;
 154   options.supports_notify_tag = true;
 155
 156   ProvidedFileSystemInfo file_system_info(
 157       kExtensionId, options, base::FilePath(FILE_PATH_LITERAL("/a/b/c")));
 158
 159   Watchers watchers;
 160   watchers[WatcherKey(fake_watcher_.entry_path, fake_watcher_.recursive)] =
 161       fake_watcher_;
 162
 163   registry_->RememberFileSystem(file_system_info, watchers);
 164
 165   TestingPrefServiceSyncable* const pref_service =
 166       profile_->GetTestingPrefService();
 167   ASSERT_TRUE(pref_service);
 168
 169   const base::DictionaryValue* const extensions =
 170       pref_service->GetDictionary(prefs::kFileSystemProviderMounted);
 171   ASSERT_TRUE(extensions);
 172
 173   const base::DictionaryValue* file_systems = NULL;
 174   ASSERT_TRUE(extensions->GetDictionaryWithoutPathExpansion(kExtensionId,
 175                                                             &file_systems));
 176   EXPECT_EQ(1u, file_systems->size());
 177
 178   const base::Value* file_system_value = NULL;
 179   const base::DictionaryValue* file_system = NULL;
 180   ASSERT_TRUE(
 181       file_systems->GetWithoutPathExpansion(kFileSystemId, &file_system_value));
 182   ASSERT_TRUE(file_system_value->GetAsDictionary(&file_system));
 183
 184   std::string file_system_id;
 185   EXPECT_TRUE(file_system->GetStringWithoutPathExpansion(kPrefKeyFileSystemId,
 186                                                          &file_system_id));
 187   EXPECT_EQ(kFileSystemId, file_system_id);
 188
 189   std::string display_name;
 190   EXPECT_TRUE(file_system->GetStringWithoutPathExpansion(kPrefKeyDisplayName,
 191                                                          &display_name));
 192   EXPECT_EQ(kDisplayName, display_name);
 193
 194   bool writable = false;
 195   EXPECT_TRUE(
 196       file_system->GetBooleanWithoutPathExpansion(kPrefKeyWritable, &writable));
 197   EXPECT_TRUE(writable);
 198
 199   bool supports_notify_tag = false;
 200   EXPECT_TRUE(file_system->GetBooleanWithoutPathExpansion(
 201       kPrefKeySupportsNotifyTag, &supports_notify_tag));
 202   EXPECT_TRUE(supports_notify_tag);
 203
 204   const base::DictionaryValue* watchers_value = NULL;
 205   ASSERT_TRUE(file_system->GetDictionaryWithoutPathExpansion(kPrefKeyWatchers,
 206                                                              &watchers_value));
 207
 208   const base::DictionaryValue* watcher = NULL;
 209   ASSERT_TRUE(watchers_value->GetDictionaryWithoutPathExpansion(
 210       fake_watcher_.entry_path.value(), &watcher));
 211
 212   std::string entry_path;
 213   EXPECT_TRUE(watcher->GetStringWithoutPathExpansion(kPrefKeyWatcherEntryPath,
 214                                                      &entry_path));
 215   EXPECT_EQ(fake_watcher_.entry_path.value(), entry_path);
 216
 217   bool recursive = false;
 218   EXPECT_TRUE(watcher->GetBooleanWithoutPathExpansion(kPrefKeyWatcherRecursive,
 219                                                       &recursive));
 220   EXPECT_EQ(fake_watcher_.recursive, recursive);
 221
 222   std::string last_tag;
 223   EXPECT_TRUE(watcher->GetStringWithoutPathExpansion(kPrefKeyWatcherLastTag,
 224                                                      &last_tag));
 225   EXPECT_EQ(fake_watcher_.last_tag, last_tag);
 226
 227   const base::ListValue* persistent_origins = NULL;
 228   ASSERT_TRUE(watcher->GetListWithoutPathExpansion(
 229       kPrefKeyWatcherPersistentOrigins, &persistent_origins));
 230   ASSERT_GT(fake_watcher_.subscribers.size(), persistent_origins->GetSize());
 231   ASSERT_EQ(1u, persistent_origins->GetSize());
 232   std::string persistent_origin;
 233   EXPECT_TRUE(persistent_origins->GetString(0, &persistent_origin));
 234   const auto& fake_subscriber_it =
 235       fake_watcher_.subscribers.find(GURL(persistent_origin));
 236   ASSERT_NE(fake_watcher_.subscribers.end(), fake_subscriber_it);
 237   EXPECT_TRUE(fake_subscriber_it->second.persistent);
 238 }
 239
 240 TEST_F(FileSystemProviderRegistryTest, ForgetFileSystem) {
 241   // Create a fake file systems in the preferences.
 242   RememberFakeFileSystem(profile_,
 243                          kExtensionId,
 244                          kFileSystemId,
 245                          kDisplayName,
 246                          true /* writable */,
 247                          true /* supports_notify_tag */,
 248                          fake_watcher_);
 249
 250   registry_->ForgetFileSystem(kExtensionId, kFileSystemId);
 251
 252   TestingPrefServiceSyncable* const pref_service =
 253       profile_->GetTestingPrefService();
 254   ASSERT_TRUE(pref_service);
 255
 256   const base::DictionaryValue* const extensions =
 257       pref_service->GetDictionary(prefs::kFileSystemProviderMounted);
 258   ASSERT_TRUE(extensions);
 259
 260   const base::DictionaryValue* file_systems = NULL;
 261   EXPECT_FALSE(extensions->GetDictionaryWithoutPathExpansion(kExtensionId,
 262                                                              &file_systems));
 263 }
 264
 265 TEST_F(FileSystemProviderRegistryTest, UpdateWatcherTag) {
 266   MountOptions options(kFileSystemId, kDisplayName);
 267   options.writable = true;
 268   options.supports_notify_tag = true;
 269
 270   ProvidedFileSystemInfo file_system_info(
 271       kExtensionId, options, base::FilePath(FILE_PATH_LITERAL("/a/b/c")));
 272
 273   Watchers watchers;
 274   watchers[WatcherKey(fake_watcher_.entry_path, fake_watcher_.recursive)] =
 275       fake_watcher_;
 276
 277   registry_->RememberFileSystem(file_system_info, watchers);
 278
 279   fake_watcher_.last_tag = "updated-tag";
 280   registry_->UpdateWatcherTag(file_system_info, fake_watcher_);
 281
 282   TestingPrefServiceSyncable* const pref_service =
 283       profile_->GetTestingPrefService();
 284   ASSERT_TRUE(pref_service);
 285
 286   const base::DictionaryValue* const extensions =
 287       pref_service->GetDictionary(prefs::kFileSystemProviderMounted);
 288   ASSERT_TRUE(extensions);
 289
 290   const base::DictionaryValue* file_systems = NULL;
 291   ASSERT_TRUE(extensions->GetDictionaryWithoutPathExpansion(kExtensionId,
 292                                                             &file_systems));
 293   EXPECT_EQ(1u, file_systems->size());
 294
 295   const base::Value* file_system_value = NULL;
 296   const base::DictionaryValue* file_system = NULL;
 297   ASSERT_TRUE(
 298       file_systems->GetWithoutPathExpansion(kFileSystemId, &file_system_value));
 299   ASSERT_TRUE(file_system_value->GetAsDictionary(&file_system));
 300
 301   const base::DictionaryValue* watchers_value = NULL;
 302   ASSERT_TRUE(file_system->GetDictionaryWithoutPathExpansion(kPrefKeyWatchers,
 303                                                              &watchers_value));
 304
 305   const base::DictionaryValue* watcher = NULL;
 306   ASSERT_TRUE(watchers_value->GetDictionaryWithoutPathExpansion(
 307       fake_watcher_.entry_path.value(), &watcher));
 308
 309   std::string last_tag;
 310   EXPECT_TRUE(watcher->GetStringWithoutPathExpansion(kPrefKeyWatcherLastTag,
 311                                                      &last_tag));
 312   EXPECT_EQ(fake_watcher_.last_tag, last_tag);
 313 }
 314
 315 }  // namespace file_system_provider
 316 }  // namespace chromeos