Change GCD Leak Test

ably · Jul 1, 2021 · 0f8e29f · 0f8e29f · maratal · Jul 1, 2021
1 parent 20b8887
commit 0f8e29f
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/Spec/GCDTest.swift b/Spec/GCDTest.swift
@@ -10,18 +10,21 @@ import XCTest
 class GCDTest: XCTestCase {
     func testScheduledBlockHandleDerefsBlockAfterInvoke() {
         let invokedExpectation = self.expectation(description: "scheduled block invoked")
-        let obj = NSObject()
-
-        var scheduledBlock = artDispatchScheduled(0, .main) {
-            _ = obj
+
+        var object = NSObject()
+        weak var weakObject = object
+
+        var scheduledBlock = artDispatchScheduled(0, .main) { [object] in
+            _ = object
             invokedExpectation.fulfill()
         }
 
+        _ = scheduledBlock
+
         waitForExpectations(timeout: 2, handler: nil)
-
-        _ = scheduledBlock // silence warning
         scheduledBlock = nil
 
-        XCTAssertEqual(CFGetRetainCount(obj), 1)
+        object = NSObject()
+        XCTAssertNil(weakObject)
     }
 }