chore(test): add API subscription integration test case for LazyLoading

Author: 5d

AmplifyPlugins/API/Tests/APIHostApp/AWSAPIPluginLazyLoadTests/LL12/DefaultPK/GraphQLLazyLoadDefaultPKTests.swift

@@ -17,86 +17,86 @@ final class GraphQLLazyLoadDefaultPKTests: GraphQLLazyLoadBaseTest {
     func testConfigure() async throws {
         await setup(withModels: DefaultPKModels())
     }
-    
+
     func testDefaultPKParentChild() async throws {
         await setup(withModels: DefaultPKModels())
         let defaultPKParent = DefaultPKParent()
         let savedParent = try await mutate(.create(defaultPKParent))
         let defaultPKChild = DefaultPKChild(parent: savedParent)
         let savedChild = try await mutate(.create(defaultPKChild))
-        
+
         assertLazyReference(savedChild._parent,
                             state: .notLoaded(identifiers: [.init(name: "id", value: savedParent.id)]))
         let loadedParent = try await savedChild.parent
         assertLazyReference(savedChild._parent,
                             state: .loaded(model: loadedParent))
     }
-    
+
     func testDefaultParentChildUpdate() async throws {
         await setup(withModels: DefaultPKModels())
         let defaultPKParent = DefaultPKParent()
         let savedParent = try await mutate(.create(defaultPKParent))
         let defaultPKChild = DefaultPKChild(parent: savedParent)
         var savedChild = try await mutate(.create(defaultPKChild))
-        
+
         let newParent = DefaultPKParent()
         let savedNewParent = try await mutate(.create(newParent))
         savedChild.setParent(savedNewParent)
         var updatedChild = try await mutate(.update(savedChild))
-                
+
         assertLazyReference(updatedChild._parent,
                             state: .notLoaded(identifiers: [.init(name: "id", value: newParent.id)]))
         let loadedParent = try await updatedChild.parent
         assertLazyReference(updatedChild._parent,
                             state: .loaded(model: loadedParent))
     }
-    
+
     func testDefaultParentChildDelete() async throws {
         await setup(withModels: DefaultPKModels())
         let defaultPKParent = DefaultPKParent()
         let savedParent = try await mutate(.create(defaultPKParent))
         let defaultPKChild = DefaultPKChild(parent: savedParent)
         let savedChild = try await mutate(.create(defaultPKChild))
-        
+
         try await mutate(.delete(savedParent))
         try await assertModelDoesNotExist(savedParent)
         try await assertModelExists(savedChild)
         try await mutate(.delete(savedChild))
         try await assertModelDoesNotExist(savedChild)
     }
-    
+
     func testDefaultPKParentChildGet() async throws {
         await setup(withModels: DefaultPKModels())
         let defaultPKParent = DefaultPKParent()
         let savedParent = try await mutate(.create(defaultPKParent))
         let defaultPKChild = DefaultPKChild(parent: savedParent)
         let savedChild = try await mutate(.create(defaultPKChild))
-        
+
         let queriedParent = try await query(.get(DefaultPKParent.self, byId: savedParent.id))!
         assertList(queriedParent.children!, state: .isNotLoaded(associatedIdentifiers: [queriedParent.id],
                                                                 associatedFields: ["parent"]))
         try await queriedParent.children?.fetch()
         assertList(queriedParent.children!, state: .isLoaded(count: 1))
-        
+
         let queriedChild = try await query(.get(DefaultPKChild.self, byId: savedChild.id))!
         assertLazyReference(queriedChild._parent,
                             state: .notLoaded(identifiers: [.init(name: "id", value: savedParent.id)]))
         let loadedParent = try await queriedChild.parent
         assertLazyReference(queriedChild._parent,
                             state: .loaded(model: loadedParent))
     }
-    
+
     func testDefaultPKParentChildList() async throws {
         await setup(withModels: DefaultPKModels())
         let defaultPKParent = DefaultPKParent()
         let savedParent = try await mutate(.create(defaultPKParent))
         let defaultPKChild = DefaultPKChild(parent: savedParent)
         let savedChild = try await mutate(.create(defaultPKChild))
-        
+
         let queriedParents = try await listQuery(.list(DefaultPKParent.self,
                                                        where: DefaultPKParent.keys.id == defaultPKParent.id))
         assertList(queriedParents, state: .isLoaded(count: 1))
-        
+
         let queriedChildren = try await listQuery(.list(DefaultPKChild.self,
                                                         where: DefaultPKChild.keys.id == defaultPKChild.id))
         assertList(queriedChildren, state: .isLoaded(count: 1))
@@ -269,7 +269,7 @@ final class GraphQLLazyLoadDefaultPKTests: GraphQLLazyLoadBaseTest {
 extension GraphQLLazyLoadDefaultPKTests: DefaultLogger { }
 
 extension GraphQLLazyLoadDefaultPKTests {
-    
+
     struct DefaultPKModels: AmplifyModelRegistration {
         public let version: String = "version"
         func registerModels(registry: ModelRegistry.Type) {

AmplifyPlugins/API/Tests/APIHostApp/AWSAPIPluginLazyLoadTests/LL12/HasOneParentChild/GraphQLLazyLoadHasOneTests.swift

@@ -17,82 +17,82 @@ final class GraphQLLazyLoadHasOneTests: GraphQLLazyLoadBaseTest {
     func testConfigure() async throws {
         await setup(withModels: HasOneParentChildModels())
     }
-    
+
     func testHasOneParentChild() async throws {
         await setup(withModels: HasOneParentChildModels())
         let hasOneChild = HasOneChild()
         let savedChild = try await mutate(.create(hasOneChild))
         let hasOneParent = HasOneParent(child: hasOneChild)
         let savedParent = try await mutate(.create(hasOneParent))
-        
+
         assertLazyReference(savedParent._child,
                             state: .notLoaded(identifiers: [.init(name: "id", value: savedChild.id)]))
         let loadedChild = try await savedParent.child
         assertLazyReference(savedParent._child,
                             state: .loaded(model: loadedChild))
     }
-    
+
     func testHasOneParentChildUpdate() async throws {
         await setup(withModels: HasOneParentChildModels())
         let hasOneChild = HasOneChild()
         let savedChild = try await mutate(.create(hasOneChild))
         let hasOneParent = HasOneParent(child: hasOneChild)
         var savedParent = try await mutate(.create(hasOneParent))
-        
+
         let newChild = HasOneChild()
         let savedNewChild = try await mutate(.create(newChild))
         savedParent.setChild(newChild)
         var updatedParent = try await mutate(.update(savedParent))
-        
+
         assertLazyReference(updatedParent._child,
                             state: .notLoaded(identifiers: [.init(name: "id", value: newChild.id)]))
         let loadedChild = try await updatedParent.child
         assertLazyReference(updatedParent._child,
                             state: .loaded(model: loadedChild))
     }
-    
+
     func testHasOneParentChildDelete() async throws {
         await setup(withModels: HasOneParentChildModels())
         let hasOneChild = HasOneChild()
         let savedChild = try await mutate(.create(hasOneChild))
         let hasOneParent = HasOneParent(child: hasOneChild)
         var savedParent = try await mutate(.create(hasOneParent))
-        
+
         try await mutate(.delete(savedParent))
         try await assertModelDoesNotExist(savedParent)
         try await assertModelExists(savedChild)
         try await mutate(.delete(savedChild))
         try await assertModelDoesNotExist(savedChild)
     }
-    
+
     func testHasOneParentChildGet() async throws {
         await setup(withModels: HasOneParentChildModels())
         let hasOneChild = HasOneChild()
         let savedChild = try await mutate(.create(hasOneChild))
         let hasOneParent = HasOneParent(child: hasOneChild)
         let savedParent = try await mutate(.create(hasOneParent))
-        
+
         let queriedParent = try await query(.get(HasOneParent.self, byId: savedParent.id))!
         assertLazyReference(queriedParent._child,
                             state: .notLoaded(identifiers: [.init(name: "id", value: savedChild.id)]))
         let loadedChild = try await queriedParent.child
         assertLazyReference(queriedParent._child,
                             state: .loaded(model: loadedChild))
-        
+
         let queriedChild = try await query(.get(HasOneChild.self, byId: savedChild.id))!
     }
-    
+
     func testHasOneParentChildList() async throws {
         await setup(withModels: HasOneParentChildModels())
         let hasOneChild = HasOneChild()
         let savedChild = try await mutate(.create(hasOneChild))
         let hasOneParent = HasOneParent(child: hasOneChild)
         let savedParent = try await mutate(.create(hasOneParent))
-        
+
         let queriedParents = try await listQuery(.list(HasOneParent.self,
                                                        where: HasOneParent.keys.id == hasOneParent.id))
         assertList(queriedParents, state: .isLoaded(count: 1))
-        
+
         let queriedChildren = try await listQuery(.list(HasOneChild.self,
                                                        where: HasOneChild.keys.id == hasOneChild.id))
         assertList(queriedChildren, state: .isLoaded(count: 1))
@@ -256,7 +256,7 @@ final class GraphQLLazyLoadHasOneTests: GraphQLLazyLoadBaseTest {
 extension GraphQLLazyLoadHasOneTests: DefaultLogger { }
 
 extension GraphQLLazyLoadHasOneTests {
-    
+
     struct HasOneParentChildModels: AmplifyModelRegistration {
         public let version: String = "version"
         func registerModels(registry: ModelRegistry.Type) {