Fix unit test

home-assistant · Nov 28, 2023 · 57470c9 · 57470c9
1 parent 109addd
commit 57470c9
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/Sources/App/ThreadCredentialsSharing/Tests/ThreadCredentialsSharingViewModelTests.swift b/Sources/App/ThreadCredentialsSharing/Tests/ThreadCredentialsSharingViewModelTests.swift
@@ -19,18 +19,12 @@ final class ThreadCredentialsSharingViewModelTests: XCTestCase {
         mockClient = nil
     }
 
-    func test_retrieveAllCredentials_calls_retrieveAllCredentials() {
-        // Given
-        let expectation = XCTestExpectation(description: "retrieveAllCredentials returned")
-
+    func test_retrieveAllCredentials_calls_retrieveAllCredentials() async {
         // When
-        Task {
-            await sut.retrieveAllCredentials()
+        await sut.retrieveAllCredentials()
 
-            // Then
-            XCTAssertTrue(mockClient.retrieveAllCredentialsCalled)
-            XCTAssertEqual(sut.credentials.count, 2)
-            expectation.fulfill()
-        }
+        // Then
+        XCTAssertTrue(mockClient.retrieveAllCredentialsCalled)
+        XCTAssertEqual(sut.credentials.count, 2)
     }
 }