From dc8015f6695ee069279dfbd681000e8b1fcd30e5 Mon Sep 17 00:00:00 2001 From: Mason Reed Date: Wed, 26 Feb 2025 20:56:18 -0500 Subject: Remove module level `Session` initialization in Rust unit tests This was preventing the enterprise license checkout from dropping, also the initialization story is much better than it was when we added the rstest fixtures, we can get away with initialization in parallel more broadly. --- rust/tests/background_task.rs | 22 +++++++++------------- 1 file changed, 9 insertions(+), 13 deletions(-) (limited to 'rust/tests/background_task.rs') diff --git a/rust/tests/background_task.rs b/rust/tests/background_task.rs index d0c329e3..6e0c354b 100644 --- a/rust/tests/background_task.rs +++ b/rust/tests/background_task.rs @@ -1,15 +1,9 @@ use binaryninja::background_task::*; use binaryninja::headless::Session; -use rstest::*; -#[fixture] -#[once] -fn session() -> Session { - Session::new().expect("Failed to initialize session") -} - -#[rstest] -fn test_background_task_registered(_session: &Session) { +#[test] +fn test_background_task_registered() { + let _session = Session::new().expect("Failed to initialize session"); let task_progress = "test registered"; let task = BackgroundTask::new(task_progress, false); BackgroundTask::running_tasks() @@ -24,8 +18,9 @@ fn test_background_task_registered(_session: &Session) { assert!(!still_running, "Task still running"); } -#[rstest] -fn test_background_task_cancellable(_session: &Session) { +#[test] +fn test_background_task_cancellable() { + let _session = Session::new().expect("Failed to initialize session"); let task_progress = "test cancellable"; let task = BackgroundTask::new(task_progress, false); BackgroundTask::running_tasks() @@ -37,8 +32,9 @@ fn test_background_task_cancellable(_session: &Session) { task.finish(); } -#[rstest] -fn test_background_task_progress(_session: &Session) { +#[test] +fn test_background_task_progress() { + let _session = Session::new().expect("Failed to initialize session"); let task = BackgroundTask::new("test progress", false); let first_progress = task.progress_text().to_string(); assert_eq!(first_progress, "test progress"); -- cgit v1.3.1