refactor: Clean up some tests

pueue/tests/daemon/integration/restart.rs

@@ -13,12 +13,12 @@ async fn test_restart_in_place() -> Result<()> {
     let shared = &daemon.settings.shared;
 
     // Add a single task that instantly finishes.
-    assert_success(add_task(shared, "sleep 0.1").await?);
+    assert_success(add_and_start_task(shared, "ls").await?);
 
     // Wait for task 0 to finish.
     let original_task = wait_for_task_condition(shared, 0, Task::is_done).await?;
 
-    // Restart task 0 with an extended sleep command with a different path.
+    // Restart task 0 with an sleep command and with a different path.
     let restart_message = RestartMessage {
         tasks: vec![TaskToRestart {
             task_id: 0,
@@ -60,13 +60,11 @@ async fn test_cannot_restart_running() -> Result<()> {
     let daemon = daemon().await?;
     let shared = &daemon.settings.shared;
 
-    // Add a single task that instantly finishes.
-    assert_success(add_task(shared, "sleep 60").await?);
-
-    // Wait for task 0 to finish.
-    let task = wait_for_task_condition(shared, 0, Task::is_running).await?;
+    // Add a long running task that starts immediately.
+    assert_success(add_and_start_task(shared, "sleep 60").await?);
+    let task = get_task(shared, 1).await?;
 
-    // Restart task 0 with an extended sleep command.
+    // Try to restart task 0
     let restart_message = RestartMessage {
         tasks: vec![TaskToRestart {
             task_id: 0,