| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/environment.h" | 5 #include "base/environment.h" |
| 6 #include "base/memory/scoped_ptr.h" | 6 #include "base/memory/scoped_ptr.h" |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
| 8 #include "testing/platform_test.h" | 8 #include "testing/platform_test.h" |
| 9 | 9 |
| 10 typedef PlatformTest EnvironmentTest; | 10 typedef PlatformTest EnvironmentTest; |
| 11 | 11 |
| 12 namespace base { | 12 namespace base { |
| 13 | 13 |
| 14 TEST_F(EnvironmentTest, GetVar) { | 14 TEST_F(EnvironmentTest, GetVar) { |
| 15 // Every setup should have non-empty PATH... | 15 // Every setup should have non-empty PATH... |
| 16 scoped_ptr<Environment> env(Environment::Create()); | 16 scoped_ptr<Environment> env(Environment::Create()); |
| 17 std::string env_value; | 17 std::string env_value; |
| 18 EXPECT_TRUE(env->GetVar("PATH", &env_value)); | 18 EXPECT_TRUE(env->GetVar("PATH", &env_value)); |
| 19 EXPECT_NE(env_value, ""); | 19 EXPECT_NE(env_value, ""); |
| 20 } | 20 } |
| 21 | 21 |
| 22 TEST_F(EnvironmentTest, GetVarReverse) { | 22 TEST_F(EnvironmentTest, GetVarReverse) { |
| 23 scoped_ptr<Environment> env(Environment::Create()); | 23 scoped_ptr<Environment> env(Environment::Create()); |
| 24 const char* kFooUpper = "FOO"; | 24 const char kFooUpper[] = "FOO"; |
| 25 const char* kFooLower = "foo"; | 25 const char kFooLower[] = "foo"; |
| 26 | 26 |
| 27 // Set a variable in UPPER case. | 27 // Set a variable in UPPER case. |
| 28 EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower)); | 28 EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower)); |
| 29 | 29 |
| 30 // And then try to get this variable passing the lower case. | 30 // And then try to get this variable passing the lower case. |
| 31 std::string env_value; | 31 std::string env_value; |
| 32 EXPECT_TRUE(env->GetVar(kFooLower, &env_value)); | 32 EXPECT_TRUE(env->GetVar(kFooLower, &env_value)); |
| 33 | 33 |
| 34 EXPECT_STREQ(env_value.c_str(), kFooLower); | 34 EXPECT_STREQ(env_value.c_str(), kFooLower); |
| 35 | 35 |
| 36 EXPECT_TRUE(env->UnSetVar(kFooUpper)); | 36 EXPECT_TRUE(env->UnSetVar(kFooUpper)); |
| 37 | 37 |
| 38 const char* kBar = "bar"; | 38 const char kBar[] = "bar"; |
| 39 // Now do the opposite, set the variable in the lower case. | 39 // Now do the opposite, set the variable in the lower case. |
| 40 EXPECT_TRUE(env->SetVar(kFooLower, kBar)); | 40 EXPECT_TRUE(env->SetVar(kFooLower, kBar)); |
| 41 | 41 |
| 42 // And then try to get this variable passing the UPPER case. | 42 // And then try to get this variable passing the UPPER case. |
| 43 EXPECT_TRUE(env->GetVar(kFooUpper, &env_value)); | 43 EXPECT_TRUE(env->GetVar(kFooUpper, &env_value)); |
| 44 | 44 |
| 45 EXPECT_STREQ(env_value.c_str(), kBar); | 45 EXPECT_STREQ(env_value.c_str(), kBar); |
| 46 | 46 |
| 47 EXPECT_TRUE(env->UnSetVar(kFooLower)); | 47 EXPECT_TRUE(env->UnSetVar(kFooLower)); |
| 48 } | 48 } |
| 49 | 49 |
| 50 TEST_F(EnvironmentTest, HasVar) { | 50 TEST_F(EnvironmentTest, HasVar) { |
| 51 // Every setup should have PATH... | 51 // Every setup should have PATH... |
| 52 scoped_ptr<Environment> env(Environment::Create()); | 52 scoped_ptr<Environment> env(Environment::Create()); |
| 53 EXPECT_TRUE(env->HasVar("PATH")); | 53 EXPECT_TRUE(env->HasVar("PATH")); |
| 54 } | 54 } |
| 55 | 55 |
| 56 TEST_F(EnvironmentTest, SetVar) { | 56 TEST_F(EnvironmentTest, SetVar) { |
| 57 scoped_ptr<Environment> env(Environment::Create()); | 57 scoped_ptr<Environment> env(Environment::Create()); |
| 58 | 58 |
| 59 const char* kFooUpper = "FOO"; | 59 const char kFooUpper[] = "FOO"; |
| 60 const char* kFooLower = "foo"; | 60 const char kFooLower[] = "foo"; |
| 61 EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower)); | 61 EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower)); |
| 62 | 62 |
| 63 // Now verify that the environment has the new variable. | 63 // Now verify that the environment has the new variable. |
| 64 EXPECT_TRUE(env->HasVar(kFooUpper)); | 64 EXPECT_TRUE(env->HasVar(kFooUpper)); |
| 65 | 65 |
| 66 std::string var_value; | 66 std::string var_value; |
| 67 EXPECT_TRUE(env->GetVar(kFooUpper, &var_value)); | 67 EXPECT_TRUE(env->GetVar(kFooUpper, &var_value)); |
| 68 EXPECT_EQ(var_value, kFooLower); | 68 EXPECT_EQ(var_value, kFooLower); |
| 69 } | 69 } |
| 70 | 70 |
| 71 TEST_F(EnvironmentTest, UnSetVar) { | 71 TEST_F(EnvironmentTest, UnSetVar) { |
| 72 scoped_ptr<Environment> env(Environment::Create()); | 72 scoped_ptr<Environment> env(Environment::Create()); |
| 73 | 73 |
| 74 const char* kFooUpper = "FOO"; | 74 const char kFooUpper[] = "FOO"; |
| 75 const char* kFooLower = "foo"; | 75 const char kFooLower[] = "foo"; |
| 76 // First set some environment variable. | 76 // First set some environment variable. |
| 77 EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower)); | 77 EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower)); |
| 78 | 78 |
| 79 // Now verify that the environment has the new variable. | 79 // Now verify that the environment has the new variable. |
| 80 EXPECT_TRUE(env->HasVar(kFooUpper)); | 80 EXPECT_TRUE(env->HasVar(kFooUpper)); |
| 81 | 81 |
| 82 // Finally verify that the environment variable was erased. | 82 // Finally verify that the environment variable was erased. |
| 83 EXPECT_TRUE(env->UnSetVar(kFooUpper)); | 83 EXPECT_TRUE(env->UnSetVar(kFooUpper)); |
| 84 | 84 |
| 85 // And check that the variable has been unset. | 85 // And check that the variable has been unset. |
| 86 EXPECT_FALSE(env->HasVar(kFooUpper)); | 86 EXPECT_FALSE(env->HasVar(kFooUpper)); |
| 87 } | 87 } |
| 88 | 88 |
| 89 #if defined(OS_WIN) | 89 #if defined(OS_WIN) |
| 90 | 90 |
| 91 TEST_F(EnvironmentTest, AlterEnvironment) { | 91 TEST_F(EnvironmentTest, AlterEnvironment) { |
| 92 const wchar_t empty[] = L"\0"; | 92 const wchar_t empty[] = L"\0"; |
| 93 const wchar_t a2[] = L"A=2\0"; | 93 const wchar_t a2[] = L"A=2\0"; |
| 94 EnvironmentMap changes; | 94 EnvironmentMap changes; |
| 95 string16 e; | 95 string16 e; |
| 96 | 96 |
| 97 e = AlterEnvironment(empty, changes); | 97 e = AlterEnvironment(empty, changes); |
| 98 EXPECT_TRUE(e[0] == 0); | 98 EXPECT_EQ(0, e[0]); |
| 99 | 99 |
| 100 changes[L"A"] = L"1"; | 100 changes[L"A"] = L"1"; |
| 101 e = AlterEnvironment(empty, changes); | 101 e = AlterEnvironment(empty, changes); |
| 102 EXPECT_EQ(string16(L"A=1\0\0", 5), e); | 102 EXPECT_EQ(string16(L"A=1\0\0", 5), e); |
| 103 | 103 |
| 104 changes.clear(); | 104 changes.clear(); |
| 105 changes[L"A"] = string16(); | 105 changes[L"A"] = string16(); |
| 106 e = AlterEnvironment(empty, changes); | 106 e = AlterEnvironment(empty, changes); |
| 107 EXPECT_EQ(string16(L"\0\0", 2), e); | 107 EXPECT_EQ(string16(L"\0\0", 2), e); |
| 108 | 108 |
| (...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 155 | 155 |
| 156 changes.clear(); | 156 changes.clear(); |
| 157 changes["A"] = std::string(); | 157 changes["A"] = std::string(); |
| 158 e = AlterEnvironment(a2, changes); | 158 e = AlterEnvironment(a2, changes); |
| 159 EXPECT_TRUE(e[0] == NULL); | 159 EXPECT_TRUE(e[0] == NULL); |
| 160 } | 160 } |
| 161 | 161 |
| 162 #endif | 162 #endif |
| 163 | 163 |
| 164 } // namespace base | 164 } // namespace base |
| OLD | NEW |