A bit more async-y

BrighterCommand · Feb 8, 2024 · 4dfbf2c · 4dfbf2c
1 parent 018207a
commit 4dfbf2c
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 6 deletions.
diff --git a/tests/Paramore.Brighter.AWS.Tests/TestDoubles/MyLargeCommandMessageMapper.cs b/tests/Paramore.Brighter.AWS.Tests/TestDoubles/MyLargeCommandMessageMapper.cs
@@ -1,4 +1,5 @@
 using System;
+using System.IO;
 using System.Text.Json;
 using System.Threading.Tasks;
 using Paramore.Brighter.Transforms.Attributes;
@@ -26,17 +27,19 @@ public MyLargeCommand MapToRequest(Message message)
 public class MyLargeCommandMessageMapperAsync : IAmAMessageMapperAsync<MyLargeCommand>
 {
     [ClaimCheck(0, thresholdInKb: 5)]
-    public Task<Message> MapToMessage(MyLargeCommand request)
+    public async Task<Message> MapToMessage(MyLargeCommand request)
     {
-        return Task.FromResult(new Message(
+        using var memoryContentStream = new MemoryStream();
+        await JsonSerializer.SerializeAsync(memoryContentStream, request, new JsonSerializerOptions(JsonSerializerDefaults.General));
+        return new Message(
             new MessageHeader(request.Id, "transform.event", MessageType.MT_COMMAND, DateTime.UtcNow),
-            new MessageBody(JsonSerializer.Serialize(request, new JsonSerializerOptions(JsonSerializerDefaults.General))))
-        );
+            new MessageBody(memoryContentStream.ToArray()));
     }
 
     [RetrieveClaim(0, retain:false)]
-    public Task<MyLargeCommand> MapToRequest(Message message)
+    public async Task<MyLargeCommand> MapToRequest(Message message)
     {
-        return Task.FromResult(JsonSerializer.Deserialize<MyLargeCommand>(message.Body.Bytes));
+        using MemoryStream stream = new(message.Body.Bytes);
+        return await JsonSerializer.DeserializeAsync<MyLargeCommand>(stream);
     }
 }
diff --git a/tests/Paramore.Brighter.AWS.Tests/Transformers/When_wrapping_a_large_message.cs b/tests/Paramore.Brighter.AWS.Tests/Transformers/When_wrapping_a_large_message.cs
@@ -36,6 +36,7 @@ public LargeMessagePayloadWrapTests()
                 new MessageMapperRegistry(null, new SimpleMessageMapperFactoryAsync(
                     _ => new MyLargeCommandMessageMapperAsync())
                     );
+
             mapperRegistry.RegisterAsync<MyLargeCommand, MyLargeCommandMessageMapperAsync>();
 
             _myCommand = new MyLargeCommand(6000);