1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
// Copyright 2020 Open Source Robotics Foundation, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#include <gtest/gtest.h>
#include <string>
#include "rcutils/env.h"
#include "rcutils/error_handling.h"
TEST(TestEnv, test_set_env) {
const char * res = nullptr;
// Invalid cases
EXPECT_FALSE(rcutils_set_env(nullptr, nullptr));
rcutils_reset_error();
EXPECT_FALSE(rcutils_set_env("=INVALID_ENV_VAR=", nullptr));
rcutils_reset_error();
EXPECT_FALSE(rcutils_set_env("=INVALID_ENV_VAR=", "InvalidEnvValue"));
rcutils_reset_error();
// Ensure we're starting clean
ASSERT_EQ(nullptr, rcutils_get_env("NEW_ENV_VAR", &res));
ASSERT_STREQ("", res);
// Simple set
ASSERT_TRUE(rcutils_set_env("NEW_ENV_VAR", "NewEnvValue"));
ASSERT_STREQ(nullptr, rcutils_get_env("NEW_ENV_VAR", &res));
EXPECT_STREQ(res, "NewEnvValue");
// Re-set
ASSERT_TRUE(rcutils_set_env("NEW_ENV_VAR", "DifferentEnvValue"));
ASSERT_STREQ(nullptr, rcutils_get_env("NEW_ENV_VAR", &res));
EXPECT_STREQ(res, "DifferentEnvValue");
// Un-set
ASSERT_TRUE(rcutils_set_env("NEW_ENV_VAR", nullptr));
ASSERT_EQ(nullptr, rcutils_get_env("NEW_ENV_VAR", &res));
EXPECT_STREQ("", res);
// Un-set again
ASSERT_TRUE(rcutils_set_env("NEW_ENV_VAR", nullptr));
ASSERT_EQ(nullptr, rcutils_get_env("NEW_ENV_VAR", &res));
EXPECT_STREQ("", res);
}
/* Tests rcutils_get_env.
*
* Expected environment variables must be set by the calling code:
*
* - EMPTY_TEST=
* - NORMAL_TEST=foo
*
* These are set in the call to `ament_add_gtest()` in the `CMakeLists.txt`.
*/
TEST(TestEnv, test_get_env) {
const char * env;
const char * ret;
ret = rcutils_get_env("NORMAL_TEST", NULL);
EXPECT_STREQ("argument env_value is null", ret);
ret = rcutils_get_env(NULL, &env);
EXPECT_STREQ("argument env_name is null", ret);
ret = rcutils_get_env("SHOULD_NOT_EXIST_TEST", &env);
EXPECT_FALSE(ret);
EXPECT_STREQ("", env);
ret = rcutils_get_env("NORMAL_TEST", &env);
EXPECT_FALSE(ret);
EXPECT_FALSE(NULL == env);
EXPECT_STREQ("foo", env);
ret = rcutils_get_env("EMPTY_TEST", &env);
EXPECT_FALSE(ret);
EXPECT_FALSE(NULL == env);
EXPECT_STREQ("", env);
}
/* Tests rcutils_get_env.
*
* Expected environment variables must be set by the calling code:
*
* - EMPTY_TEST=
* - NORMAL_TEST=foo
*
* These are set in the call to `ament_add_gtest()` in the `CMakeLists.txt`.
*/
TEST(TestEnv, test_set_env_overwrite) {
const char * env;
const char * ret;
// Do not overwrite environment variable if preset if overwrite is set false.
EXPECT_TRUE(rcutils_set_env_overwrite("NORMAL_TEST", "NewEnvValue", false));
ret = rcutils_get_env("NORMAL_TEST", &env);
EXPECT_TRUE(NULL == ret);
EXPECT_STREQ("foo", env);
// Overwrite environment variable if present if overwrite is set true.
EXPECT_TRUE(rcutils_set_env_overwrite("NORMAL_TEST", "NewEnvValue", true));
ret = rcutils_get_env("NORMAL_TEST", &env);
EXPECT_TRUE(NULL == ret);
EXPECT_STREQ("NewEnvValue", env);
}
TEST(TestEnv, test_get_home) {
EXPECT_STRNE(NULL, rcutils_get_home_dir());
const char * home = NULL;
#ifdef _WIN32
// Assert pre-condition that USERPROFILE is defined
const char * ret = rcutils_get_env("USERPROFILE", &home);
ASSERT_EQ(NULL, ret);
// Check USERPROFILE is not defined
EXPECT_TRUE(rcutils_set_env("USERPROFILE", NULL));
EXPECT_EQ(NULL, rcutils_get_home_dir());
#else
// Assert pre-condition that HOME is defined
const char * ret = rcutils_get_env("HOME", &home);
ASSERT_EQ(NULL, ret);
// Check HOME is not defined
EXPECT_TRUE(rcutils_set_env("HOME", NULL));
EXPECT_EQ(NULL, rcutils_get_home_dir());
#endif
}
|
