| 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; |
| (...skipping 68 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 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) | |
| 90 | |
| 91 TEST_F(EnvironmentTest, AlterEnvironment) { | |
| 92 const wchar_t empty[] = L"\0"; | |
| 93 const wchar_t a2[] = L"A=2\0"; | |
| 94 EnvironmentMap changes; | |
| 95 string16 e; | |
| 96 | |
| 97 e = AlterEnvironment(empty, changes); | |
| 98 EXPECT_EQ(0, e[0]); | |
| 99 | |
| 100 changes[L"A"] = L"1"; | |
| 101 e = AlterEnvironment(empty, changes); | |
| 102 EXPECT_EQ(string16(L"A=1\0\0", 5), e); | |
| 103 | |
| 104 changes.clear(); | |
| 105 changes[L"A"] = string16(); | |
| 106 e = AlterEnvironment(empty, changes); | |
| 107 EXPECT_EQ(string16(L"\0\0", 2), e); | |
| 108 | |
| 109 changes.clear(); | |
| 110 e = AlterEnvironment(a2, changes); | |
| 111 EXPECT_EQ(string16(L"A=2\0\0", 5), e); | |
| 112 | |
| 113 changes.clear(); | |
| 114 changes[L"A"] = L"1"; | |
| 115 e = AlterEnvironment(a2, changes); | |
| 116 EXPECT_EQ(string16(L"A=1\0\0", 5), e); | |
| 117 | |
| 118 changes.clear(); | |
| 119 changes[L"A"] = string16(); | |
| 120 e = AlterEnvironment(a2, changes); | |
| 121 EXPECT_EQ(string16(L"\0\0", 2), e); | |
| 122 } | |
| 123 | |
| 124 #else | |
| 125 | |
| 126 TEST_F(EnvironmentTest, AlterEnvironment) { | 89 TEST_F(EnvironmentTest, AlterEnvironment) { |
| 127 const char* const empty[] = { NULL }; | 90 const char* const empty[] = { NULL }; |
| 128 const char* const a2[] = { "A=2", NULL }; | 91 const char* const a2[] = { "A=2", NULL }; |
| 129 EnvironmentMap changes; | 92 EnvironmentMap changes; |
| 130 scoped_ptr<char*[]> e; | 93 scoped_ptr<char*[]> e; |
| 131 | 94 |
| 132 e = AlterEnvironment(empty, changes).Pass(); | 95 e = AlterEnvironment(empty, changes).Pass(); |
| 133 EXPECT_TRUE(e[0] == NULL); | 96 EXPECT_TRUE(e[0] == NULL); |
| 134 | 97 |
| 135 changes["A"] = "1"; | 98 changes["A"] = "1"; |
| (...skipping 16 matching lines...) Expand all Loading... |
| 152 e = AlterEnvironment(a2, changes); | 115 e = AlterEnvironment(a2, changes); |
| 153 EXPECT_EQ(std::string("A=1"), e[0]); | 116 EXPECT_EQ(std::string("A=1"), e[0]); |
| 154 EXPECT_TRUE(e[1] == NULL); | 117 EXPECT_TRUE(e[1] == NULL); |
| 155 | 118 |
| 156 changes.clear(); | 119 changes.clear(); |
| 157 changes["A"] = std::string(); | 120 changes["A"] = std::string(); |
| 158 e = AlterEnvironment(a2, changes); | 121 e = AlterEnvironment(a2, changes); |
| 159 EXPECT_TRUE(e[0] == NULL); | 122 EXPECT_TRUE(e[0] == NULL); |
| 160 } | 123 } |
| 161 | 124 |
| 162 #endif | |
| 163 | |
| 164 } // namespace base | 125 } // namespace base |
| OLD | NEW |