AppIntentsTesting Framework

import XCTest
import AppIntentsTesting

final class CalendarIntentTests: XCTestCase {

    // Bundle identifier of the host app under test
    private let appBundleID = "com.example.CometCal"

    private var intentDefs: IntentDefinitions!
    private var entityDefs: EntityDefinitions!

    override func setUp() async throws {
        try await super.setUp()
        intentDefs = try IntentDefinitions(bundleIdentifier: appBundleID)
        entityDefs = try EntityDefinitions(bundleIdentifier: appBundleID)

        // Seed deterministic test data using a test-only intent
        let seedDef = intentDefs.intents["SeedSampleEventsIntent"]
        let seedIntent = try seedDef.makeIntent(parameters: [:])
        try await seedIntent.run()
    }

    // MARK: - Test: Create a calendar and assert return value

    func testCreateCalendarReturnsCorrectTitle() async throws {
        let createDef = intentDefs.intents["CreateCalendarIntent"]

        let intent = try createDef.makeIntent(parameters: [
            "name": "Occupy Saturn",
            "color": "red"   // AppEnum raw string value
        ])

        let result = try await intent.run()

        // Dynamic member lookup surfaces the entity's 'title' property
        let title = try result.value.title as String
        XCTAssertEqual(title, "Occupy Saturn")
    }

    // MARK: - Test: Entity string query returns matching events

    func testEventStringQueryFiltersCorrectly() async throws {
        let eventEntityDef = entityDefs.entities["EventEntity"]

        // Executes EventEntity's EntityStringQuery on-device
        let matches = try await eventEntityDef.entities(matching: "Cosmic Ray Calibration")

        XCTAssertEqual(matches.count, 1)
        let eventTitle = try matches[0].title as String
        XCTAssertEqual(eventTitle, "Cosmic Ray Calibration")
    }

    // MARK: - Test: Chain two intents (create then update)

    func testCreateThenUpdateEventTitle() async throws {
        let createDef = intentDefs.intents["CreateEventIntent"]
        let updateDef = intentDefs.intents["UpdateEventIntent"]

        // Create the event; passing a String for a CalendarEntity parameter
        // triggers the CalendarEntity's EntityStringQuery automatically.
        let createIntent = try createDef.makeIntent(parameters: [
            "title": "Asteroid Dodgeball Practice",
            "date": "2026-09-01T14:00:00Z",
            "duration": 3600,
            "calendar": "Mission Control"  // resolved via EntityStringQuery
        ])
        let createResult = try await createIntent.run()

        // Pass the returned EventEntity directly into the update intent
        let updateIntent = try updateDef.makeIntent(parameters: [
            "event": createResult.value,
            "title": "Asteroid Dodgeball Rules Review"
        ])
        let updateResult = try await updateIntent.run()

        let updatedTitle = try updateResult.value.title as String
        XCTAssertEqual(updatedTitle, "Asteroid Dodgeball Rules Review")
    }
}