adds a unit test for deallocation

DP-3T · May 12, 2021 · 7efb508 · 7efb508
1 parent 4bf4c19
commit 7efb508
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/Tests/DP3TSDKTests/DP3TSDKTests.swift b/Tests/DP3TSDKTests/DP3TSDKTests.swift
@@ -319,8 +319,19 @@ class DP3TSDKTests: XCTestCase {
         wait(for: [secondInfected], timeout: 1.0)
 
         XCTAssertEqual(sdk.status.infectionStatus, .infected)
+    }
 
+    func testCorrectDeallocation() {
+        weak var weakRef = sdk
+        sdk.reset()
+        sdk = nil
+        let expt = expectation(description: "deallocated")
 
+        DispatchQueue.main.async {
+            XCTAssertNil(weakRef)
+            expt.fulfill()
+        }
 
+        wait(for: [expt], timeout: 1.0)
     }
 }