123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164 |
- // Copyright (c) 2011 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 "butil/environment.h"
- #include "butil/memory/scoped_ptr.h"
- #include <gtest/gtest.h>
- #include <gtest/gtest.h>
- typedef testing::Test EnvironmentTest;
- namespace butil {
- TEST_F(EnvironmentTest, GetVar) {
- // Every setup should have non-empty PATH...
- scoped_ptr<Environment> env(Environment::Create());
- std::string env_value;
- EXPECT_TRUE(env->GetVar("PATH", &env_value));
- EXPECT_NE(env_value, "");
- }
- TEST_F(EnvironmentTest, GetVarReverse) {
- scoped_ptr<Environment> env(Environment::Create());
- const char* kFooUpper = "FOO";
- const char* kFooLower = "foo";
- // Set a variable in UPPER case.
- EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower));
- // And then try to get this variable passing the lower case.
- std::string env_value;
- EXPECT_TRUE(env->GetVar(kFooLower, &env_value));
- EXPECT_STREQ(env_value.c_str(), kFooLower);
- EXPECT_TRUE(env->UnSetVar(kFooUpper));
- const char* kBar = "bar";
- // Now do the opposite, set the variable in the lower case.
- EXPECT_TRUE(env->SetVar(kFooLower, kBar));
- // And then try to get this variable passing the UPPER case.
- EXPECT_TRUE(env->GetVar(kFooUpper, &env_value));
- EXPECT_STREQ(env_value.c_str(), kBar);
- EXPECT_TRUE(env->UnSetVar(kFooLower));
- }
- TEST_F(EnvironmentTest, HasVar) {
- // Every setup should have PATH...
- scoped_ptr<Environment> env(Environment::Create());
- EXPECT_TRUE(env->HasVar("PATH"));
- }
- TEST_F(EnvironmentTest, SetVar) {
- scoped_ptr<Environment> env(Environment::Create());
- const char* kFooUpper = "FOO";
- const char* kFooLower = "foo";
- EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower));
- // Now verify that the environment has the new variable.
- EXPECT_TRUE(env->HasVar(kFooUpper));
- std::string var_value;
- EXPECT_TRUE(env->GetVar(kFooUpper, &var_value));
- EXPECT_EQ(var_value, kFooLower);
- }
- TEST_F(EnvironmentTest, UnSetVar) {
- scoped_ptr<Environment> env(Environment::Create());
- const char* kFooUpper = "FOO";
- const char* kFooLower = "foo";
- // First set some environment variable.
- EXPECT_TRUE(env->SetVar(kFooUpper, kFooLower));
- // Now verify that the environment has the new variable.
- EXPECT_TRUE(env->HasVar(kFooUpper));
- // Finally verify that the environment variable was erased.
- EXPECT_TRUE(env->UnSetVar(kFooUpper));
- // And check that the variable has been unset.
- EXPECT_FALSE(env->HasVar(kFooUpper));
- }
- #if defined(OS_WIN)
- TEST_F(EnvironmentTest, AlterEnvironment) {
- const wchar_t empty[] = L"\0";
- const wchar_t a2[] = L"A=2\0";
- EnvironmentMap changes;
- string16 e;
- e = AlterEnvironment(empty, changes);
- EXPECT_TRUE(e[0] == 0);
- changes[L"A"] = L"1";
- e = AlterEnvironment(empty, changes);
- EXPECT_EQ(string16(L"A=1\0\0", 5), e);
- changes.clear();
- changes[L"A"] = string16();
- e = AlterEnvironment(empty, changes);
- EXPECT_EQ(string16(L"\0\0", 2), e);
- changes.clear();
- e = AlterEnvironment(a2, changes);
- EXPECT_EQ(string16(L"A=2\0\0", 5), e);
- changes.clear();
- changes[L"A"] = L"1";
- e = AlterEnvironment(a2, changes);
- EXPECT_EQ(string16(L"A=1\0\0", 5), e);
- changes.clear();
- changes[L"A"] = string16();
- e = AlterEnvironment(a2, changes);
- EXPECT_EQ(string16(L"\0\0", 2), e);
- }
- #else
- TEST_F(EnvironmentTest, AlterEnvironment) {
- const char* const empty[] = { NULL };
- const char* const a2[] = { "A=2", NULL };
- EnvironmentMap changes;
- scoped_ptr<char*[]> e;
- e = AlterEnvironment(empty, changes).Pass();
- EXPECT_TRUE(e[0] == NULL);
- changes["A"] = "1";
- e = AlterEnvironment(empty, changes);
- EXPECT_EQ(std::string("A=1"), e[0]);
- EXPECT_TRUE(e[1] == NULL);
- changes.clear();
- changes["A"] = std::string();
- e = AlterEnvironment(empty, changes);
- EXPECT_TRUE(e[0] == NULL);
- changes.clear();
- e = AlterEnvironment(a2, changes);
- EXPECT_EQ(std::string("A=2"), e[0]);
- EXPECT_TRUE(e[1] == NULL);
- changes.clear();
- changes["A"] = "1";
- e = AlterEnvironment(a2, changes);
- EXPECT_EQ(std::string("A=1"), e[0]);
- EXPECT_TRUE(e[1] == NULL);
- changes.clear();
- changes["A"] = std::string();
- e = AlterEnvironment(a2, changes);
- EXPECT_TRUE(e[0] == NULL);
- }
- #endif
- } // namespace butil
|