Add tests

Author: ra1028

Tests/AnyComponentTests.swift

@@ -66,6 +66,31 @@ final class AnyComponentTests: XCTestCase {
         XCTAssertEqual((anyContent as? MockComponent.Content)?.frame, frame)
     }
 
+    func testIntrinsicContentSizeForView() {
+        struct TestComponent: Component {
+            func renderContent() -> UILabel {
+                UILabel()
+            }
+
+            func render(in content: UILabel) {
+                content.text = "Test"
+            }
+        }
+
+        let component = TestComponent()
+        let content = component.renderContent()
+        let anyComponent = AnyComponent(component)
+        let anyContent = anyComponent.renderContent()
+
+        component.render(in: content)
+        anyComponent.render(in: anyContent)
+
+        XCTAssertEqual(
+            component.intrinsicContentSize(for: content),
+            anyComponent.intrinsicContentSize(for: anyContent)
+        )
+    }
+
     func testShouldContentUpdate() {
         let component1 = MockComponent(shouldContentUpdate: true)
         let component2 = MockComponent(shouldContentUpdate: false)

Tests/ComponentTests.swift

@@ -22,6 +22,24 @@ final class ComponentTests: XCTestCase {
         XCTAssertEqual(content.frame, frame)
     }
 
+    func testIntrinsicContentSizeForView() {
+        struct TestComponent: Component {
+            func renderContent() -> UILabel {
+                UILabel()
+            }
+
+            func render(in content: UILabel) {
+                content.text = "Test"
+            }
+        }
+
+        let component = TestComponent()
+        let content = component.renderContent()
+        component.render(in: content)
+
+        XCTAssertEqual(component.intrinsicContentSize(for: content), content.intrinsicContentSize)
+    }
+
     func testShouldContentUpdateWhenEquatable() {
         let component1 = A.Component(value: 100)
         let component2 = A.Component(value: 200)

Tests/ComponentWrapper/ComponentWrappingTests.swift

@@ -4,14 +4,16 @@ import XCTest
 final class ComponentWrappingTests: XCTestCase {
     func testForwardingActions() {
         let reuseIdentifier = "testForwardingActions"
-        let referenceSize =  CGSize(width: 200, height: 200)
+        let referenceSize = CGSize(width: 200, height: 200)
+        let intrinsicContentSize = CGSize(width: 300, height: 300)
         let shouldContentUpdate = true
         let shouldRender = true
         let content = UIView()
 
         let mock = MockComponent(
             reuseIdentifier: reuseIdentifier,
             referenceSize: referenceSize,
+            intrinsicContentSize: intrinsicContentSize,
             shouldContentUpdate: shouldContentUpdate,
             shouldRender: shouldRender,
             content: content
@@ -26,6 +28,7 @@ final class ComponentWrappingTests: XCTestCase {
         XCTAssertEqual(wrapper.renderContent(), content)
         XCTAssertEqual(mock.contentCapturedOnRender, content)
         XCTAssertEqual(wrapper.referenceSize(in: .zero), referenceSize)
+        XCTAssertEqual(wrapper.intrinsicContentSize(for: content), intrinsicContentSize)
         XCTAssertEqual(wrapper.shouldContentUpdate(with: wrapper), shouldContentUpdate)
         XCTAssertEqual(wrapper.shouldRender(next: wrapper, in: content), shouldRender)
         XCTAssertEqual(mock.contentCapturedOnLayout, content)

Tests/SwiftUISupport/ComponentSwiftUISupportTests.swift

@@ -74,6 +74,27 @@ final class ComponentSwiftUISupportTests: XCTestCase {
 
         XCTAssertEqual(hostingController.view.sizeThatFits(.zero), TestComponent.testSize)
     }
+
+    func testIntrinsicContentSize() {
+        struct TestComponent: Component, View {
+            static let testSize = CGSize(width: 123, height: 456)
+
+            func renderContent() -> UIView {
+                UIView()
+            }
+
+            func render(in content: UIView) {}
+
+            func intrinsicContentSize(for content: UIView) -> CGSize {
+                Self.testSize
+            }
+        }
+
+        let component = TestComponent()
+        let hostingController = UIHostingController(rootView: component.body)
+
+        XCTAssertEqual(hostingController.view.sizeThatFits(.zero), TestComponent.testSize)
+    }
 }
 
 #endif

Tests/TestTools.swift

@@ -56,6 +56,7 @@ enum B {
 class MockComponent: Component, Equatable {
     let reuseIdentifier: String
     let referenceSize: CGSize?
+    let intrinsicContentSize: CGSize
     let shouldContentUpdate: Bool
     let shouldRender: Bool
     let content: UIView
@@ -68,12 +69,14 @@ class MockComponent: Component, Equatable {
     init(
         reuseIdentifier: String = "MockComponent",
         referenceSize: CGSize? = nil,
+        intrinsicContentSize: CGSize = .zero,
         shouldContentUpdate: Bool = false,
         shouldRender: Bool = false,
         content: UIView = UIView()
         ) {
         self.reuseIdentifier = reuseIdentifier
         self.referenceSize = referenceSize
+        self.intrinsicContentSize = intrinsicContentSize
         self.shouldContentUpdate = shouldContentUpdate
         self.shouldRender = shouldRender
         self.content = content
@@ -103,6 +106,10 @@ class MockComponent: Component, Equatable {
         contentCapturedOnLayout = content
     }
 
+    func intrinsicContentSize(for content: UIView) -> CGSize {
+        intrinsicContentSize
+    }
+
     func contentWillDisplay(_ content: UIView) {
         contentCapturedOnWillDisplay = content
     }