@@ -191,11 +191,13 @@ fn build_config_and_process(
191
191
192
192
#[ cfg( test) ]
193
193
mod tests {
194
+ use std:: env;
194
195
use crate :: exporter:: config:: build_config_and_process;
195
196
use opentelemetry:: sdk:: trace:: Config ;
196
197
use opentelemetry:: sdk:: Resource ;
197
198
use opentelemetry:: KeyValue ;
198
199
use std:: sync:: Arc ;
200
+ use crate :: new_agent_pipeline;
199
201
200
202
#[ test]
201
203
fn test_set_service_name ( ) {
@@ -241,13 +243,17 @@ mod tests {
241
243
KeyValue :: new( "service.name" , "override_service" )
242
244
) ;
243
245
244
- // todo: move it to a separate test function
245
- // // OTEL_SERVICE_NAME env var also works
246
- // env::set_var("OTEL_SERVICE_NAME", "test service");
247
- // builder = crate::PipelineBuilder::default();
248
- // let exporter = builder.init_sync_exporter().unwrap();
249
- // assert_eq!(exporter.process.service_name, "test service");
250
- // env::set_var("OTEL_SERVICE_NAME", "")
246
+
247
+ }
248
+
249
+ #[ test]
250
+ fn test_read_from_env ( ) {
251
+ // OTEL_SERVICE_NAME env var also works
252
+ env:: set_var ( "OTEL_SERVICE_NAME" , "test service" ) ;
253
+ let builder = new_agent_pipeline ( ) ;
254
+ let exporter = builder. build_sync_agent_exporter ( ) . unwrap ( ) ;
255
+ assert_eq ! ( exporter. process. service_name, "test service" ) ;
256
+ env:: set_var ( "OTEL_SERVICE_NAME" , "" )
251
257
}
252
258
}
253
259
0 commit comments