add swiftUI code

2025-01-22 14:09:10 +08:00
parent 68e7b7347c
commit 8a99853829
2531 changed files with 486215 additions and 0 deletions
@@ -0,0 +1,103 @@
+import XCTest
+import GRDB
+
+class AssociationHasManyRowScopeTests: GRDBTestCase {
+
+    func testSingularTable() throws {
+        struct Child : TableRecord {
+        }
+        
+        struct Parent : TableRecord {
+            static let children = hasMany(Child.self)
+        }
+        
+        let dbQueue = try makeDatabaseQueue()
+        try dbQueue.inDatabase { db in
+            try db.create(table: "parent") { t in
+                t.autoIncrementedPrimaryKey("id")
+            }
+            try db.create(table: "child") { t in
+                t.autoIncrementedPrimaryKey("id")
+                t.belongsTo("parent")
+            }
+            try db.execute(sql: """
+                INSERT INTO parent (id) VALUES (1);
+                INSERT INTO child (id, parentId) VALUES (2, 1);
+                """)
+            
+            let request = Parent.including(required: Parent.children)
+            let row = try Row.fetchOne(db, request)!
+            XCTAssertEqual(row.unscoped, ["id": 1])
+            XCTAssertEqual(row.scopes["child"], ["id": 2, "parentId": 1])
+        }
+    }
+    
+    func testPluralTable() throws {
+        struct Child : TableRecord {
+            static let databaseTableName = "children"
+        }
+        
+        struct Parent : TableRecord {
+            static let databaseTableName = "parents"
+            static let children = hasMany(Child.self)
+        }
+        
+        let dbQueue = try makeDatabaseQueue()
+        try dbQueue.inDatabase { db in
+            try db.create(table: "parents") { t in
+                t.autoIncrementedPrimaryKey("id")
+            }
+            try db.create(table: "children") { t in
+                t.autoIncrementedPrimaryKey("id")
+                t.belongsTo("parent")
+            }
+            try db.execute(sql: """
+                INSERT INTO parents (id) VALUES (1);
+                INSERT INTO children (id, parentId) VALUES (2, 1);
+                """)
+            
+            let request = Parent.including(required: Parent.children)
+            let row = try Row.fetchOne(db, request)!
+            XCTAssertEqual(row.unscoped, ["id": 1])
+            XCTAssertEqual(row.scopes["child"], ["id": 2, "parentId": 1])
+        }
+    }
+    
+    func testCustomKey() throws {
+        struct Child : TableRecord {
+        }
+        
+        struct Parent : TableRecord {
+            static let littlePuppies = hasMany(Child.self, key: "littlePuppies")
+            static let kittens = hasMany(Child.self).forKey("kittens")
+        }
+        
+        let dbQueue = try makeDatabaseQueue()
+        try dbQueue.inDatabase { db in
+            try db.create(table: "parent") { t in
+                t.autoIncrementedPrimaryKey("id")
+            }
+            try db.create(table: "child") { t in
+                t.autoIncrementedPrimaryKey("id")
+                t.belongsTo("parent")
+            }
+            try db.execute(sql: """
+                INSERT INTO parent (id) VALUES (1);
+                INSERT INTO child (id, parentId) VALUES (2, 1);
+                """)
+            
+            do {
+                let request = Parent.including(required: Parent.littlePuppies)
+                let row = try Row.fetchOne(db, request)!
+                XCTAssertEqual(row.unscoped, ["id": 1])
+                XCTAssertEqual(row.scopes["littlePuppy"], ["id": 2, "parentId": 1])
+            }
+            do {
+                let request = Parent.including(required: Parent.kittens)
+                let row = try Row.fetchOne(db, request)!
+                XCTAssertEqual(row.unscoped, ["id": 1])
+                XCTAssertEqual(row.scopes["kitten"], ["id": 2, "parentId": 1])
+            }
+        }
+    }
+}