Pulse/Tests/PulseTests/URLSessionProxyTests.swift

// The MIT License (MIT)
//
// Copyright (c) 2020-2022 Alexander Grebenyuk (github.com/kean).

import XCTest
import Foundation
import CoreData
@testable import Pulse

final class URLSessionProxyTests: XCTestCase {
    let directory = TemporaryDirectory()
    var store: LoggerStore!
    var logger: NetworkLogger!

    override func setUp() {
        super.setUp()

        let storeURL = directory.url.appending(filename: "logs.pulse")
        store = try! LoggerStore(storeURL: storeURL, options: [.create, .synchronous])
        logger = NetworkLogger(store: store)

        Experimental.URLSessionProxy.shared.logger = logger
        Experimental.URLSessionProxy.shared.isEnabled = true
    }

    override func tearDown() {
        super.tearDown()

        try? store.destroy()

        directory.remove()

        Experimental.URLSessionProxy.shared.isEnabled = false
    }

    func testRecordSuccess() throws {
        // WHEN
        let dataURL = directory.url.appending(filename: "logs-archive-v2.pulse")
        try Resources.pulseArchive.write(to: dataURL)
        let didComplete = self.expectation(description: "TaskCompleted")
        let dataTask = URLSession.shared.dataTask(with: dataURL) { _, _, _ in
            didComplete.fulfill()
        }
        dataTask.resume()
        wait(for: [didComplete], timeout: 5)

        // RECORD
        let tasks = try store.allTasks()
        let task = try XCTUnwrap(tasks.first)

        // THEN
        XCTAssertEqual(tasks.count, 1)

        XCTAssertEqual(task.url, dataURL.absoluteString)
        XCTAssertEqual(task.host, nil)
        XCTAssertEqual(task.httpMethod, "GET")
        XCTAssertNil(task.errorDomain)
        XCTAssertEqual(task.errorCode, 0)
        XCTAssertEqual(task.requestState, NetworkTaskEntity.State.success.rawValue)

        let message = try XCTUnwrap(task.message)
        XCTAssertEqual(message.label.name, "network")
    }

    func testRecordError() throws {
        // GIVEN file that doesn't exist
        let dataURL = FileManager.default.temporaryDirectory.appending(filename: UUID().uuidString)

        // WHEN
        let didComplete = self.expectation(description: "TaskCompleted")
        let dataTask = URLSession.shared.dataTask(with: dataURL) { _, _, _ in
            didComplete.fulfill()
        }
        dataTask.resume()
        wait(for: [didComplete], timeout: 5)

        // RECORD
        let tasks = try store.allTasks()
        let task = try XCTUnwrap(tasks.first)

        // THEN
        XCTAssertEqual(tasks.count, 1)

        XCTAssertEqual(task.url, dataURL.absoluteString)
        XCTAssertEqual(task.host, nil)
        XCTAssertEqual(task.httpMethod, "GET")
        XCTAssertEqual(task.errorDomain, URLError.errorDomain)
        XCTAssertEqual(task.errorCode, -1100)
        XCTAssertEqual(task.requestState, NetworkTaskEntity.State.failure.rawValue)

        let message = try XCTUnwrap(task.message)
        XCTAssertEqual(message.label.name, "network")
    }
}