Initial commit

2019-06-24 12:03:35 +09:00
commit 3e216e1aa4
98 changed files with 12591 additions and 0 deletions
@@ -0,0 +1,186 @@
+#if os(iOS) || os(tvOS)
+
+import XCTest
+import UIKit
+@testable import DiffableDataSources
+
+final class TableViewDiffableDataSourceTests: XCTestCase {
+    func testInit() {
+        let tableView = MockTableView()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            UITableViewCell()
+        }
+
+        XCTAssertTrue(tableView.dataSource === dataSource)
+    }
+
+    func testApply() {
+        let tableView = MockTableView()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            UITableViewCell()
+        }
+
+        let snapshot = DiffableDataSourceSnapshot<Int, Int>()
+
+        dataSource.apply(snapshot)
+        XCTAssertEqual(tableView.isPerformBatchUpdatesCalledCount, 0)
+
+        snapshot.appendSections([0])
+        snapshot.appendItems([0])
+
+        dataSource.apply(snapshot)
+        XCTAssertEqual(tableView.isPerformBatchUpdatesCalledCount, 1)
+
+        dataSource.apply(snapshot)
+        XCTAssertEqual(tableView.isPerformBatchUpdatesCalledCount, 1)
+
+        snapshot.appendItems([1])
+
+        dataSource.apply(snapshot)
+        XCTAssertEqual(tableView.isPerformBatchUpdatesCalledCount, 2)
+    }
+
+    func testSnapshot() {
+        let tableView = MockTableView()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            UITableViewCell()
+        }
+
+        let snapshot1 = dataSource.snapshot()
+        XCTAssertEqual(snapshot1.sectionIdentifiers, [])
+        XCTAssertEqual(snapshot1.itemIdentifiers, [])
+
+        dataSource.snapshot().appendSections([0, 1, 2])
+        let snapshot2 = dataSource.snapshot()
+        XCTAssertEqual(snapshot2.sectionIdentifiers, [])
+        XCTAssertEqual(snapshot2.itemIdentifiers, [])
+
+        let snapshotToApply = DiffableDataSourceSnapshot<Int, Int>()
+        snapshotToApply.appendSections([0, 1, 2])
+        snapshotToApply.appendItems([0, 1, 2])
+        dataSource.apply(snapshotToApply)
+
+        let snapshot3 = dataSource.snapshot()
+        XCTAssertEqual(snapshot3.sectionIdentifiers, [0, 1, 2])
+        XCTAssertEqual(snapshot3.itemIdentifiers, [0, 1, 2])
+
+        dataSource.snapshot().appendSections([3, 4, 5])
+
+        let snapshot4 = dataSource.snapshot()
+        XCTAssertEqual(snapshot4.sectionIdentifiers, [0, 1, 2])
+        XCTAssertEqual(snapshot4.itemIdentifiers, [0, 1, 2])
+
+        snapshot4.appendSections([3, 4, 5])
+        snapshot4.appendItems([3, 4, 5])
+        dataSource.apply(snapshot4)
+
+        let snapshot5 = dataSource.snapshot()
+        XCTAssertEqual(snapshot5.sectionIdentifiers, [0, 1, 2, 3, 4, 5])
+        XCTAssertEqual(snapshot5.itemIdentifiers, [0, 1, 2, 3, 4, 5])
+    }
+
+    func testItemIdentifier() {
+        let tableView = MockTableView()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            UITableViewCell()
+        }
+
+        let snapshot = DiffableDataSourceSnapshot<Int, Int>()
+        snapshot.appendSections([0, 1, 2])
+        snapshot.appendItems([0, 1, 2], toSection: 0)
+        dataSource.apply(snapshot)
+
+        XCTAssertEqual(dataSource.itemIdentifier(for: IndexPath(item: 1, section: 0)), 1)
+        XCTAssertEqual(dataSource.itemIdentifier(for: IndexPath(item: 100, section: 100)), nil)
+    }
+
+    func testIndexPath() {
+        let tableView = MockTableView()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            UITableViewCell()
+        }
+
+        let snapshot = DiffableDataSourceSnapshot<Int, Int>()
+        snapshot.appendSections([0, 1, 2])
+        snapshot.appendItems([0, 1, 2], toSection: 0)
+        dataSource.apply(snapshot)
+
+        XCTAssertEqual(dataSource.indexPath(for: 2), IndexPath(item: 2, section: 0))
+        XCTAssertEqual(dataSource.indexPath(for: 100), nil)
+    }
+
+    func testNumberOfSections() {
+        let tableView = MockTableView()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            UITableViewCell()
+        }
+
+        XCTAssertEqual(dataSource.numberOfSections(in: tableView), 0)
+
+        let snapshot = DiffableDataSourceSnapshot<Int, Int>()
+        snapshot.appendSections([0, 1, 2])
+        snapshot.appendItems([0, 1, 2], toSection: 0)
+        dataSource.apply(snapshot)
+
+        XCTAssertEqual(dataSource.numberOfSections(in: tableView), 3)
+    }
+
+    func testNumberOfRowsInSection() {
+        let tableView = MockTableView()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            UITableViewCell()
+        }
+
+        let snapshot = DiffableDataSourceSnapshot<Int, Int>()
+        snapshot.appendSections([0, 1, 2])
+        snapshot.appendItems([0, 1, 2], toSection: 0)
+        dataSource.apply(snapshot)
+
+        XCTAssertEqual(dataSource.tableView(tableView, numberOfRowsInSection: 0), 3)
+    }
+
+    func testCellForRowAt() {
+        let tableView = MockTableView()
+        let cell = UITableViewCell()
+        let dataSource = TableViewDiffableDataSource<Int, Int>(tableView: tableView) { _, _, _ in
+            cell
+        }
+
+        let snapshot = DiffableDataSourceSnapshot<Int, Int>()
+        snapshot.appendSections([0, 1, 2])
+        snapshot.appendItems([0, 1, 2], toSection: 0)
+        dataSource.apply(snapshot)
+
+        XCTAssertEqual(
+            dataSource.tableView(tableView, cellForRowAt: IndexPath(item: 1, section: 0)),
+            cell
+        )
+    }
+}
+
+final class MockTableView: UITableView {
+    var isPerformBatchUpdatesCalledCount = 0
+
+    init() {
+        super.init(frame: .zero, style: .plain)
+
+        let window = UIWindow()
+        window.addSubview(self)
+    }
+
+    @available(*, unavailable)
+    required init?(coder aDecoder: NSCoder) {
+        fatalError("init(coder:) has not been implemented")
+    }
+
+    override func performBatchUpdates(_ updates: (() -> Void)?, completion: ((Bool) -> Void)? = nil) {
+        isPerformBatchUpdatesCalledCount += 1
+        updates?()
+        completion?(true)
+    }
+
+    override func insertSections(_ sections: IndexSet, with animation: RowAnimation) {}
+    override func insertRows(at indexPaths: [IndexPath], with animation: RowAnimation) {}
+}
+
+#endif