@@ -80,33 +80,30 @@ final class APIStressTests: XCTestCase {
8080 DispatchQueue . concurrentPerform ( iterations: concurrencyLimit) { index in
8181 Task {
8282 let subscription = Amplify . API. subscribe ( request: . subscription( of: Post . self, type: . onCreate) )
83- Task {
84- for try await subscriptionEvent in subscription {
85- switch subscriptionEvent {
86- case . connection( let state) :
87- switch state {
88- case . connecting:
89- break
90- case . connected:
91- connectedInvoked. fulfill ( )
92- case . disconnected:
93- disconnectedInvoked. fulfill ( )
94- }
95- case . data( let result) :
96- switch result {
97- case . success( let post) :
98- if post. title == title {
99- progressInvoked. fulfill ( )
100- }
101- case . failure( let error) :
102- XCTFail ( " \( error) " )
83+ await sequenceActor. append ( sequence: subscription)
84+ for try await subscriptionEvent in subscription {
85+ switch subscriptionEvent {
86+ case . connection( let state) :
87+ switch state {
88+ case . connecting:
89+ break
90+ case . connected:
91+ connectedInvoked. fulfill ( )
92+ case . disconnected:
93+ disconnectedInvoked. fulfill ( )
94+ }
95+ case . data( let result) :
96+ switch result {
97+ case . success( let post) :
98+ if post. title == title {
99+ progressInvoked. fulfill ( )
103100 }
101+ case . failure( let error) :
102+ XCTFail ( " \( error) " )
104103 }
105104 }
106- completedInvoked. fulfill ( )
107105 }
108-
109- await sequenceActor. append ( sequence: subscription)
106+ completedInvoked. fulfill ( )
110107 }
111108 }
112109
0 commit comments