Fix flaky tests

Author: ra1028

Tests/AtomsTests/Atom/AsyncPhaseAtomTests.swift

@@ -70,9 +70,7 @@ final class AsyncPhaseAtomTests: XCTestCase {
                 await context.refresh(atom)
             }
 
-            Task {
-                refreshTask.cancel()
-            }
+            refreshTask.cancel()
 
             let phase = await refreshTask.value
             XCTAssertTrue(phase.isSuspending)

Tests/AtomsTests/Atom/TaskAtomTests.swift

@@ -66,16 +66,13 @@ final class TaskAtomTests: XCTestCase {
 
         do {
             // Cancellation
-            let refreshTask0 = Task {
+            let refreshTask = Task {
                 await context.refresh(atom)
             }
 
-            Task {
-                refreshTask0.cancel()
-            }
-
-            let task = await refreshTask0.value
+            refreshTask.cancel()
 
+            let task = await refreshTask.value
             XCTAssertTrue(task.isCancelled)
         }
 
@@ -89,19 +86,15 @@ final class TaskAtomTests: XCTestCase {
 
         do {
             // Override cancellation
-
             context.override(atom) { _ in Task { 400 } }
 
             let refreshTask = Task {
                 await context.refresh(atom)
             }
 
-            Task {
-                refreshTask.cancel()
-            }
+            refreshTask.cancel()
 
             let task = await refreshTask.value
-
             XCTAssertTrue(task.isCancelled)
         }
     }

Tests/AtomsTests/Atom/ThrowingTaskAtomTests.swift

@@ -85,12 +85,9 @@ final class ThrowingTaskAtomTests: XCTestCase {
                 await context.refresh(atom)
             }
 
-            Task {
-                refreshTask.cancel()
-            }
+            refreshTask.cancel()
 
             let task = await refreshTask.value
-
             XCTAssertTrue(task.isCancelled)
         }
 
@@ -104,20 +101,16 @@ final class ThrowingTaskAtomTests: XCTestCase {
 
         do {
             // Override cancellation
-
             context.override(atom) { _ in Task { 400 } }
 
-            let refreshTask1 = Task {
+            let refreshTask = Task {
                 await context.refresh(atom)
             }
 
-            Task {
-                refreshTask1.cancel()
-            }
-
-            let task1 = await refreshTask1.value
+            refreshTask.cancel()
 
-            XCTAssertTrue(task1.isCancelled)
+            let task = await refreshTask.value
+            XCTAssertTrue(task.isCancelled)
         }
     }