Home | History | Annotate | Download | only in base
      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 "base/path_service.h"
      6 
      7 #include "base/basictypes.h"
      8 #include "base/file_util.h"
      9 #include "base/files/file_path.h"
     10 #include "base/files/scoped_temp_dir.h"
     11 #include "base/strings/string_util.h"
     12 #include "build/build_config.h"
     13 #include "testing/gtest/include/gtest/gtest-spi.h"
     14 #include "testing/gtest/include/gtest/gtest.h"
     15 #include "testing/platform_test.h"
     16 
     17 #if defined(OS_WIN)
     18 #include <userenv.h>
     19 #include "base/win/windows_version.h"
     20 // userenv.dll is required for GetDefaultUserProfileDirectory().
     21 #pragma comment(lib, "userenv.lib")
     22 #endif
     23 
     24 namespace {
     25 
     26 // Returns true if PathService::Get returns true and sets the path parameter
     27 // to non-empty for the given PathService::DirType enumeration value.
     28 bool ReturnsValidPath(int dir_type) {
     29   base::FilePath path;
     30   bool result = PathService::Get(dir_type, &path);
     31 
     32   // Some paths might not exist on some platforms in which case confirming
     33   // |result| is true and !path.empty() is the best we can do.
     34   bool check_path_exists = true;
     35 #if defined(OS_POSIX)
     36   // If chromium has never been started on this account, the cache path may not
     37   // exist.
     38   if (dir_type == base::DIR_CACHE)
     39     check_path_exists = false;
     40 #endif
     41 #if defined(OS_LINUX)
     42   // On the linux try-bots: a path is returned (e.g. /home/chrome-bot/Desktop),
     43   // but it doesn't exist.
     44   if (dir_type == base::DIR_USER_DESKTOP)
     45     check_path_exists = false;
     46 #endif
     47 #if defined(OS_WIN)
     48   if (dir_type == base::DIR_DEFAULT_USER_QUICK_LAUNCH) {
     49     // On Windows XP, the Quick Launch folder for the "Default User" doesn't
     50     // exist by default. At least confirm that the path returned begins with the
     51     // Default User's profile path.
     52     if (base::win::GetVersion() < base::win::VERSION_VISTA) {
     53       wchar_t default_profile_path[MAX_PATH];
     54       DWORD size = arraysize(default_profile_path);
     55       return (result &&
     56               ::GetDefaultUserProfileDirectory(default_profile_path, &size) &&
     57               StartsWith(path.value(), default_profile_path, false));
     58     }
     59   } else if (dir_type == base::DIR_TASKBAR_PINS) {
     60     // There is no pinned-to-taskbar shortcuts prior to Win7.
     61     if (base::win::GetVersion() < base::win::VERSION_WIN7)
     62       check_path_exists = false;
     63   }
     64 #endif
     65 #if defined(OS_MACOSX)
     66   if (dir_type != base::DIR_EXE && dir_type != base::DIR_MODULE &&
     67       dir_type != base::FILE_EXE && dir_type != base::FILE_MODULE) {
     68     if (path.ReferencesParent())
     69       return false;
     70   }
     71 #else
     72   if (path.ReferencesParent())
     73     return false;
     74 #endif
     75   return result && !path.empty() && (!check_path_exists ||
     76                                      base::PathExists(path));
     77 }
     78 
     79 #if defined(OS_WIN)
     80 // Function to test any directory keys that are not supported on some versions
     81 // of Windows. Checks that the function fails and that the returned path is
     82 // empty.
     83 bool ReturnsInvalidPath(int dir_type) {
     84   base::FilePath path;
     85   bool result = PathService::Get(dir_type, &path);
     86   return !result && path.empty();
     87 }
     88 #endif
     89 
     90 }  // namespace
     91 
     92 // On the Mac this winds up using some autoreleased objects, so we need to
     93 // be a PlatformTest.
     94 typedef PlatformTest PathServiceTest;
     95 
     96 // Test that all PathService::Get calls return a value and a true result
     97 // in the development environment.  (This test was created because a few
     98 // later changes to Get broke the semantics of the function and yielded the
     99 // correct value while returning false.)
    100 TEST_F(PathServiceTest, Get) {
    101   for (int key = base::PATH_START + 1; key < base::PATH_END; ++key) {
    102 #if defined(OS_ANDROID)
    103     if (key == base::FILE_MODULE || key == base::DIR_USER_DESKTOP ||
    104         key == base::DIR_HOME)
    105       continue;  // Android doesn't implement these.
    106 #elif defined(OS_IOS)
    107     if (key == base::DIR_USER_DESKTOP)
    108       continue;  // iOS doesn't implement DIR_USER_DESKTOP;
    109 #endif
    110     EXPECT_PRED1(ReturnsValidPath, key);
    111   }
    112 #if defined(OS_WIN)
    113   for (int key = base::PATH_WIN_START + 1; key < base::PATH_WIN_END; ++key) {
    114     bool valid = true;
    115     switch(key) {
    116       case base::DIR_LOCAL_APP_DATA_LOW:
    117         // DIR_LOCAL_APP_DATA_LOW is not supported prior Vista and is expected
    118         // to fail.
    119         valid = base::win::GetVersion() >= base::win::VERSION_VISTA;
    120         break;
    121       case base::DIR_APP_SHORTCUTS:
    122         // DIR_APP_SHORTCUTS is not supported prior Windows 8 and is expected to
    123         // fail.
    124         valid = base::win::GetVersion() >= base::win::VERSION_WIN8;
    125         break;
    126     }
    127 
    128     if (valid)
    129       EXPECT_TRUE(ReturnsValidPath(key)) << key;
    130     else
    131       EXPECT_TRUE(ReturnsInvalidPath(key)) << key;
    132   }
    133 #elif defined(OS_MACOSX)
    134   for (int key = base::PATH_MAC_START + 1; key < base::PATH_MAC_END; ++key) {
    135     EXPECT_PRED1(ReturnsValidPath, key);
    136   }
    137 #elif defined(OS_ANDROID)
    138   for (int key = base::PATH_ANDROID_START + 1; key < base::PATH_ANDROID_END;
    139        ++key) {
    140     EXPECT_PRED1(ReturnsValidPath, key);
    141   }
    142 #elif defined(OS_POSIX)
    143   for (int key = base::PATH_POSIX_START + 1; key < base::PATH_POSIX_END;
    144        ++key) {
    145     EXPECT_PRED1(ReturnsValidPath, key);
    146   }
    147 #endif
    148 }
    149 
    150 // Test that all versions of the Override function of PathService do what they
    151 // are supposed to do.
    152 TEST_F(PathServiceTest, Override) {
    153   int my_special_key = 666;
    154   base::ScopedTempDir temp_dir;
    155   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
    156   base::FilePath fake_cache_dir(temp_dir.path().AppendASCII("cache"));
    157   // PathService::Override should always create the path provided if it doesn't
    158   // exist.
    159   EXPECT_TRUE(PathService::Override(my_special_key, fake_cache_dir));
    160   EXPECT_TRUE(base::PathExists(fake_cache_dir));
    161 
    162   base::FilePath fake_cache_dir2(temp_dir.path().AppendASCII("cache2"));
    163   // PathService::OverrideAndCreateIfNeeded should obey the |create| parameter.
    164   PathService::OverrideAndCreateIfNeeded(my_special_key,
    165                                          fake_cache_dir2,
    166                                          false,
    167                                          false);
    168   EXPECT_FALSE(base::PathExists(fake_cache_dir2));
    169   EXPECT_TRUE(PathService::OverrideAndCreateIfNeeded(my_special_key,
    170                                                      fake_cache_dir2,
    171                                                      false,
    172                                                      true));
    173   EXPECT_TRUE(base::PathExists(fake_cache_dir2));
    174 
    175 #if defined(OS_POSIX)
    176   base::FilePath non_existent(
    177       base::MakeAbsoluteFilePath(temp_dir.path()).AppendASCII("non_existent"));
    178   EXPECT_TRUE(non_existent.IsAbsolute());
    179   EXPECT_FALSE(base::PathExists(non_existent));
    180 #if !defined(OS_ANDROID)
    181   // This fails because MakeAbsoluteFilePath fails for non-existent files.
    182   // Earlier versions of Bionic libc don't fail for non-existent files, so
    183   // skip this check on Android.
    184   EXPECT_FALSE(PathService::OverrideAndCreateIfNeeded(my_special_key,
    185                                                       non_existent,
    186                                                       false,
    187                                                       false));
    188 #endif
    189   // This works because indicating that |non_existent| is absolute skips the
    190   // internal MakeAbsoluteFilePath call.
    191   EXPECT_TRUE(PathService::OverrideAndCreateIfNeeded(my_special_key,
    192                                                      non_existent,
    193                                                      true,
    194                                                      false));
    195   // Check that the path has been overridden and no directory was created.
    196   EXPECT_FALSE(base::PathExists(non_existent));
    197   base::FilePath path;
    198   EXPECT_TRUE(PathService::Get(my_special_key, &path));
    199   EXPECT_EQ(non_existent, path);
    200 #endif
    201 }
    202 
    203 // Check if multiple overrides can co-exist.
    204 TEST_F(PathServiceTest, OverrideMultiple) {
    205   int my_special_key = 666;
    206   base::ScopedTempDir temp_dir;
    207   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
    208   base::FilePath fake_cache_dir1(temp_dir.path().AppendASCII("1"));
    209   EXPECT_TRUE(PathService::Override(my_special_key, fake_cache_dir1));
    210   EXPECT_TRUE(base::PathExists(fake_cache_dir1));
    211   ASSERT_EQ(1, base::WriteFile(fake_cache_dir1.AppendASCII("t1"), ".", 1));
    212 
    213   base::FilePath fake_cache_dir2(temp_dir.path().AppendASCII("2"));
    214   EXPECT_TRUE(PathService::Override(my_special_key + 1, fake_cache_dir2));
    215   EXPECT_TRUE(base::PathExists(fake_cache_dir2));
    216   ASSERT_EQ(1, base::WriteFile(fake_cache_dir2.AppendASCII("t2"), ".", 1));
    217 
    218   base::FilePath result;
    219   EXPECT_TRUE(PathService::Get(my_special_key, &result));
    220   // Override might have changed the path representation but our test file
    221   // should be still there.
    222   EXPECT_TRUE(base::PathExists(result.AppendASCII("t1")));
    223   EXPECT_TRUE(PathService::Get(my_special_key + 1, &result));
    224   EXPECT_TRUE(base::PathExists(result.AppendASCII("t2")));
    225 }
    226 
    227 TEST_F(PathServiceTest, RemoveOverride) {
    228   // Before we start the test we have to call RemoveOverride at least once to
    229   // clear any overrides that might have been left from other tests.
    230   PathService::RemoveOverride(base::DIR_TEMP);
    231 
    232   base::FilePath original_user_data_dir;
    233   EXPECT_TRUE(PathService::Get(base::DIR_TEMP, &original_user_data_dir));
    234   EXPECT_FALSE(PathService::RemoveOverride(base::DIR_TEMP));
    235 
    236   base::ScopedTempDir temp_dir;
    237   ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
    238   EXPECT_TRUE(PathService::Override(base::DIR_TEMP, temp_dir.path()));
    239   base::FilePath new_user_data_dir;
    240   EXPECT_TRUE(PathService::Get(base::DIR_TEMP, &new_user_data_dir));
    241   EXPECT_NE(original_user_data_dir, new_user_data_dir);
    242 
    243   EXPECT_TRUE(PathService::RemoveOverride(base::DIR_TEMP));
    244   EXPECT_TRUE(PathService::Get(base::DIR_TEMP, &new_user_data_dir));
    245   EXPECT_EQ(original_user_data_dir, new_user_data_dir);
    246 }
    247