Inject URLSession in EventLoggingNetworkService

Automattic · Mar 25, 2020 · fac0f65 · fac0f65
1 parent 4f20dc6
commit fac0f65
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/Automattic-Tracks-iOS/Services/EventLoggingNetworkService.swift b/Automattic-Tracks-iOS/Services/EventLoggingNetworkService.swift
@@ -4,9 +4,15 @@ typealias NetworkResultCallback = (Result<Data?, Error>) -> Void
 
 open class EventLoggingNetworkService {
 
+
+    private let urlSession: URLSession
+
+    init(urlSession: URLSession = URLSession.shared) {
+        self.urlSession = urlSession
+    }
+
     func uploadFile(request: URLRequest, fileURL: URL, completion: @escaping NetworkResultCallback) {
-        URLSession.shared
-            .uploadTask(with: request, fromFile: fileURL, completionHandler: { data, response, error in
+        urlSession.uploadTask(with: request, fromFile: fileURL, completionHandler: { data, response, error in
 
             if let error = error {
                 completion(.failure(error))