fix: [TMP] Try fix flaky tests, part 2

Signed-off-by: Fabrizio Demaria <[email protected]>
spotify · Jan 12, 2024 · 957f127 · 957f127
1 parent 2fd8824
commit 957f127
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 37 deletions.
diff --git a/Sources/ConfidenceProvider/Apply/FlagApplierWithRetries.swift b/Sources/ConfidenceProvider/Apply/FlagApplierWithRetries.swift
@@ -131,27 +131,25 @@ final class FlagApplierWithRetries: FlagApplier {
             sdk: Sdk(id: metadata.name, version: metadata.version)
         )
 
-        performRequest(request: request) { result in
-            Task {
-                switch result {
-                case .success:
-                    await completion(true)
-                case .failure(let error):
-                    self.logApplyError(error: error)
-                    await completion(false)
-                }
+        await performRequest(request: request) { result in
+            switch result {
+            case .success:
+                await completion(true)
+            case .failure(let error):
+                self.logApplyError(error: error)
+                await completion(false)
             }
         }
     }
 
     private func performRequest(
         request: ApplyFlagsRequest,
-        completion: @escaping (ApplyFlagResult) -> Void
-    ) {
+        completion: @escaping (ApplyFlagResult) async -> Void
+    ) async {
         do {
-            try httpClient.post(path: ":apply", data: request, completion: completion)
+            try await httpClient.post(path: ":apply", data: request, completion: completion)
         } catch {
-            completion(.failure(handleError(error: error)))
+            await completion(.failure(handleError(error: error)))
         }
     }
 

diff --git a/Sources/ConfidenceProvider/Http/HttpClient.swift b/Sources/ConfidenceProvider/Http/HttpClient.swift
@@ -6,8 +6,8 @@ protocol HttpClient {
     func post<T: Decodable>(
         path: String,
         data: Codable,
-        completion: @escaping (HttpClientResult<T>) -> Void
-    ) throws
+        completion: @escaping (HttpClientResult<T>) async -> Void
+    ) async throws
 
     func post<T: Decodable>(path: String, data: Codable) async throws -> HttpClientResponse<T>
 }

diff --git a/Sources/ConfidenceProvider/Http/NetworkClient.swift b/Sources/ConfidenceProvider/Http/NetworkClient.swift
@@ -44,28 +44,30 @@ final class NetworkClient: HttpClient {
     func post<T: Decodable>(
         path: String,
         data: Codable,
-        completion: @escaping (HttpClientResult<T>) -> Void
-    ) throws {
+        completion: @escaping (HttpClientResult<T>) async -> Void
+    ) async throws {
         let request = try buildRequest(path: path, data: data)
-        perform(request: request, retry: self.retry) { response, data, error in
-            if let error {
-                completion(.failure(error))
-                return
-            }
 
-            guard let response, let data else {
-                completion(.failure(ConfidenceError.internalError(message: "Bad response")))
-                return
-            }
 
-            do {
-                let httpClientResult: HttpClientResponse<T> =
-                    try self.buildResponse(response: response, data: data)
-                completion(.success(httpClientResult))
-            } catch {
-                completion(.failure(error))
-            }
+        let (data, response) = try await URLSession.shared.data(for: request)
+        guard let response = response as? HTTPURLResponse, response.statusCode == 202 else {
+            await completion(.failure(HttpClientError.internalError))
+            return
         }
+//
+//        guard let response, let data else {
+//            completion(.failure(ConfidenceError.internalError(message: "Bad response")))
+//            return
+//        }
+
+        do {
+            let httpClientResult: HttpClientResponse<T> =
+            try self.buildResponse(response: response, data: data)
+            await completion(.success(httpClientResult))
+        } catch {
+            await completion(.failure(error))
+        }
+
     }
 
     private func perform(

diff --git a/Tests/ConfidenceProviderTests/Helpers/HttpClientMock.swift b/Tests/ConfidenceProviderTests/Helpers/HttpClientMock.swift
@@ -22,13 +22,13 @@ final class HttpClientMock: HttpClient {
     func post<T>(
         path: String,
         data: Codable,
-        completion: @escaping (ConfidenceProvider.HttpClientResult<T>) -> Void
-    ) throws where T: Decodable {
+        completion: @escaping (ConfidenceProvider.HttpClientResult<T>) async -> Void
+    ) async throws where T: Decodable {
         do {
             let result: HttpClientResponse<T> = try handlePost(path: path, data: data)
-            completion(.success(result))
+            await completion(.success(result))
         } catch {
-            completion(.failure(error))
+            await completion(.failure(error))
         }
     }