@@ -8,7 +8,7 @@ use std::{
8
8
} ;
9
9
10
10
lazy_static ! {
11
- pub static ref COUNTER : AtomicUsize = AtomicUsize :: new( 1 ) ;
11
+ static ref IP_COUNTER : AtomicUsize = AtomicUsize :: new( 1 ) ;
12
12
}
13
13
14
14
const COUNTER_FILE : & str = ".ip_counter" ;
@@ -21,23 +21,26 @@ pub struct TestEnvironment {
21
21
pub base_storage_path : String ,
22
22
}
23
23
24
- pub fn get_test_environment ( test_name : & str ) -> TestEnvironment {
24
+ pub fn prepare_test_environment ( test_name : & str ) -> TestEnvironment {
25
25
let test_environment_path = format ! ( "logs/{}" , test_name) ;
26
26
let test_environment_path = Path :: new ( & test_environment_path) ;
27
27
28
28
// Remove if the previous test environment exists
29
29
if test_environment_path. exists ( ) {
30
30
fs:: remove_dir_all ( test_environment_path)
31
- . expect ( "Failed to remove previous logs directory " ) ;
31
+ . expect ( "Failed to remove previous test environment " ) ;
32
32
}
33
33
34
34
// Caution: Storing ip_counter in the tempdir result in different test binaries using duplicate loopback IP addresses,
35
35
// so we store the ip_counter file in the harness/logs directory.
36
- get_test_environment_with_counter_file ( test_name, COUNTER_FILE )
36
+ prepare_test_environment_with_counter_file ( test_name, COUNTER_FILE )
37
37
}
38
38
39
39
// Function with adjustable counter file path for testing
40
- fn get_test_environment_with_counter_file ( test_name : & str , counter_file : & str ) -> TestEnvironment {
40
+ fn prepare_test_environment_with_counter_file (
41
+ test_name : & str ,
42
+ counter_file : & str ,
43
+ ) -> TestEnvironment {
41
44
let path = Path :: new ( counter_file) ;
42
45
43
46
let mut file = OpenOptions :: new ( )
@@ -123,11 +126,11 @@ mod tests {
123
126
write ! ( file, "{}" , MAX_COUNTER - 1 ) . expect ( "Failed to write to counter file" ) ;
124
127
125
128
// First call: should return last IP address 127.255.255.254
126
- let env = get_test_environment_with_counter_file ( test_name, counter_file_str) ;
129
+ let env = prepare_test_environment_with_counter_file ( test_name, counter_file_str) ;
127
130
assert_eq ! ( env. loopback_address, "127.255.255.254" ) ;
128
131
129
132
// Second call: counter wraps around, should return first IP address 127.0.0.1
130
- let env = get_test_environment_with_counter_file ( test_name, counter_file_str) ;
133
+ let env = prepare_test_environment_with_counter_file ( test_name, counter_file_str) ;
131
134
assert_eq ! ( env. loopback_address, "127.0.0.1" ) ;
132
135
}
133
136
@@ -140,16 +143,16 @@ mod tests {
140
143
let counter_file_str = counter_file_path. to_str ( ) . unwrap ( ) ;
141
144
142
145
// First allocation
143
- let env = get_test_environment_with_counter_file ( test_name, counter_file_str) ;
146
+ let env = prepare_test_environment_with_counter_file ( test_name, counter_file_str) ;
144
147
assert_eq ! ( env. loopback_address, "127.0.0.1" ) ;
145
148
assert_eq ! ( env. base_storage_path, "./logs/test_basic" ) ;
146
149
147
150
// Second allocation
148
- let env = get_test_environment_with_counter_file ( test_name, counter_file_str) ;
151
+ let env = prepare_test_environment_with_counter_file ( test_name, counter_file_str) ;
149
152
assert_eq ! ( env. loopback_address, "127.0.0.2" ) ;
150
153
151
154
// Third allocation
152
- let env = get_test_environment_with_counter_file ( test_name, counter_file_str) ;
155
+ let env = prepare_test_environment_with_counter_file ( test_name, counter_file_str) ;
153
156
assert_eq ! ( env. loopback_address, "127.0.0.3" ) ;
154
157
}
155
158
@@ -167,7 +170,7 @@ mod tests {
167
170
write ! ( file, "{}" , counter_value) . expect ( "Failed to write to counter file" ) ;
168
171
169
172
// First call: should return IP address 127.1.1.1
170
- let env = get_test_environment_with_counter_file ( test_name, counter_file_str) ;
173
+ let env = prepare_test_environment_with_counter_file ( test_name, counter_file_str) ;
171
174
assert_eq ! ( env. loopback_address, "127.1.1.1" ) ;
172
175
}
173
176
@@ -184,6 +187,6 @@ mod tests {
184
187
let counter_file_str = counter_file_path. to_str ( ) . unwrap ( ) ;
185
188
186
189
// Function call should panic
187
- get_test_environment_with_counter_file ( test_name, counter_file_str) ;
190
+ prepare_test_environment_with_counter_file ( test_name, counter_file_str) ;
188
191
}
189
192
}
0 commit comments