facebook-ios-sdk/TestTools/TestTools/TestGraphRequestConnection.swift

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 * All rights reserved.
 *
 * This source code is licensed under the license found in the
 * LICENSE file in the root directory of this source tree.
 */

import FBSDKCoreKit
import Foundation

@objcMembers
public final class TestGraphRequestConnection: NSObject, GraphRequestConnecting {
  var shouldExecuteCompletion = false
  var error: Error?
  var requestConnectionResult: [String: Any]?

  public override convenience init() {
    self.init(shouldExecuteCompletion: false)
  }

  public convenience init(shouldExecuteCompletion: Bool) {
    self.init(
      shouldExecuteCompletion: shouldExecuteCompletion,
      error: nil,
      requestConnectionResult: nil
    )
  }

  public init(shouldExecuteCompletion: Bool, error: Error?, requestConnectionResult: [String: Any]?) {
    super.init()
    self.shouldExecuteCompletion = shouldExecuteCompletion
    self.error = error
    self.requestConnectionResult = requestConnectionResult
  }

  public var requests: NSMutableArray {
    NSMutableArray(
      array: graphRequests.compactMap {
        GraphRequestMetadata(
          request: $0.request,
          completionHandler: $0.completion,
          batchParameters: nil
        )
      }
    )
  }

  public typealias Completion = @convention(block) (GraphRequestConnecting, Any?, Error?) -> Void

  public struct Request {
    public let request: GraphRequestProtocol
    public let completion: GraphRequestCompletion
  }

  public var timeout: TimeInterval = 0
  public var delegate: GraphRequestConnectionDelegate? // swiftlint:disable:this weak_delegate
  public var capturedRequest: GraphRequestProtocol?
  public var capturedRequests = [GraphRequestProtocol]()
  public var capturedCompletion: GraphRequestCompletion?
  public var capturedCompletions = [Completion]()
  public var startCallCount = 0
  public var cancelCallCount = 0

  // Enables us to use the internal requests property
  public var graphRequests = [Request]()

  public func add(_ request: GraphRequestProtocol, completion handler: @escaping GraphRequestCompletion) {
    graphRequests.append(
      Request(
        request: request,
        completion: handler
      )
    )
    capturedRequest = request
    capturedRequests.append(request)
    capturedCompletion = handler
    capturedCompletions.append(handler)
  }

  public func start() {
    startCallCount += 1
    if shouldExecuteCompletion {
      for completion in capturedCompletions {
        completion(self, requestConnectionResult, error)
      }
    }
  }

  public func cancel() {
    cancelCallCount += 1
  }
}