Extending test coverage for multi-line label values

Author: SteveWilkes

ReadableExpressions.UnitTests/WhenTranslatingGotos.cs

@@ -158,7 +158,6 @@ public void ShouldNotIncludeLabelNamesWithoutAGoto()
             Assert.AreEqual(EXPECTED.TrimStart(), translated);
         }
 
-
         [TestMethod]
         public void ShouldTranslateAReturnStatementWithABlock()
         {
@@ -195,5 +194,42 @@ public void ShouldTranslateAReturnStatementWithABlock()
 
             Assert.AreEqual(EXPECTED.TrimStart(), translated);
         }
+
+        [TestMethod]
+        public void ShouldTranslateALabelWithABlockDefaultValue()
+        {
+            var returnLabelTarget = Expression.Label(typeof(int), "Return");
+
+            var intVariable = Expression.Variable(typeof(int), "i");
+            var variableInit = Expression.Assign(intVariable, Expression.Constant(0));
+            var variablePlusOne = Expression.Add(intVariable, Expression.Constant(1));
+            var variableAdditionOne = Expression.Assign(intVariable, variablePlusOne);
+            var variablePlusTwo = Expression.Add(intVariable, Expression.Constant(2));
+            var variableAdditionTwo = Expression.Assign(intVariable, variablePlusTwo);
+
+            var variableBlock = Expression.Block(variableAdditionTwo, intVariable);
+
+            var returnVariableBlock = Expression.Label(returnLabelTarget, variableBlock);
+
+            var returnBlock = Expression.Block(
+                new[] { intVariable },
+                variableInit,
+                variableAdditionOne,
+                returnVariableBlock);
+
+            const string EXPECTED = @"
+var i = 0;
+i = i + 1;
+
+return 
+{
+    i = i + 2;
+    return i;
+};";
+
+            var translated = returnBlock.ToReadableString();
+
+            Assert.AreEqual(EXPECTED.TrimStart(), translated);
+        }
     }
 }

ReadableExpressions/Translators/LabelExpressionTranslator.cs

@@ -20,6 +20,7 @@ public override string Translate(Expression expression, TranslationContext conte
             {
                 return labelNamePart;
             }
+
             var labelValuePart = $"{Environment.NewLine}return {context.TranslateAsCodeBlock(label.DefaultValue)};";
 
             return labelNamePart + labelValuePart;