| // 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 "chrome/browser/media_galleries/media_galleries_test_util.h" |
| |
| #if defined(OS_WIN) |
| #include <windows.h> |
| #endif |
| |
| #include "base/base_paths.h" |
| #include "base/file_util.h" |
| #include "base/files/file_path.h" |
| #include "base/path_service.h" |
| #include "base/strings/stringprintf.h" |
| #include "chrome/browser/extensions/extension_service.h" |
| #include "chrome/browser/media_galleries/fileapi/picasa_finder.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/common/chrome_paths.h" |
| #include "extensions/browser/extension_prefs.h" |
| #include "extensions/browser/extension_system.h" |
| #include "extensions/common/extension.h" |
| #include "extensions/common/manifest_constants.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| #if defined(OS_MACOSX) |
| #include "base/mac/foundation_util.h" |
| #include "base/strings/sys_string_conversions.h" |
| #include "chrome/browser/media_galleries/fileapi/iapps_finder_impl.h" |
| #include "components/policy/core/common/preferences_mock_mac.h" |
| #endif // OS_MACOSX |
| |
| #if defined(OS_WIN) |
| #include "base/test/test_reg_util_win.h" |
| #include "base/win/registry.h" |
| #endif // OS_WIN |
| |
| scoped_refptr<extensions::Extension> AddMediaGalleriesApp( |
| const std::string& name, |
| const std::vector<std::string>& media_galleries_permissions, |
| Profile* profile) { |
| scoped_ptr<base::DictionaryValue> manifest(new base::DictionaryValue); |
| manifest->SetString(extensions::manifest_keys::kName, name); |
| manifest->SetString(extensions::manifest_keys::kVersion, "0.1"); |
| manifest->SetInteger(extensions::manifest_keys::kManifestVersion, 2); |
| base::ListValue* background_script_list = new base::ListValue; |
| background_script_list->Append( |
| base::Value::CreateStringValue("background.js")); |
| manifest->Set(extensions::manifest_keys::kPlatformAppBackgroundScripts, |
| background_script_list); |
| |
| base::ListValue* permission_detail_list = new base::ListValue; |
| for (size_t i = 0; i < media_galleries_permissions.size(); i++) |
| permission_detail_list->Append( |
| base::Value::CreateStringValue(media_galleries_permissions[i])); |
| base::DictionaryValue* media_galleries_permission = |
| new base::DictionaryValue(); |
| media_galleries_permission->Set("mediaGalleries", permission_detail_list); |
| base::ListValue* permission_list = new base::ListValue; |
| permission_list->Append(media_galleries_permission); |
| manifest->Set(extensions::manifest_keys::kPermissions, permission_list); |
| |
| extensions::ExtensionPrefs* extension_prefs = |
| extensions::ExtensionPrefs::Get(profile); |
| base::FilePath path = extension_prefs->install_directory().AppendASCII(name); |
| std::string errors; |
| scoped_refptr<extensions::Extension> extension = |
| extensions::Extension::Create(path, extensions::Manifest::INTERNAL, |
| *manifest.get(), |
| extensions::Extension::NO_FLAGS, &errors); |
| EXPECT_TRUE(extension.get() != NULL) << errors; |
| EXPECT_TRUE(extensions::Extension::IdIsValid(extension->id())); |
| if (!extension.get() || !extensions::Extension::IdIsValid(extension->id())) |
| return NULL; |
| |
| extension_prefs->OnExtensionInstalled( |
| extension.get(), |
| extensions::Extension::ENABLED, |
| syncer::StringOrdinal::CreateInitialOrdinal(), |
| std::string()); |
| ExtensionService* extension_service = |
| extensions::ExtensionSystem::Get(profile)->extension_service(); |
| extension_service->AddExtension(extension.get()); |
| extension_service->EnableExtension(extension->id()); |
| |
| return extension; |
| } |
| |
| EnsureMediaDirectoriesExists::EnsureMediaDirectoriesExists() |
| : num_galleries_(0) { |
| Init(); |
| } |
| |
| EnsureMediaDirectoriesExists::~EnsureMediaDirectoriesExists() { |
| #if defined(OS_MACOSX) |
| iapps::SetMacPreferencesForTesting(NULL); |
| picasa::SetMacPreferencesForTesting(NULL); |
| #endif // OS_MACOSX |
| } |
| |
| base::FilePath EnsureMediaDirectoriesExists::GetFakeAppDataPath() const { |
| DCHECK(fake_dir_.IsValid()); |
| return fake_dir_.path().AppendASCII("appdata"); |
| } |
| |
| #if defined(OS_WIN) |
| base::FilePath EnsureMediaDirectoriesExists::GetFakeLocalAppDataPath() const { |
| DCHECK(fake_dir_.IsValid()); |
| return fake_dir_.path().AppendASCII("localappdata"); |
| } |
| |
| void EnsureMediaDirectoriesExists::SetCustomPicasaAppDataPath( |
| const base::FilePath& path) { |
| base::win::RegKey key(HKEY_CURRENT_USER, picasa::kPicasaRegistryPath, |
| KEY_SET_VALUE); |
| key.WriteValue(picasa::kPicasaRegistryAppDataPathKey, path.value().c_str()); |
| } |
| #endif // OS_WIN |
| |
| #if defined(OS_MACOSX) |
| void EnsureMediaDirectoriesExists::SetCustomPicasaAppDataPath( |
| const base::FilePath& path) { |
| mac_preferences_->AddTestItem( |
| base::mac::NSToCFCast(picasa::kPicasaAppDataPathMacPreferencesKey), |
| base::SysUTF8ToNSString(path.value()), |
| false); |
| } |
| #endif // OS_MACOSX |
| |
| #if defined(OS_WIN) || defined(OS_MACOSX) |
| base::FilePath |
| EnsureMediaDirectoriesExists::GetFakePicasaFoldersRootPath() const { |
| DCHECK(fake_dir_.IsValid()); |
| return fake_dir_.path().AppendASCII("picasa_folders"); |
| } |
| #endif // OS_WIN || OS_MACOSX |
| |
| #if defined(OS_MACOSX) |
| base::FilePath EnsureMediaDirectoriesExists::GetFakeITunesRootPath() const { |
| DCHECK(fake_dir_.IsValid()); |
| return fake_dir_.path().AppendASCII("itunes"); |
| } |
| |
| base::FilePath EnsureMediaDirectoriesExists::GetFakeIPhotoRootPath() const { |
| DCHECK(fake_dir_.IsValid()); |
| return fake_dir_.path().AppendASCII("iphoto"); |
| } |
| #endif // OS_MACOSX |
| |
| void EnsureMediaDirectoriesExists::Init() { |
| #if defined(OS_CHROMEOS) || defined(OS_ANDROID) |
| return; |
| #else |
| |
| ASSERT_TRUE(fake_dir_.CreateUniqueTempDir()); |
| |
| #if defined(OS_WIN) || defined(OS_MACOSX) |
| // This is to control whether or not tests think iTunes (on Windows) and |
| // Picasa are installed. |
| app_data_override_.reset(new base::ScopedPathOverride( |
| base::DIR_APP_DATA, GetFakeAppDataPath())); |
| #endif // OS_WIN || OS_MACOSX |
| |
| #if defined(OS_WIN) |
| // Picasa on Windows is by default in the DIR_LOCAL_APP_DATA directory. |
| local_app_data_override_.reset(new base::ScopedPathOverride( |
| base::DIR_LOCAL_APP_DATA, GetFakeLocalAppDataPath())); |
| // Picasa also looks in the registry for an alternate path. |
| registry_override_.OverrideRegistry(HKEY_CURRENT_USER, L"hkcu_picasa"); |
| #endif // OS_WIN |
| |
| #if defined(OS_MACOSX) |
| mac_preferences_.reset(new MockPreferences); |
| |
| // iTunes override. |
| base::FilePath itunes_xml = |
| GetFakeITunesRootPath().AppendASCII("iTunes Library.xml"); |
| mac_preferences_->AddTestItem( |
| base::mac::NSToCFCast(iapps::kITunesRecentDatabasePathsKey), |
| base::mac::NSToCFCast(iapps::NSArrayFromFilePath(itunes_xml)), |
| false); |
| |
| // iPhoto override. |
| base::FilePath iphoto_xml = |
| GetFakeIPhotoRootPath().AppendASCII("AlbumData.xml"); |
| mac_preferences_->AddTestItem( |
| base::mac::NSToCFCast(iapps::kIPhotoRecentDatabasesKey), |
| base::mac::NSToCFCast(iapps::NSArrayFromFilePath(iphoto_xml)), |
| false); |
| |
| iapps::SetMacPreferencesForTesting(mac_preferences_.get()); |
| picasa::SetMacPreferencesForTesting(mac_preferences_.get()); |
| #endif // OS_MACOSX |
| |
| music_override_.reset(new base::ScopedPathOverride( |
| chrome::DIR_USER_MUSIC, fake_dir_.path().AppendASCII("music"))); |
| pictures_override_.reset(new base::ScopedPathOverride( |
| chrome::DIR_USER_PICTURES, fake_dir_.path().AppendASCII("pictures"))); |
| video_override_.reset(new base::ScopedPathOverride( |
| chrome::DIR_USER_VIDEOS, fake_dir_.path().AppendASCII("videos"))); |
| num_galleries_ = 3; |
| #endif // OS_CHROMEOS || OS_ANDROID |
| } |
| |
| base::FilePath MakeMediaGalleriesTestingPath(const std::string& dir) { |
| #if defined(OS_WIN) |
| return base::FilePath(FILE_PATH_LITERAL("C:\\")).AppendASCII(dir); |
| #elif defined(OS_POSIX) |
| return base::FilePath(FILE_PATH_LITERAL("/")).Append(dir); |
| #else |
| NOTREACHED(); |
| #endif |
| } |