divkit/client/ios/DivKitTests/DivKitTests.swift

@testable @_spi(Internal) import DivKit
import LayoutKit
import Serialization
import VGSL
import XCTest

final class DivKitTests: XCTestCase {
  func test_multithreaded_blockCreation() {
    let expectation = XCTestExpectation()
    guard let data = try? dataFromFile(named: "heavy", subdirectory: nil) else {
      XCTFail()
      return
    }
    let components = DivKitComponents()
    let counter = Atomic(initialValue: 0)

    for i in 0...50 {
      let cardId = DivCardID(rawValue: "card_\(i)")
      let result = try! components.parseDivDataWithTemplates(data, cardId: cardId)
      onBackgroundThread {
        let context = components.makeContext(cardId: cardId, cachedImageHolders: [])
        _ = try! result.value!.makeBlock(context: context)

        counter.accessWrite { counter in
          counter += 1
          if counter == 50 {
            expectation.fulfill()
          }
        }
      }
    }
    wait(for: [expectation])
  }
}

extension TemplateValue where ResolvedValue: DivBlockModeling {
  static func make(
    fromFile filename: String,
    subdirectory: String? = nil,
    context: DivBlockModelingContext
  ) throws -> Block {
    let dict = try jsonDictFromFile(
      named: filename,
      subdirectory: "unit_test_data/" + (subdirectory ?? "")
    )
    let templatesDict = (dict["templates"] as? [String: Any]) ?? [:]
    let templates = DivTemplates(dictionary: templatesDict)
    let div = templates.parseValue(type: Self.self, from: dict)
    return try div.value!.makeBlock(context: context)
  }
}

private func dataFromFile(named name: String, subdirectory: String?) throws -> Data {
  let bundle = Bundle(for: DivKitTests.self)
  let url = bundle.url(forResource: name, withExtension: "json", subdirectory: subdirectory)!
  return try Data(contentsOf: url)
}

private func jsonDictFromFile(named name: String, subdirectory: String?) throws -> [String: Any] {
  let data = try dataFromFile(named: name, subdirectory: subdirectory)
  return try JSONSerialization.jsonObject(with: data, options: []) as! [String: Any]
}

extension Deserializable {
  init(file name: String) throws {
    let bundle = Bundle(for: DivKitTests.self)
    let url = bundle.url(forResource: name, withExtension: "json")!
    let data = try Data(contentsOf: url)
    try self.init(JSONData: data)
  }
}

public func XCTAssertThrowsError<T: Equatable & Error>(
  _ expression: @autoclosure () throws -> some Any,
  _ expectedError: T
) {
  XCTAssertThrowsError(try expression()) { error in
    XCTAssertEqual(error as? T, expectedError)
  }
}