Add more defer { task.removeAllObservers() } in tests

firebase · Jan 22, 2024 · 3d37807 · 3d37807
1 parent 9804dd4
commit 3d37807
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/FirebaseStorage/Tests/Integration/StorageIntegration.swift b/FirebaseStorage/Tests/Integration/StorageIntegration.swift
@@ -244,6 +244,9 @@ class StorageResultTests: StorageIntegrationCommon {
       self.assertResultSuccess(result)
       putFileExpectation.fulfill()
     }
+    defer {
+      task.removeAllObservers()
+    }
 
     task.observe(StorageTaskStatus.success) { snapshot in
       XCTAssertEqual(snapshot.description, "<State: Success>")
@@ -292,7 +295,6 @@ class StorageResultTests: StorageIntegrationCommon {
       self.assertResultSuccess(result)
       putFileExpectation.fulfill()
     }
-
     defer {
       task.removeAllObservers()
     }
@@ -347,7 +349,6 @@ class StorageResultTests: StorageIntegrationCommon {
       self.assertResultSuccess(result)
       putFileExpectation.fulfill()
     }
-
     defer {
       task.removeAllObservers()
     }
@@ -552,6 +553,9 @@ class StorageResultTests: StorageIntegrationCommon {
       switch result {
       case .success:
         let task = ref.write(toFile: fileURL)
+        defer {
+          task.removeAllObservers()
+        }
 
         task.observe(StorageTaskStatus.success) { snapshot in
           do {
@@ -594,6 +598,9 @@ class StorageResultTests: StorageIntegrationCommon {
       switch result {
       case .success:
         let task = ref.write(toFile: fileURL)
+        defer {
+          task.removeAllObservers()
+        }
 
         task.observe(StorageTaskStatus.success) { snapshot in
           XCTFail("Error processing success snapshot")
@@ -703,6 +710,9 @@ class StorageResultTests: StorageIntegrationCommon {
 
     let fileURL = URL(fileURLWithPath: "\(NSTemporaryDirectory())/hello.txt")
     let task = ref.write(toFile: fileURL)
+    defer {
+      task.removeAllObservers()
+    }
     var downloadedBytes: Int64 = 0
     var resumeAtBytes = 256 * 1024
 
@@ -742,6 +752,9 @@ class StorageResultTests: StorageIntegrationCommon {
 
     let fileURL = URL(fileURLWithPath: "\(NSTemporaryDirectory())/hello.txt")
     let task = ref.write(toFile: fileURL)
+    defer {
+      task.removeAllObservers()
+    }
     var downloadedBytes: Int64 = 0
     var resumeAtBytes = 256 * 1024