Skip to content

Commit b1fa94b

Browse files
committed
refactor testMultipleSubscriptions()
1 parent 13c4933 commit b1fa94b

File tree

1 file changed

+20
-23
lines changed

1 file changed

+20
-23
lines changed

AmplifyPlugins/API/Tests/APIHostApp/GraphQLAPIStressTests/GraphQLAPIStressTests.swift

Lines changed: 20 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)