Fix test on running_actions_manager_test (#1141)

TraceMachina · Jul 11, 2024 · 18db991 · 18db991
1 parent 991332f
commit 18db991
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/nativelink-scheduler/src/simple_scheduler.rs b/nativelink-scheduler/src/simple_scheduler.rs
@@ -424,11 +424,15 @@ impl WorkerScheduler for SimpleScheduler {
     }
 
     async fn remove_timedout_workers(&self, now_timestamp: WorkerTimestamp) -> Result<(), Error> {
-        self.worker_scheduler.remove_timedout_workers(now_timestamp).await
+        self.worker_scheduler
+            .remove_timedout_workers(now_timestamp)
+            .await
     }
 
     async fn set_drain_worker(&self, worker_id: &WorkerId, is_draining: bool) -> Result<(), Error> {
-        self.worker_scheduler.set_drain_worker(worker_id, is_draining).await
+        self.worker_scheduler
+            .set_drain_worker(worker_id, is_draining)
+            .await
     }
 
     fn register_metrics(self: Arc<Self>, registry: &mut Registry) {

diff --git a/nativelink-worker/tests/running_actions_manager_test.rs b/nativelink-worker/tests/running_actions_manager_test.rs
@@ -42,10 +42,12 @@ use nativelink_store::ac_utils::{get_and_decode_digest, serialize_and_upload_mes
 use nativelink_store::fast_slow_store::FastSlowStore;
 use nativelink_store::filesystem_store::FilesystemStore;
 use nativelink_store::memory_store::MemoryStore;
+#[cfg_attr(target_family = "windows", allow(unused_imports))]
+use nativelink_util::action_messages::SymlinkInfo;
 use nativelink_util::action_messages::{
-    ActionResult, DirectoryInfo, ExecutionMetadata, FileInfo, NameOrPath,
+    ActionResult, ActionUniqueKey, ActionUniqueQualifier, DirectoryInfo, ExecutionMetadata,
+    FileInfo, NameOrPath, OperationId,
 };
-use nativelink_util::action_messages::{ActionUniqueKey, ActionUniqueQualifier, OperationId};
 use nativelink_util::common::{fs, DigestInfo};
 use nativelink_util::digest_hasher::{DigestHasher, DigestHasherFunc};
 use nativelink_util::store_trait::{Store, StoreLike};