WIP

Author: davdroman

Tests/Tests/ViewTypes/PickerWithWheelStyleTests.swift

@@ -1,27 +1,24 @@
 #if !os(tvOS) && !os(macOS)
 import SwiftUI
 import SwiftUIIntrospect
-import XCTest
+import Testing
 
 @MainActor
-final class PickerWithWheelStyleTests: XCTestCase {
+@Suite
+struct PickerWithWheelStyleTests {
     #if canImport(UIKit)
     typealias PlatformPickerWithWheelStyle = UIPickerView
     #endif
 
-    func testPickerWithWheelStyle() {
-        XCTAssertViewIntrospection(of: PlatformPickerWithWheelStyle.self) { spies in
-            let spy0 = spies[0]
-            let spy1 = spies[1]
-            let spy2 = spies[2]
-
+    @Test func introspect() async throws {
+        let (entity1, entity2, entity3) = try await introspection(of: PlatformPickerWithWheelStyle.self) { spy1, spy2, spy3 in
             VStack {
                 Picker("Pick", selection: .constant("1")) {
                     Text("1").tag("1")
                 }
                 .pickerStyle(.wheel)
                 #if os(iOS) || os(visionOS)
-                .introspect(.picker(style: .wheel), on: .iOS(.v13, .v14, .v15, .v16, .v17, .v18, .v26), .visionOS(.v1, .v2, .v26), customize: spy0)
+                .introspect(.picker(style: .wheel), on: .iOS(.v13, .v14, .v15, .v16, .v17, .v18, .v26), .visionOS(.v1, .v2, .v26), customize: spy1)
                 #endif
                 .cornerRadius(8)
 
@@ -31,7 +28,7 @@ final class PickerWithWheelStyleTests: XCTestCase {
                 }
                 .pickerStyle(.wheel)
                 #if os(iOS) || os(visionOS)
-                .introspect(.picker(style: .wheel), on: .iOS(.v13, .v14, .v15, .v16, .v17, .v18, .v26), .visionOS(.v1, .v2, .v26), customize: spy1)
+                .introspect(.picker(style: .wheel), on: .iOS(.v13, .v14, .v15, .v16, .v17, .v18, .v26), .visionOS(.v1, .v2, .v26), customize: spy2)
                 #endif
                 .cornerRadius(8)
 
@@ -42,16 +39,15 @@ final class PickerWithWheelStyleTests: XCTestCase {
                 }
                 .pickerStyle(.wheel)
                 #if os(iOS) || os(visionOS)
-                .introspect(.picker(style: .wheel), on: .iOS(.v13, .v14, .v15, .v16, .v17, .v18, .v26), .visionOS(.v1, .v2, .v26), customize: spy2)
+                .introspect(.picker(style: .wheel), on: .iOS(.v13, .v14, .v15, .v16, .v17, .v18, .v26), .visionOS(.v1, .v2, .v26), customize: spy3)
                 #endif
             }
-        } extraAssertions: {
-            #if canImport(UIKit)
-            XCTAssertEqual($0[safe: 0]?.numberOfRows(inComponent: 0), 1)
-            XCTAssertEqual($0[safe: 1]?.numberOfRows(inComponent: 0), 2)
-            XCTAssertEqual($0[safe: 2]?.numberOfRows(inComponent: 0), 3)
-            #endif
         }
+        #if canImport(UIKit)
+        #expect(entity1.numberOfRows(inComponent: 0) == 1)
+        #expect(entity2.numberOfRows(inComponent: 0) == 2)
+        #expect(entity3.numberOfRows(inComponent: 0) == 3)
+        #endif
     }
 }
 #endif