hapi/ios/HapiTests/TranscriptGeometryTests.swift

import UIKit
import XCTest
@testable import HapiUI

@MainActor
final class TranscriptGeometryTests: XCTestCase {
    func testIndexedQueriesMatchLinearGeometryThroughWindowAndWidthChanges() throws {
        let layout = TranscriptLayout()
        let collection = UICollectionView(frame: CGRect(x: 0, y: 0, width: 390, height: 844), collectionViewLayout: layout)
        var ids = (0..<800).map(String.init)
        var heights: [String: CGFloat] = [:]
        layout.setItems(ids, width: 390)
        layout.prepare()
        for index in stride(from: 0, to: ids.count, by: 7) {
            let height = CGFloat(20 + index % 19 * 87)
            heights[ids[index]] = height
            let original = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: index, section: 0)))
            let preferred = original.copy() as! UICollectionViewLayoutAttributes
            preferred.size.height = height
            _ = layout.invalidationContext(forPreferredLayoutAttributes: preferred, withOriginalAttributes: original)
        }
        layout.prepare()

        @MainActor func verify(width: CGFloat) {
            var y: CGFloat = 12
            let contentWidth = HapiReadingLayout.contentWidth(in: width)
            let expected = ids.map { id -> CGRect in
                let frame = CGRect(x: (width - contentWidth) / 2, y: y, width: contentWidth, height: heights[id] ?? 100)
                y = frame.maxY + 12
                return frame
            }
            let actualHeight = layout.collectionViewContentSize.height
            XCTAssertEqual(actualHeight, y, accuracy: 0.01)
            let rects = stride(from: -500, to: Int(y) + 500, by: 377).map {
                CGRect(x: 0, y: $0, width: Int(width), height: 844)
            } + [.zero, .null, .infinite, CGRect(x: 1000, y: 0, width: 100, height: 100000)]
            for rect in rects {
                let indices = expected.indices.filter { expected[$0].intersects(rect) }
                let actual = layout.layoutAttributesForElements(in: rect) ?? []
                XCTAssertEqual(actual.map(\.indexPath.item), indices)
                XCTAssertEqual(actual.map(\.frame), indices.map { expected[$0] })
            }
        }
        verify(width: 390)
        ids = (800..<820).map(String.init) + Array(ids.dropLast(20))
        layout.setItems(ids, width: 390)
        layout.prepare()
        verify(width: 390)
        // Width changes discard stale measurements but retain correct indices.
        heights.removeAll()
        layout.setItems(ids, width: 430)
        layout.prepare()
        verify(width: 430)
        layout.setItems([], width: 430)
        layout.prepare()
        XCTAssertTrue(layout.layoutAttributesForElements(in: .infinite)?.isEmpty == true)
        XCTAssertNil(layout.layoutAttributesForItem(at: IndexPath(item: 0, section: 0)))
        XCTAssertEqual(collection.collectionViewLayout, layout)
    }

    func testStaleSelfSizingAttributesDoNotMoveTheWidthChangeAnchor() throws {
        let layout = TranscriptLayout()
        let collection = UICollectionView(frame: CGRect(x: 0, y: 0, width: 390, height: 100), collectionViewLayout: layout)
        let ids = ["above", "anchor", "below"]
        layout.setItems(ids, width: 390)
        layout.prepare()
        let first = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: 0, section: 0)))
        let measured = first.copy() as! UICollectionViewLayoutAttributes
        measured.size.height = 200
        _ = layout.invalidationContext(forPreferredLayoutAttributes: measured, withOriginalAttributes: first)
        layout.prepare()
        let stale = try XCTUnwrap(layout.layoutAttributesForItem(at: first.indexPath))
        layout.followsTail = false
        layout.anchorForUpdate = ReadingAnchor(id: "anchor", offset: -31)
        layout.setItems(ids, width: 1024)
        layout.prepare()
        let anchor = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: 1, section: 0)))
        collection.contentOffset.y = anchor.frame.minY + 31
        let preferred = stale.copy() as! UICollectionViewLayoutAttributes
        preferred.size.height = 50
        let context = layout.invalidationContext(forPreferredLayoutAttributes: preferred, withOriginalAttributes: stale)
        XCTAssertEqual(context.contentOffsetAdjustment.y, -50, accuracy: 0.01)
        collection.contentOffset.y += context.contentOffsetAdjustment.y
        layout.prepare()
        let updatedAnchor = try XCTUnwrap(layout.layoutAttributesForItem(at: anchor.indexPath))
        XCTAssertEqual(updatedAnchor.frame.minY - collection.contentOffset.y, -31, accuracy: 0.01)
    }

    func testMetricsInvalidateOffscreenHeightsButTabletRecenteringDoesNot() throws {
        let layout = TranscriptLayout()
        let collection = UICollectionView(frame: CGRect(x: 0, y: 0, width: 1024, height: 768), collectionViewLayout: layout)
        let ids = (0..<80).map(String.init)
        let spacing = ids.indices.map { $0 == 30 ? CGFloat(24) : 12 }
        layout.setItems(ids, width: 1024, spacingBefore: spacing)
        layout.prepare()
        let original = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: 30, section: 0)))
        let preferred = original.copy() as! UICollectionViewLayoutAttributes
        preferred.size.height = 450
        _ = layout.invalidationContext(forPreferredLayoutAttributes: preferred, withOriginalAttributes: original)
        layout.prepare()
        layout.setItems(ids, width: 768, spacingBefore: spacing)
        layout.prepare()
        let centered = try XCTUnwrap(layout.layoutAttributesForItem(at: original.indexPath))
        XCTAssertEqual(centered.frame.width, 720)
        XCTAssertEqual(centered.frame.minX, 24)
        XCTAssertEqual(centered.frame.height, 450, "Same reading width retains valid heights")
        let preceding = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: 29, section: 0)))
        XCTAssertEqual(centered.frame.minY - preceding.frame.maxY, 24)
        layout.setItems(ids, width: 768, spacingBefore: spacing, invalidateMeasurements: true)
        layout.prepare()
        XCTAssertEqual(layout.layoutAttributesForItem(at: original.indexPath)?.frame.height, 100)
        XCTAssertEqual(collection.collectionViewLayout, layout)
    }

    #if DEBUG
    func testVisibleQueriesVisitOnlyAViewportSizedRange() {
        let layout = TranscriptLayout()
        layout.setItems((0..<800).map(String.init), width: 390)
        layout.prepare()
        for step in 0..<200 {
            let rect = CGRect(x: 0, y: 1000 + step * 350, width: 390, height: 844)
            let visible = layout.layoutAttributesForElements(in: rect) ?? []
            XCTAssertFalse(visible.isEmpty)
            XCTAssertTrue(visible.allSatisfy { $0.frame.intersects(rect) })
            XCTAssertLessThanOrEqual(visible.count, 9)
        }
        print("Transcript geometry: 800 rows / 200 viewport queries / \(layout.visibleQueryProbeCount) probes")
        // log2(800) + visible rows + a small boundary allowance, not 800
        // CGRect intersection tests on every frame.
        XCTAssertLessThanOrEqual(layout.visibleQueryProbeCount, 200 * 32)
    }

    func testMeasuringRowsDoesNotAllocateLayoutAttributesForTheWholeWindow() throws {
        let layout = TranscriptLayout()
        let collection = UICollectionView(frame: CGRect(x: 0, y: 0, width: 390, height: 844), collectionViewLayout: layout)
        layout.setItems((0..<800).map(String.init), width: 390)
        layout.prepare()
        let old = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: 300, section: 0)))
        let oldFrame = old.frame
        let allocations = layout.attributeCreationCount
        for index in 0..<20 {
            let original = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: index, section: 0)))
            let preferred = original.copy() as! UICollectionViewLayoutAttributes
            preferred.size.height += 20
            _ = layout.invalidationContext(forPreferredLayoutAttributes: preferred, withOriginalAttributes: original)
            layout.prepare()
        }
        let updated = try XCTUnwrap(layout.layoutAttributesForItem(at: IndexPath(item: 300, section: 0)))
        XCTAssertEqual(updated.frame.minY, oldFrame.minY + 400, accuracy: 0.01)
        XCTAssertEqual(old.frame, oldFrame, "UIKit's previously returned attributes must remain immutable")
        XCTAssertEqual(collection.collectionViewLayout, layout)
        let created = layout.attributeCreationCount - allocations
        print("Transcript geometry: 20 self-sizing updates / \(created) attributes created")
        XCTAssertLessThanOrEqual(created, 40)
    }
    #endif
}