Skip to content

Commit 7cccf0b

Browse files
committed
Wrap unit test calls to set_var in unsafe {}
Newer rustc's complain about it, previous ones didn't 🤷
1 parent 8b7d289 commit 7cccf0b

File tree

3 files changed

+17
-9
lines changed

3 files changed

+17
-9
lines changed

cli/src/main.rs

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,9 @@ mod tests {
7575
let mut flags = Flags::default();
7676
flags.table = None;
7777

78-
std::env::set_var("TABLE_LOCATION", "s3://test-bucket-from-env/table");
78+
unsafe {
79+
std::env::set_var("TABLE_LOCATION", "s3://test-bucket-from-env/table");
80+
}
7981

8082
let location = table_location(&flags).expect("Failed to load table location");
8183
assert_eq!(location, "s3://test-bucket-from-env/table");

crates/oxbow/src/lib.rs

Lines changed: 8 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -476,7 +476,7 @@ fn coerce_field(
476476
let coerced = coerce_field(field.clone());
477477
let list_field = Field::new(field.name(), DataType::List(coerced), field.is_nullable());
478478
return Arc::new(list_field);
479-
},
479+
}
480480
DataType::Struct(fields) => {
481481
let coerced: Vec<deltalake::arrow::datatypes::FieldRef> =
482482
fields.iter().map(|f| coerce_field(f.clone())).collect();
@@ -486,7 +486,7 @@ fn coerce_field(
486486
field.is_nullable(),
487487
);
488488
return Arc::new(struct_field);
489-
},
489+
}
490490
_ => {}
491491
};
492492
field.clone()
@@ -832,7 +832,9 @@ mod tests {
832832
#[tokio::test]
833833
async fn attempt_to_convert_without_auth() {
834834
let region = std::env::var("AWS_REGION");
835-
std::env::set_var("AWS_REGION", "custom");
835+
unsafe {
836+
std::env::set_var("AWS_REGION", "custom");
837+
}
836838
deltalake::aws::register_handlers(None);
837839

838840
let files: Vec<ObjectMeta> = vec![];
@@ -847,7 +849,9 @@ mod tests {
847849
assert!(result.is_err());
848850

849851
if let Ok(region) = region {
850-
std::env::set_var("AWS_REGION", region);
852+
unsafe {
853+
std::env::set_var("AWS_REGION", region);
854+
}
851855
}
852856
}
853857

lambdas/glue-sync/src/main.rs

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -389,10 +389,12 @@ mod tests {
389389
records: vec![message],
390390
};
391391
let lambda_event = LambdaEvent::new(event, lambda_runtime::Context::default());
392-
std::env::set_var(
393-
GLUE_REGEX_ENV,
394-
r#"(?P<database>\w+)\/(?P<table>\w+)\/_delta_log\/.*.json"#,
395-
);
392+
unsafe {
393+
std::env::set_var(
394+
GLUE_REGEX_ENV,
395+
r#"(?P<database>\w+)\/(?P<table>\w+)\/_delta_log\/.*.json"#,
396+
);
397+
}
396398
let result = function_handler(lambda_event).await;
397399
assert!(result.is_ok());
398400
}

0 commit comments

Comments
 (0)