Updated unit tests

nickolas-pohilets · nickolas-pohilets · commit 292f3e1073f8 · 2022-07-29T20:34:54.000+02:00
diff --git a/test/Concurrency/voucher_propagation.swift b/test/Concurrency/voucher_propagation.swift
@@ -76,7 +76,7 @@ actor Counter {
   func get() -> Int { n }
 }
 
-actor ActorWithIsolatedDeinit {
+actor ActorWithSelfIsolatedDeinit {
   let expectedVoucher: voucher_t?
   let group: DispatchGroup
   
@@ -102,6 +102,25 @@ actor ActorWithIsolatedDeinit {
   }
 }
 
+actor ActorWithDeinitIsolatedOnAnother {
+  let expectedVoucher: voucher_t?
+  let group: DispatchGroup
+  
+  init(expectedVoucher: voucher_t?, group: DispatchGroup) {
+    self.expectedVoucher = expectedVoucher
+    self.group = group
+  }
+  
+  @AnotherActor
+  deinit {
+    expectTrue(isCurrentExecutor(AnotherActor.shared.unownedExecutor))
+    let currentVoucher = voucher_copy()
+    expectEqual(expectedVoucher, currentVoucher)
+    os_release(currentVoucher)
+    group.leave()
+  }
+}
+
 class ClassWithIsolatedDeinit {
   let expectedVoucher: voucher_t?
   let group: DispatchGroup
@@ -413,11 +432,21 @@ if #available(SwiftStdlib 5.1, *) {
     withVouchers { v1, v2, v3 in
       let group = DispatchGroup()
       group.enter()
+      group.enter()
+      group.enter()
       Task {
         await AnotherActor.shared.performTesting {
           adopt(voucher: v1)
           _ = ClassWithIsolatedDeinit(expectedVoucher: v1, group: group)
         }
+        await AnotherActor.shared.performTesting {
+          adopt(voucher: v2)
+          _ = ActorWithSelfIsolatedDeinit(expectedVoucher: v2, group: group)
+        }
+        await AnotherActor.shared.performTesting {
+          adopt(voucher: v3)
+          _ = ActorWithDeinitIsolatedOnAnother(expectedVoucher: v3, group: group)
+        }
       }
       group.wait()
     }
@@ -430,12 +459,12 @@ if #available(SwiftStdlib 5.1, *) {
       group.enter()
       Task {
         do {
-          adopt(voucher: v2)
-          _ = ActorWithIsolatedDeinit(expectedVoucher: v2, group: group)
+          adopt(voucher: v1)
+          _ = ActorWithDeinitIsolatedOnAnother(expectedVoucher: v1, group: group)
         }
         do {
-          adopt(voucher: v3)
-          _ = ClassWithIsolatedDeinit(expectedVoucher: v3, group: group)
+          adopt(voucher: v2)
+          _ = ClassWithIsolatedDeinit(expectedVoucher: v2, group: group)
         }
       }
       group.wait()
diff --git a/test/Distributed/Runtime/distributed_actor_deinit.swift b/test/Distributed/Runtime/distributed_actor_deinit.swift
@@ -14,6 +14,28 @@ import Distributed
 @preconcurrency import Dispatch
 import StdlibUnittest
 
+#if canImport(Darwin)
+import Darwin
+typealias ThreadID = pthread_t
+func getCurrentThreadID() -> ThreadID { pthread_self() }
+func equalThreadIDs(_ a: ThreadID, _ b: ThreadID) -> Bool { pthread_equal(a, b) != 0 }
+#elseif canImport(Glibc)
+import Glibc
+typealias ThreadID = pthread_t
+func getCurrentThreadID() -> ThreadID { pthread_self() }
+func equalThreadIDs(_ a: ThreadID, _ b: ThreadID) -> Bool { pthread_equal(a, b) != 0 }
+#elseif os(Windows)
+import WinSDK
+typealias ThreadID = UInt32
+func getCurrentThreadID() -> ThreadID { GetCurrentThreadId() }
+func equalThreadIDs(_ a: ThreadID, _ b: ThreadID) -> Bool { a == b }
+#endif
+
+var mainThread: ThreadID?
+func isMainThread() -> Bool {
+    return equalThreadIDs(getCurrentThreadID(), mainThread!)
+}
+
 // Look up a symbol using dlsym and cast the result to an arbitrary type.
 func lookup<T>(_ name: String) -> T {
   let RTLD_DEFAULT = UnsafeMutableRawPointer(bitPattern: -2)
@@ -33,7 +55,7 @@ func isCurrent(_ a: AnyActor) -> Bool {
   return isCurrentExecutor(getExecutor(a))
 }
 
-func isMainThread() -> Bool {
+func isMainExecutor() -> Bool {
   isCurrentExecutor(getMainExecutor())
 }
 
@@ -59,7 +81,7 @@ distributed actor DA_userDefined_nonisolated {
   }
 
   nonisolated deinit {
-    print("Deinitializing \(self.id) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self))")
+    print("Deinitializing \(self.id) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self)) mainThread:\(isMainThread())")
   }
 }
 
@@ -69,7 +91,7 @@ distributed actor DA_userDefined_isolated {
   }
 
   deinit {
-    print("Deinitializing \(self.id) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self))")
+    print("Deinitializing \(self.id) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self)) mainThread:\(isMainThread())")
   }
 }
 
@@ -84,7 +106,7 @@ distributed actor DA_state_nonisolated {
   }
 
   nonisolated deinit {
-    print("Deinitializing \(self.id) name=\(name) age=\(age) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self))")
+    print("Deinitializing \(self.id) name=\(name) age=\(age) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self)) mainThread:\(isMainThread())")
     return
   }
 }
@@ -100,7 +122,28 @@ distributed actor DA_state_isolated {
   }
 
   deinit {
-    print("Deinitializing \(self.id) name=\(name) age=\(age) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self))")
+    print("Deinitializing \(self.id) name=\(name) age=\(age) remote:\(__isRemoteActor(self)) isolated:\(isCurrent(self)) mainThread:\(isMainThread())")
+    return
+  }
+}
+
+@globalActor actor AnotherActor: GlobalActor {
+  static let shared = AnotherActor()
+}
+
+distributed actor DA_state_isolated_on_another {
+  var name: String
+  var age: Int
+
+  init(name: String, age: Int, system: FakeActorSystem) {
+    self.name = name
+    self.age = age
+    self.actorSystem = system
+  }
+
+  @AnotherActor
+  deinit {
+    print("Deinitializing \(self.id) name=\(name) age=\(age) remote:\(__isRemoteActor(self)) isolated-self:\(isCurrent(self)) isolated-other:\(isCurrent(AnotherActor.shared)) mainThread:\(isMainThread())")
     return
   }
 }
@@ -129,7 +172,7 @@ final class FakeActorSystem: @unchecked Sendable, DistributedActorSystem {
   }
   
   deinit {
-    print("Deinit ActorSystem: mainThread=\(isMainThread())")
+    print("Deinit ActorSystem: mainExecutor=\(isMainExecutor()) mainThread=\(isMainThread())")
     group.leave()
   }
 
@@ -254,7 +297,7 @@ func test() {
   // CHECK: assign type:DA, address:[[ADDRESS:.*]]
   // CHECK: ready actor:main.DA, address:ActorAddress(address: "[[ADDR1:addr-[0-9]]]")
   // CHECK: resign address:ActorAddress(address: "[[ADDR1]]")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=true
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=true mainThread=true
 
   group.enter()
   _ = { () -> DA_userDefined in
@@ -264,7 +307,7 @@ func test() {
   // CHECK: assign type:DA_userDefined, address:[[ADDRESS:.*]]
   // CHECK: ready actor:main.DA_userDefined, address:ActorAddress(address: "[[ADDR2:addr-[0-9]]]")
   // CHECK: resign address:ActorAddress(address: "[[ADDR2]]")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=true
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=true mainThread=true
 
   // resign must happen as the _last thing_ after user-deinit completed
   group.enter()
@@ -274,9 +317,9 @@ func test() {
   group.wait()
   // CHECK: assign type:DA_userDefined_nonisolated, address:[[ADDRESS:.*]]
   // CHECK: ready actor:main.DA_userDefined_nonisolated, address:ActorAddress(address: "[[ADDR3:addr-[0-9]]]")
-  // CHECK: Deinitializing ActorAddress(address: "[[ADDR3]]") remote:false isolated:false
+  // CHECK: Deinitializing ActorAddress(address: "[[ADDR3]]") remote:false isolated:false mainThread:true
   // CHECK-NEXT: resign address:ActorAddress(address: "[[ADDR3]]")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=true
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=true mainThread=true
   
   // resign must happen as the _last thing_ after user-deinit completed
   group.enter()
@@ -286,9 +329,9 @@ func test() {
   group.wait()
   // CHECK: assign type:DA_userDefined_isolated, address:[[ADDRESS:.*]]
   // CHECK: ready actor:main.DA_userDefined_isolated, address:ActorAddress(address: "[[ADDR4:addr-[0-9]]]")
-  // CHECK: Deinitializing ActorAddress(address: "[[ADDR4]]") remote:false isolated:true
+  // CHECK: Deinitializing ActorAddress(address: "[[ADDR4]]") remote:false isolated:true mainThread:true
   // CHECK-NEXT: resign address:ActorAddress(address: "[[ADDR4]]")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=false
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=false mainThread=true
 
   // resign must happen as the _last thing_ after user-deinit completed
   group.enter()
@@ -298,9 +341,9 @@ func test() {
   group.wait()
   // CHECK: assign type:DA_state_nonisolated, address:[[ADDRESS:.*]]
   // CHECK: ready actor:main.DA_state_nonisolated, address:ActorAddress(address: "[[ADDR5:addr-[0-9]]]")
-  // CHECK: Deinitializing ActorAddress(address: "[[ADDR5]]") name=Foo age=37 remote:false isolated:false
+  // CHECK: Deinitializing ActorAddress(address: "[[ADDR5]]") name=Foo age=37 remote:false isolated:false mainThread:true
   // CHECK-NEXT: resign address:ActorAddress(address: "[[ADDR5]]")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=true
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=true mainThread=true
   
   // resign must happen as the _last thing_ after user-deinit completed
   group.enter()
@@ -310,9 +353,21 @@ func test() {
   group.wait()
   // CHECK: assign type:DA_state_isolated, address:[[ADDRESS:.*]]
   // CHECK: ready actor:main.DA_state_isolated, address:ActorAddress(address: "[[ADDR6:addr-[0-9]]]")
-  // CHECK: Deinitializing ActorAddress(address: "[[ADDR6]]") name=Bar age=42 remote:false isolated:true
+  // CHECK: Deinitializing ActorAddress(address: "[[ADDR6]]") name=Bar age=42 remote:false isolated:true mainThread:true
+  // CHECK-NEXT: resign address:ActorAddress(address: "[[ADDR6]]")
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=false mainThread=true
+  
+  // resign must happen as the _last thing_ after user-deinit completed
+  group.enter()
+  _ = { () -> DA_state_isolated_on_another in
+    DA_state_isolated_on_another(name: "Baz", age:57, system: DefaultDistributedActorSystem(group: group))
+  }()
+  group.wait()
+  // CHECK: assign type:DA_state_isolated_on_another, address:[[ADDRESS:.*]]
+  // CHECK: ready actor:main.DA_state_isolated_on_another, address:ActorAddress(address: "[[ADDR6:addr-[0-9]]]")
+  // CHECK: Deinitializing ActorAddress(address: "[[ADDR6]]") name=Baz age=57 remote:false isolated-self:false isolated-other:true mainThread:false
   // CHECK-NEXT: resign address:ActorAddress(address: "[[ADDR6]]")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=false
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=false mainThread=false
 
   // a remote actor should not resign it's address, it was never "assigned" it
   group.enter()
@@ -324,7 +379,7 @@ func test() {
   // CHECK-NEXT: resolve type:DA_userDefined_nonisolated, address:ActorAddress(address: "remote-1")
   // MUST NOT run deinit body for a remote distributed actor
   // CHECK-NOT: Deinitializing ActorAddress(address: "remote-1")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=true
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=true mainThread=true
   
   // a remote actor should not resign it's address, it was never "assigned" it
   group.enter()
@@ -336,14 +391,27 @@ func test() {
   // CHECK-NEXT: resolve type:DA_userDefined_isolated, address:ActorAddress(address: "remote-2")
   // MUST NOT run deinit body for a remote distributed actor
   // CHECK-NOT: Deinitializing ActorAddress(address: "remote-2")
-  // CHECK-NEXT: Deinit ActorSystem: mainThread=true
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=true mainThread=true
+  
+  // a remote actor should not resign it's address, it was never "assigned" it
+  group.enter()
+  _ = { () -> DA_state_isolated_on_another in
+    let address = ActorAddress(parse: "remote-3")
+    return try! DA_state_isolated_on_another.resolve(id: address, using: DefaultDistributedActorSystem(group: group))
+  }()
+  group.wait()
+  // CHECK-NEXT: resolve type:DA_state_isolated_on_another, address:ActorAddress(address: "remote-3")
+  // MUST NOT run deinit body for a remote distributed actor
+  // CHECK-NOT: Deinitializing ActorAddress(address: "remote-3")
+  // CHECK-NEXT: Deinit ActorSystem: mainExecutor=true mainThread=true
 
   print("DONE")
   // CHECK-NEXT: DONE
 }
 
 @main struct Main {
   static func main() async {
+    mainThread = getCurrentThreadID()
     test()
   }
 }
