fix tests

Author: sebsto

Sources/AWSLambdaRuntime/LambdaStreaming+Codable.swift

@@ -82,33 +82,52 @@ public struct StreamingLambdaCodableAdapter<
         context: LambdaContext
     ) async throws {
 
-        // for some reasons I don't understand the "body" param contains the complete FunctionURL request
-        // so, 1/ we decode the event we receive, 2/ we base64 decode the body, 3/ we decode a FunctionURLRequest again,
-        // then 4/ we can access the actual payload body, decode it pass it to the handler
-        let functionUrlEvent1 = try self.decoder.decode(FunctionURLRequest.self, from: event)
-
-        if let base64EncodedString = functionUrlEvent1.body,
-            // this is the minimal way to base64 decode without importing new dependencies
-            let decodedData = Data(base64Encoded: base64EncodedString),
-            let decodedString = String(data: decodedData, encoding: .utf8)
-        {
-
-            // decode the FunctionURL event inside the body
-            let functionUrlEvent2 = try self.decoder.decode(
-                FunctionURLRequest.self,
-                from: ByteBuffer(string: decodedString)
-            )
-
-            // finally decode the actual payload passed by the caller
-            let decodedEvent = try self.decoder.decode(
-                Handler.Event.self,
-                from: ByteBuffer(string: functionUrlEvent2.body ?? "")
-            )
-
-            // and invoke the user-provided handler
-            try await self.handler.handle(decodedEvent, responseWriter: responseWriter, context: context)
+        // try to decode the event as a FunctionURLRequest
+        if let functionUrlEvent1 = isFunctionURLRequest(event) {
+
+            // for some reasons I don't understand the "body" param contains the complete FunctionURL request
+            // so, 1/ we decode the event we receive, 2/ we base64 decode the body, 3/ we decode a FunctionURLRequest again,
+            // then 4/ we can access the actual payload body, decode it pass it to the handler
+            if let base64EncodedString = functionUrlEvent1.body,
+                // this is the minimal way to base64 decode without importing new dependencies
+                let decodedData = Data(base64Encoded: base64EncodedString),
+                let decodedString = String(data: decodedData, encoding: .utf8)
+            {
+
+                // decode the FunctionURL event inside the body
+                let functionUrlEvent2 = try self.decoder.decode(
+                    FunctionURLRequest.self,
+                    from: ByteBuffer(string: decodedString)
+                )
+
+                // finally decode the actual payload passed by the caller
+                let decodedEvent = try self.decoder.decode(
+                    Handler.Event.self,
+                    from: ByteBuffer(string: functionUrlEvent2.body ?? "")
+                )
+
+                // and invoke the user-provided handler
+                try await self.handler.handle(decodedEvent, responseWriter: responseWriter, context: context)
+            } else {
+                context.logger.trace("Can't decode FunctionURLRequest's body", metadata: ["Event": "\(event)"])
+            }
+            
         } else {
-            context.logger.trace("Can't decode FunctionURLRequest's body", metadata: ["Event": "\(event)"])
+            // otherwise, decode the event as a user-provided JSON event
+            let decodedEvent = try self.decoder.decode(Handler.Event.self, from: event)
+            try await self.handler.handle(decodedEvent, responseWriter: responseWriter, context: context)
+        }
+    }
+
+    /// Check if the payload is an FunctionURLlRequest or a direct invocation
+    /// - Parameter event: The raw ByteBuffer event to check.
+    /// - Returns: the FunctionURLRequest if the event is a FunctionURLRequest, nil otherwise
+    @inlinable
+    package func isFunctionURLRequest(_ event: ByteBuffer) -> FunctionURLRequest? {
+        do {
+            return try self.decoder.decode(FunctionURLRequest.self, from: event)
+        } catch {
+            return nil
         }
     }
 }