1 // Copyright (c) 2012 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/media_galleries/media_galleries_test_util.h" 6 7 #include "base/base_paths.h" 8 #include "base/file_util.h" 9 #include "base/files/file_path.h" 10 #include "base/path_service.h" 11 #include "base/strings/stringprintf.h" 12 #include "chrome/browser/extensions/extension_prefs.h" 13 #include "chrome/browser/extensions/extension_service.h" 14 #include "chrome/browser/extensions/extension_system.h" 15 #include "chrome/browser/profiles/profile.h" 16 #include "chrome/common/chrome_paths.h" 17 #include "chrome/common/extensions/extension.h" 18 #include "chrome/common/extensions/extension_manifest_constants.h" 19 #include "testing/gtest/include/gtest/gtest.h" 20 21 namespace chrome { 22 23 scoped_refptr<extensions::Extension> AddMediaGalleriesApp( 24 const std::string& name, 25 const std::vector<std::string>& media_galleries_permissions, 26 Profile* profile) { 27 scoped_ptr<DictionaryValue> manifest(new DictionaryValue); 28 manifest->SetString(extension_manifest_keys::kName, name); 29 manifest->SetString(extension_manifest_keys::kVersion, "0.1"); 30 manifest->SetInteger(extension_manifest_keys::kManifestVersion, 2); 31 ListValue* background_script_list = new ListValue; 32 background_script_list->Append(Value::CreateStringValue("background.js")); 33 manifest->Set(extension_manifest_keys::kPlatformAppBackgroundScripts, 34 background_script_list); 35 36 ListValue* permission_detail_list = new ListValue; 37 for (size_t i = 0; i < media_galleries_permissions.size(); i++) 38 permission_detail_list->Append( 39 Value::CreateStringValue(media_galleries_permissions[i])); 40 DictionaryValue* media_galleries_permission = new DictionaryValue(); 41 media_galleries_permission->Set("mediaGalleries", permission_detail_list); 42 ListValue* permission_list = new ListValue; 43 permission_list->Append(media_galleries_permission); 44 manifest->Set(extension_manifest_keys::kPermissions, permission_list); 45 46 extensions::ExtensionPrefs* extension_prefs = 47 extensions::ExtensionPrefs::Get(profile); 48 base::FilePath path = extension_prefs->install_directory().AppendASCII(name); 49 std::string errors; 50 scoped_refptr<extensions::Extension> extension = 51 extensions::Extension::Create(path, extensions::Manifest::INTERNAL, 52 *manifest.get(), 53 extensions::Extension::NO_FLAGS, &errors); 54 EXPECT_TRUE(extension.get() != NULL) << errors; 55 EXPECT_TRUE(extensions::Extension::IdIsValid(extension->id())); 56 if (!extension.get() || !extensions::Extension::IdIsValid(extension->id())) 57 return NULL; 58 59 extension_prefs->OnExtensionInstalled( 60 extension.get(), 61 extensions::Extension::ENABLED, 62 extensions::Blacklist::NOT_BLACKLISTED, 63 syncer::StringOrdinal::CreateInitialOrdinal()); 64 ExtensionService* extension_service = 65 extensions::ExtensionSystem::Get(profile)->extension_service(); 66 extension_service->AddExtension(extension.get()); 67 extension_service->EnableExtension(extension->id()); 68 69 return extension; 70 } 71 72 EnsureMediaDirectoriesExists::EnsureMediaDirectoriesExists() 73 : num_galleries_(0) { 74 Init(); 75 } 76 77 EnsureMediaDirectoriesExists::~EnsureMediaDirectoriesExists() { 78 } 79 80 void EnsureMediaDirectoriesExists::Init() { 81 #if defined(OS_CHROMEOS) || defined(OS_ANDROID) 82 return; 83 #else 84 85 ASSERT_TRUE(fake_dir_.CreateUniqueTempDir()); 86 87 #if defined(OS_WIN) || defined(OS_MACOSX) 88 // This is to make sure the tests don't think iTunes is installed (unless 89 // we control it specifically). 90 appdir_override_.reset(new base::ScopedPathOverride( 91 base::DIR_APP_DATA, fake_dir_.path().AppendASCII("itunes"))); 92 #endif 93 94 music_override_.reset(new base::ScopedPathOverride( 95 chrome::DIR_USER_MUSIC, fake_dir_.path().AppendASCII("music"))); 96 pictures_override_.reset(new base::ScopedPathOverride( 97 chrome::DIR_USER_PICTURES, fake_dir_.path().AppendASCII("pictures"))); 98 video_override_.reset(new base::ScopedPathOverride( 99 chrome::DIR_USER_VIDEOS, fake_dir_.path().AppendASCII("videos"))); 100 num_galleries_ = 3; 101 #endif 102 } 103 104 } // namespace chrome 105