Ensure WebSocket connection error notifies client

Author: tylerjroach

appsync/aws-sdk-appsync-events/src/androidTest/java/com/amazonaws/sdk/appsync/events/EventsWebSocketClientTests.kt

@@ -18,6 +18,7 @@ import androidx.test.platform.app.InstrumentationRegistry
 import app.cash.turbine.test
 import app.cash.turbine.turbineScope
 import com.amazonaws.sdk.appsync.core.authorizers.ApiKeyAuthorizer
+import com.amazonaws.sdk.appsync.events.data.ConnectException
 import com.amazonaws.sdk.appsync.events.data.InvalidInputException
 import com.amazonaws.sdk.appsync.events.data.PublishResult
 import com.amazonaws.sdk.appsync.events.data.UnauthorizedException
@@ -47,6 +48,7 @@ import org.junit.Test
 internal class EventsWebSocketClientTests {
     private val eventsConfig = getEventsConfig(InstrumentationRegistry.getInstrumentation().targetContext)
     private val apiKeyAuthorizer = ApiKeyAuthorizer(eventsConfig.apiKey)
+    private val badApiKeyAuthorizer = ApiKeyAuthorizer("bad-api-key")
     private val webSocketLogCapture = EventsLibraryLogCapture()
     private val defaultChannel = "default/${UUID.randomUUID()}"
     private val customChannel = "custom/${UUID.randomUUID()}"
@@ -60,6 +62,12 @@ internal class EventsWebSocketClientTests {
             loggerProvider = { _ -> webSocketLogCapture }
         )
     )
+
+    private val badWebSocketClient = events.createWebSocketClient(
+        badApiKeyAuthorizer,
+        badApiKeyAuthorizer,
+        badApiKeyAuthorizer
+    )
     private val backgroundScope = CoroutineScope(Dispatchers.IO)
 
     @After
@@ -109,6 +117,15 @@ internal class EventsWebSocketClientTests {
         )
     }
 
+    @Test
+    fun testConnectionFailure() = runBlockingWithTimeout {
+        turbineScope(timeout = 10.seconds) {
+            badWebSocketClient.subscribe(defaultChannel).test(timeout = 10.seconds) {
+                awaitError() shouldBe ConnectException(UnauthorizedException())
+            }
+        }
+    }
+
     @Test
     fun testPublishWithBadAuth(): Unit = runBlockingWithTimeout {
         // Publish the message

appsync/aws-sdk-appsync-events/src/main/java/com/amazonaws/sdk/appsync/events/data/WebSocketMessage.kt

@@ -97,7 +97,10 @@ internal sealed class WebSocketMessage {
 
         @Serializable
         @SerialName("connection_error")
-        internal data class ConnectionError(val errors: List<EventsError>) : Received()
+        internal data class ConnectionError(
+            override val id: String? = null,
+            override val errors: List<EventsError>
+        ) : Received(), ErrorContainer
 
         @Serializable
         internal sealed class Subscription : Received() {
@@ -189,7 +192,8 @@ internal sealed class WebSocketMessage {
 
     internal data class Closed(val reason: WebSocketDisconnectReason) : WebSocketMessage()
 
-    // All errors contain an id and errors list
+    // All errors contain an errors list
+    // Most events contain an id, which is either a channel id or a publish id
     internal interface ErrorContainer {
         val id: String?
         val errors: List<EventsError>