Fix flaky test

home-assistant · Dec 7, 2023 · a6b987b · a6b987b
1 parent 65244eb
commit a6b987b
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/Tests/Shared/Webhook/WebhookResponseUpdateComplications.test.swift b/Tests/Shared/Webhook/WebhookResponseUpdateComplications.test.swift
@@ -8,7 +8,6 @@ class WebhookResponseUpdateComplicationsTests: XCTestCase {
     private var api: FakeHomeAssistantAPI!
     private var webhookManager: FakeWebhookManager!
     private var realm: Realm!
-    private var handler: WebhookResponseUpdateComplications?
 
     override func setUpWithError() throws {
         try super.setUpWithError()
@@ -25,6 +24,12 @@ class WebhookResponseUpdateComplicationsTests: XCTestCase {
         FakeWatchComplication.rawRenderedUpdates = [:]
     }
 
+    override func tearDown() {
+        api = nil
+        webhookManager = nil
+        realm = nil
+    }
+
     func testNoComplicationGivesNoRequest() {
         XCTAssertNil(WebhookResponseUpdateComplications.request(for: .init()))
     }
@@ -127,8 +132,8 @@ class WebhookResponseUpdateComplicationsTests: XCTestCase {
             realm.add(complications)
         }
 
-        handler = WebhookResponseUpdateComplications(api: api)
-        handler?.watchComplicationClass = FakeWatchComplication.self
+        var handler = WebhookResponseUpdateComplications(api: api)
+        handler.watchComplicationClass = FakeWatchComplication.self
 
         let request = WebhookResponseUpdateComplications.request(for: Set(complications))!
         let result: [String: Any] = [
@@ -137,8 +142,6 @@ class WebhookResponseUpdateComplicationsTests: XCTestCase {
             "c3|fwc3k1": 3,
         ]
 
-        guard let handler = handler else { return }
-
         let expectation = self.expectation(description: "result")
         handler.handle(request: .value(request), result: .value(result)).done { handlerResult in
             XCTAssertNil(handlerResult.notification)