Fix tests

RomanPodymov · Dec 13, 2023 · 1f1d204 · 1f1d204
1 parent 0dde595
commit 1f1d204
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/Sources/CommonAppleKit/CAListViewCell.swift b/Sources/CommonAppleKit/CAListViewCell.swift
@@ -17,8 +17,9 @@ open class CAListViewCell<RootView: CAView>: CACollectionViewCell {
 
     #if canImport(AppKit)
         open override func loadView() {
-            view = createRootView()
-            rootView = view
+            let rootView = createRootView()
+            view = rootView
+            self.rootView = rootView
         }
     #elseif canImport(UIKit)
         open var representedObject: Any?

diff --git a/Tests/CommonAppleKitTests/CommonAppleKitTests.swift b/Tests/CommonAppleKitTests/CommonAppleKitTests.swift
@@ -13,7 +13,9 @@ private final class ControllerA: CAViewController { }
 
 private final class ControllerB: CAViewController { }
 
-private final class CellA: CAListViewCell { }
+private final class CellA: CAListViewCell<CellARootView> { }
+
+private final class CellARootView: CAView { }
 
 private final class ViewA: CAView {
     override init(frame: CARect) {
@@ -82,12 +84,12 @@ final class CommonAppleKitTests: XCTestCase {
     }
 
     func testListView() {
-        let listView = CAListView<CellA>(frame: .zero, itemSize: .init(width: 200, height: 100), cellId: "id")
+        let listView = CAListView<CellA, CellARootView>(frame: .zero, itemSize: .init(width: 200, height: 100), cellId: "id")
         listView.content = []
     }
 
     func testScrollableListView() {
-        let listView = CAScrollableListView<CellA>(frame: .zero, itemSize: .init(width: 200, height: 100), cellId: "id")
+        let listView = CAScrollableListView<CellA, CellARootView>(frame: .zero, itemSize: .init(width: 200, height: 100), cellId: "id")
         listView.content = []
     }
 }