fix(api): Web socket decode connection errors (#4606)

aws-amplify · Apr 16, 2024 · 549f355 · 549f355
1 parent cb9f7e2
commit 549f355
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/packages/api/amplify_api_dart/lib/src/graphql/web_socket/types/web_socket_types.dart b/packages/api/amplify_api_dart/lib/src/graphql/web_socket/types/web_socket_types.dart
@@ -61,6 +61,7 @@ abstract class WebSocketMessagePayload {
     MessageType.connectionAck: ConnectionAckMessagePayload.fromJson,
     MessageType.data: SubscriptionDataPayload.fromJson,
     MessageType.error: WebSocketError.fromJson,
+    MessageType.connectionError: WebSocketError.fromJson,
   };
 
   static WebSocketMessagePayload? fromJson(

diff --git a/packages/api/amplify_api_dart/test/web_socket/web_socket_types_test.dart b/packages/api/amplify_api_dart/test/web_socket/web_socket_types_test.dart
@@ -65,6 +65,21 @@ void main() {
       },
       MessageType.error,
     ),
+    MessageTypeTestEntry(
+      {
+        'id': 'abc-456',
+        'type': 'connection_error',
+        'payload': {
+          'errors': [
+            {
+              'errorType': 'UnknownConnectionError',
+              'message': 'Unknown connection id abc-456',
+            }
+          ],
+        },
+      },
+      MessageType.connectionError,
+    ),
   ];
 
   group('WebSocketMessage should create expected messages from JSON', () {