|
8 | 8 | * the file licenses/APL2.txt. |
9 | 9 | */ |
10 | 10 | #include <folly/portability/GTest.h> |
11 | | - |
12 | | -#include <platform/checked_snprintf.h> |
| 11 | +#include <folly/portability/Stdlib.h> |
13 | 12 | #include <platform/sysinfo.h> |
14 | 13 |
|
15 | | -char cpu_count_env_var[256]; |
16 | | - |
17 | 14 | TEST(GetAvailableCpu, NoVariable) { |
18 | 15 | EXPECT_NE(0u, cb::get_available_cpu_count()); |
19 | 16 | } |
20 | 17 |
|
21 | 18 | TEST(GetAvailableCpu, CorrectVariable_ExactNumber) { |
22 | | - checked_snprintf(cpu_count_env_var, |
23 | | - sizeof(cpu_count_env_var), |
24 | | - "COUCHBASE_CPU_COUNT=10000"); |
25 | | - putenv(cpu_count_env_var); |
| 19 | + setenv("COUCHBASE_CPU_COUNT", "10000", 1); |
26 | 20 | EXPECT_EQ(10000u, cb::get_available_cpu_count()); |
27 | 21 | } |
28 | 22 |
|
29 | 23 | TEST(GetAvailableCpu, CorrectVariable_LeadingSpace) { |
30 | | - checked_snprintf(cpu_count_env_var, |
31 | | - sizeof(cpu_count_env_var), |
32 | | - "COUCHBASE_CPU_COUNT= 9999"); |
33 | | - putenv(cpu_count_env_var); |
| 24 | + setenv("COUCHBASE_CPU_COUNT", " 9999", 1); |
34 | 25 | EXPECT_EQ(9999u, cb::get_available_cpu_count()); |
35 | 26 | } |
36 | 27 |
|
37 | 28 | TEST(GetAvailableCpu, CorrectVariable_TrailingSpace) { |
38 | | - checked_snprintf(cpu_count_env_var, |
39 | | - sizeof(cpu_count_env_var), |
40 | | - "COUCHBASE_CPU_COUNT=9998 "); |
41 | | - putenv(cpu_count_env_var); |
| 29 | + setenv("COUCHBASE_CPU_COUNT", "9998 ", 1); |
42 | 30 | EXPECT_EQ(9998u, cb::get_available_cpu_count()); |
43 | 31 | } |
44 | 32 |
|
45 | 33 | TEST(GetAvailableCpu, CorrectVariable_LeadingTab) { |
46 | | - checked_snprintf(cpu_count_env_var, |
47 | | - sizeof(cpu_count_env_var), |
48 | | - "COUCHBASE_CPU_COUNT=\t9997"); |
49 | | - putenv(cpu_count_env_var); |
| 34 | + setenv("COUCHBASE_CPU_COUNT", "\t9997", 1); |
50 | 35 | EXPECT_EQ(9997u, cb::get_available_cpu_count()); |
51 | 36 | } |
52 | 37 |
|
53 | 38 | TEST(GetAvailableCpu, CorrectVariable_TrailigTab) { |
54 | | - checked_snprintf(cpu_count_env_var, |
55 | | - sizeof(cpu_count_env_var), |
56 | | - "COUCHBASE_CPU_COUNT=9996\t"); |
57 | | - putenv(cpu_count_env_var); |
| 39 | + setenv("COUCHBASE_CPU_COUNT", "9996\t", 1); |
58 | 40 | EXPECT_EQ(9996u, cb::get_available_cpu_count()); |
59 | 41 | } |
60 | 42 |
|
61 | 43 | TEST(GetAvailableCpu, InvalidValue) { |
62 | | - checked_snprintf(cpu_count_env_var, |
63 | | - sizeof(cpu_count_env_var), |
64 | | - "COUCHBASE_CPU_COUNT=1a"); |
65 | | - putenv(cpu_count_env_var); |
| 44 | + setenv("COUCHBASE_CPU_COUNT", "1a", 1); |
66 | 45 | EXPECT_THROW(cb::get_available_cpu_count(), std::logic_error); |
67 | 46 |
|
68 | | - checked_snprintf(cpu_count_env_var, |
69 | | - sizeof(cpu_count_env_var), |
70 | | - "COUCHBASE_CPU_COUNT=1 a"); |
71 | | - putenv(cpu_count_env_var); |
| 47 | + setenv("COUCHBASE_CPU_COUNT", "1 a", 1); |
72 | 48 | EXPECT_THROW(cb::get_available_cpu_count(), std::logic_error); |
73 | 49 |
|
74 | | - checked_snprintf(cpu_count_env_var, |
75 | | - sizeof(cpu_count_env_var), |
76 | | - "COUCHBASE_CPU_COUNT=a1"); |
77 | | - putenv(cpu_count_env_var); |
| 50 | + setenv("COUCHBASE_CPU_COUNT", "a1", 1); |
78 | 51 | EXPECT_THROW(cb::get_available_cpu_count(), std::logic_error); |
79 | 52 |
|
80 | | - checked_snprintf(cpu_count_env_var, |
81 | | - sizeof(cpu_count_env_var), |
82 | | - "COUCHBASE_CPU_COUNT=a 1"); |
83 | | - putenv(cpu_count_env_var); |
| 53 | + setenv("COUCHBASE_CPU_COUNT", "a 1", 1); |
84 | 54 | EXPECT_THROW(cb::get_available_cpu_count(), std::logic_error); |
85 | 55 | } |
86 | 56 |
|
|
0 commit comments