Add migration test

Author: itsmeichigo

Modules/Sources/Storage/Model/MIGRATIONS.md

@@ -2,6 +2,10 @@
 
 This file documents changes in the WCiOS Storage data model. Please explain any changes to the data model as well as any custom migrations.
 
+## Model 127 (Release 23.4.0.0)
+- @itsmeichigo 2025-09-23
+  - Added new `Booking` entity.
+
 ## Model 126 (Release 23.3.0.0)
 - @rafaelkayumov 2025-09-15
   - Added `isGarden` attribute to `Site` entity.

Modules/Tests/StorageTests/CoreData/MigrationTests.swift

@@ -2755,6 +2755,32 @@ final class MigrationTests: XCTestCase {
         let updatedCategory = migratedLabel.value(forKey: "hazmatCategory") as? String
         XCTAssertEqual(updatedCategory, category)
     }
+
+    func test_migrating_126_to_127_adds_new_booking_entity() throws {
+        // Given
+        let sourceContainer = try startPersistentContainer("Model 126")
+        let sourceContext = sourceContainer.viewContext
+
+        try sourceContext.save()
+
+        // Confidence Check. Booking should not exist in Model 126
+        XCTAssertNil(NSEntityDescription.entity(forEntityName: "Booking", in: sourceContext))
+
+        // When
+        let targetContainer = try migrate(sourceContainer, to: "Model 127")
+        let targetContext = targetContainer.viewContext
+
+        // Then
+        XCTAssertEqual(try targetContext.count(entityName: "Booking"), 0)
+
+        let booking = insertBooking(to: targetContext)
+        XCTAssertNoThrow(try targetContext.save())
+
+        XCTAssertEqual(try targetContext.count(entityName: "Booking"), 1)
+        let insertedBooking = try XCTUnwrap(targetContext.firstObject(ofType: Booking.self))
+        XCTAssertEqual(insertedBooking, booking)
+        XCTAssertEqual(insertedBooking.parentID, 0) // default value
+    }
 }
 
 // MARK: - Persistent Store Setup and Migrations
@@ -3679,4 +3705,12 @@ private extension MigrationTests {
             "id": "test-address"
         ])
     }
+
+    @discardableResult
+    func insertBooking(to context: NSManagedObjectContext) -> NSManagedObject {
+        context.insert(entityName: "Booking", properties: [
+            "siteID": 1,
+            "bookingID": 23
+        ])
+    }
 }