improve tests and examples

Author: hjgraca

examples/Kafka/JsonClassLibrary/src/Function.cs

@@ -21,7 +21,16 @@ public string FunctionHandler(ConsumerRecords<string, CustomerProfile> records,
     {
         foreach (var record in records)
         {
-            Logger.LogInformation("Record Value: {@record}", record.Value);
+            Logger.LogInformation("Processing messagem from topic: {topic}", record.Topic);
+            Logger.LogInformation("Partition: {partition}, Offset: {offset}", record.Partition, record.Offset);
+            Logger.LogInformation("Produced at: {timestamp}", record.Timestamp);
+            
+            foreach (var header in record.Headers.DecodedValues())
+            {
+                Logger.LogInformation($"{header.Key}: {header.Value}");
+            }
+            
+            Logger.LogInformation("Processing order for: {fullName}", record.Value.FullName);
         }
 
         return "Processed " + records.Count() + " records";

libraries/tests/AWS.Lambda.Powertools.Kafka.Tests/Protobuf/HandlerTests.cs

@@ -298,4 +298,65 @@ private async Task<string> HandlerWithProtobufKeys(ConsumerRecords<ProtobufKey,
 
         return "Successfully processed Protobuf Kafka events with complex keys";
     }
+    
+    [Fact]
+    public void SimpleHandlerTest()
+    {
+        string Handler(ConsumerRecords<int, ProtobufProduct> records, ILambdaContext context)
+        {
+            foreach (var record in records)
+            {
+                var product = record.Value;
+                context.Logger.LogInformation($"Processing {product.Name} at ${product.Price}");
+            }
+
+            return "Successfully processed Protobuf Kafka events";
+        }
+        // Simulate the handler execution
+        var mockLogger = new TestLambdaLogger();
+        var mockContext = new TestLambdaContext
+        {
+            Logger = mockLogger
+        };
+
+        var records = new ConsumerRecords<int, ProtobufProduct>
+        {
+            Records = new Dictionary<string, List<ConsumerRecord<int, ProtobufProduct>>>
+            {
+                { "mytopic-0", new List<ConsumerRecord<int, ProtobufProduct>>
+                    {
+                        new()
+                        {
+                            Topic = "mytopic",
+                            Partition = 0,
+                            Offset = 15,
+                            Key = 42,
+                            Value = new ProtobufProduct { Name = "Test Product", Id = 1, Price = 99.99 }
+                        }
+                    }
+                }
+            }
+        };
+        
+        // Call the handler
+        var result = Handler(records, mockContext);
+        
+        // Assert the result
+        Assert.Equal("Successfully processed Protobuf Kafka events", result);
+        
+        // Verify the context logger output
+        Assert.Contains("Processing Test Product at $99.99", mockLogger.Buffer.ToString());
+        
+        // Verify the records were processed
+        Assert.Single(records.Records);
+        Assert.Contains("mytopic-0", records.Records.Keys);
+        Assert.Single(records.Records["mytopic-0"]);
+        Assert.Equal("mytopic", records.Records["mytopic-0"][0].Topic);
+        Assert.Equal(0, records.Records["mytopic-0"][0].Partition);
+        Assert.Equal(15, records.Records["mytopic-0"][0].Offset);
+        Assert.Equal(42, records.Records["mytopic-0"][0].Key);
+        Assert.Equal("Test Product", records.Records["mytopic-0"][0].Value.Name);
+        Assert.Equal(1, records.Records["mytopic-0"][0].Value.Id);
+        Assert.Equal(99.99, records.Records["mytopic-0"][0].Value.Price);
+    }
 }