Add test for new case

home-assistant · Apr 2, 2024 · 560c62a · 560c62a
1 parent f5547e7
commit 560c62a
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/Source/Internal/ResponseController/HAResponseController.swift b/Source/Internal/ResponseController/HAResponseController.swift
@@ -126,6 +126,7 @@ internal class HAResponseControllerImpl: HAResponseController {
             phase = .disconnected(error: error, forReset: false)
         case .peerClosed:
             HAGlobal.log(.info, "Peer closed")
+            phase = .disconnected(error: nil, forReset: false)
         }
     }
 

diff --git a/Tests/HAResponseController.test.swift b/Tests/HAResponseController.test.swift
@@ -72,6 +72,14 @@ internal class HAResponseControllerTests: XCTestCase {
         }
     }
 
+    func testPeerClosedEvent() {
+        fireConnected()
+        controller.didReceive(event: .peerClosed)
+        waitForCallback()
+        XCTAssertEqual(delegate.lastPhase, .disconnected(error: nil, forReset: false))
+        XCTAssertNil(delegate.lastReceived)
+    }
+
     func testAuthFlow() throws {
         fireConnected()
         try fireText(