From 3e816532fcf0fe5dd2b866f90bea7bd5257a39de Mon Sep 17 00:00:00 2001
From: Soumya Ranjan Mahunt <soumya.mahunt@gmail.com>
Date: Sat, 23 Jul 2022 13:44:30 +0530
Subject: [PATCH] test: add tests for cancellation source

---
 .../CancellationSourceTests.swift             | 188 ++++++++++++++++++
 1 file changed, 188 insertions(+)
 create mode 100644 Tests/AsyncObjectsTests/CancellationSourceTests.swift

diff --git a/Tests/AsyncObjectsTests/CancellationSourceTests.swift b/Tests/AsyncObjectsTests/CancellationSourceTests.swift
new file mode 100644
index 00000000..e58dc5fb
--- /dev/null
+++ b/Tests/AsyncObjectsTests/CancellationSourceTests.swift
@@ -0,0 +1,188 @@
+import XCTest
+@testable import AsyncObjects
+
+class CancellationSourceTests: XCTestCase {
+
+    func testTaskCancellation() async throws {
+        let source = await CancellationSource()
+        let task = Task {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+        }
+        await source.register(task: task)
+        await source.cancel()
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func testTaskCancellationWithTimeout() async throws {
+        let source = CancellationSource(cancelAfterNanoseconds: UInt64(5E9))
+        let task = Task {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+        }
+        await source.register(task: task)
+        try await Task.sleep(nanoseconds: UInt64(6E9))
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func testTaskCancellationWithLinkedSource() async throws {
+        let parentSource = await CancellationSource()
+        let source = await CancellationSource(linkedWith: parentSource)
+        let task = Task {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+        }
+        await source.register(task: task)
+        await parentSource.cancel()
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func testTaskCancellationWithMultipleLinkedSources() async throws {
+        let parentSource1 = await CancellationSource()
+        let parentSource2 = await CancellationSource()
+        let source = await CancellationSource(
+            linkedWith: parentSource1, parentSource2
+        )
+        let task = Task {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+        }
+        await source.register(task: task)
+        await parentSource1.cancel()
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func testTaskCancellationWithSourcePassedOnInitialization() async throws {
+        let source = await CancellationSource()
+        let task = await Task(cancellationSource: source) {
+            do {
+                try await Task.sleep(nanoseconds: UInt64(10E9))
+                XCTFail()
+            } catch {}
+        }
+        await source.cancel()
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func testDetachedTaskCancellationWithSourcePassedOnInitialization()
+        async throws
+    {
+        let source = await CancellationSource()
+        let task = await Task.detached(cancellationSource: source) {
+            do {
+                try await Task.sleep(nanoseconds: UInt64(10E9))
+                XCTFail()
+            } catch {}
+        }
+        await source.cancel()
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func testThrowingTaskCancellationWithSourcePassedOnInitialization()
+        async throws
+    {
+        let source = await CancellationSource()
+        let task = try await Task(cancellationSource: source) {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+        }
+        await source.cancel()
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func testThrowingDetachedTaskCancellationWithSourcePassedOnInitialization()
+        async throws
+    {
+        let source = await CancellationSource()
+        let task = try await Task.detached(cancellationSource: source) {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+        }
+        await source.cancel()
+        XCTAssertTrue(task.isCancelled)
+    }
+
+    func createTaskWithCancellationSource(
+        _ source: CancellationSource
+    ) -> Task<Void, Never> {
+        return Task(cancellationSource: source) {
+            do {
+                try await Task.sleep(nanoseconds: UInt64(10E9))
+                XCTAssertTrue(Task.isCancelled)
+            } catch {}
+        }
+    }
+
+    func testTaskCancellationWithSourcePassedOnSyncInitialization() async throws
+    {
+        let source = await CancellationSource()
+        let task = createTaskWithCancellationSource(source)
+        Task {
+            try await Task.sleep(nanoseconds: UInt64(2E9))
+            await source.cancel()
+        }
+        await task.value
+    }
+
+    func createDetachedTaskWithCancellationSource(
+        _ source: CancellationSource
+    ) -> Task<Void, Never> {
+        return Task.detached(cancellationSource: source) {
+            do {
+                try await Task.sleep(nanoseconds: UInt64(10E9))
+                XCTFail()
+            } catch {}
+        }
+    }
+
+    func testDetachedTaskCancellationWithSourcePassedOnSyncInitialization()
+        async throws
+    {
+        let source = await CancellationSource()
+        let task = createDetachedTaskWithCancellationSource(source)
+        Task {
+            try await Task.sleep(nanoseconds: UInt64(2E9))
+            await source.cancel()
+        }
+        await task.value
+    }
+
+    func createThrowingTaskWithCancellationSource(
+        _ source: CancellationSource
+    ) throws -> Task<Void, Error> {
+        return try Task(cancellationSource: source) {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+            XCTFail()
+        }
+    }
+
+    func testThrowingTaskCancellationWithSourcePassedOnSyncInitialization()
+        async throws
+    {
+        let source = await CancellationSource()
+        let task = try createThrowingTaskWithCancellationSource(source)
+        Task {
+            try await Task.sleep(nanoseconds: UInt64(2E9))
+            await source.cancel()
+        }
+        let value: Void? = try? await task.value
+        XCTAssertNil(value)
+    }
+
+    func createThrowingDetachedTaskWithCancellationSource(
+        _ source: CancellationSource
+    ) throws -> Task<Void, Error> {
+        return try Task.detached(cancellationSource: source) {
+            try await Task.sleep(nanoseconds: UInt64(10E9))
+            XCTFail()
+        }
+    }
+
+    func
+        testThrowingDetachedTaskCancellationWithSourcePassedOnSyncInitialization()
+        async throws
+    {
+        let source = await CancellationSource()
+        let task = try createThrowingDetachedTaskWithCancellationSource(source)
+        Task {
+            try await Task.sleep(nanoseconds: UInt64(2E9))
+            await source.cancel()
+        }
+        let value: Void? = try? await task.value
+        XCTAssertNil(value)
+    }
+}