Fix test

Cysharp · Dec 7, 2023 · 11b41b2 · 11b41b2
1 parent 1203b62
commit 11b41b2
Show file tree

Hide file tree

Showing 2 changed files with 45 additions and 8 deletions.
diff --git a/src/ZLogger.MessagePack/MessagePackZLoggerFormatter.cs b/src/ZLogger.MessagePack/MessagePackZLoggerFormatter.cs
@@ -98,20 +98,24 @@ public void FormatLogEntry<TEntry>(IBufferWriter<byte> writer, TEntry entry) whe
         }
         if ((IncludeProperties & IncludeProperties.ScopeKeyValues) != 0)
         {
-            if (entry.LogInfo.ScopeState != null)
+            if (entry.LogInfo.ScopeState == null)
+            {
+                propCount--;
+            }
+            else
             {
                 var scopeProperties = entry.LogInfo.ScopeState.Properties;
                 foreach (var t in scopeProperties)
                 {
                     if (t.Key != "{OriginalFormat}") scopePropCount++;
                 }
-            }
 
-            // flatten
-            if (PropertyNames.ScopeKeyValues == null)
-            {
-                propCount--;
-                propCount += scopePropCount;
+                // flatten
+                if (PropertyNames.ScopeKeyValues == null)
+                {
+                    propCount--;
+                    propCount += scopePropCount;
+                }
             }
         }
 

diff --git a/tests/ZLogger.MessagePack.Tests/FormatterTest.cs b/tests/ZLogger.MessagePack.Tests/FormatterTest.cs
@@ -263,13 +263,46 @@ public void NestedParameterKeyValues()
         logger = loggerFactory.CreateLogger("test");
 
 
+        logger.ZLogInformation($"{456:@y}");
+
+        var msg = processor.Dequeue();
+        ((string)msg["Message"]).Should().Be("456");
+        ((int)msg["attributes"]["y"]).Should().Be(456);
+    }
+
+    [Fact]
+    public void NestedScopeKeyValues()
+    {
+        var formatter = new MessagePackZLoggerFormatter
+        {
+            PropertyNames = MessagePackPropertyNames.Default with
+            {
+                ParameterKeyValues = MessagePackEncodedText.Encode("attributes"),
+                ScopeKeyValues = MessagePackEncodedText.Encode("scope")
+            }
+        };
+
+        processor = new TestProcessor(formatter);
+
+        var loggerFactory = LoggerFactory.Create(x =>
+        {
+            x.SetMinimumLevel(LogLevel.Debug);
+            x.AddZLoggerLogProcessor(options =>
+            {
+                options.IncludeScopes = true;
+                options.TimeProvider = timeProvider;
+                return processor;
+            });
+        });
+        logger = loggerFactory.CreateLogger("test");
+
         using (logger.BeginScope("{X}", 123))
         {
             logger.ZLogInformation($"{456:@y}");
         }
 
         var msg = processor.Dequeue();
-        ((string)msg["Message"]).Should().Be("100");
+        ((string)msg["Message"]).Should().Be("456");
         ((int)msg["scope"]["X"]).Should().Be(123);
         ((int)msg["attributes"]["y"]).Should().Be(456);
     }