test update

borisdj · Sep 14, 2024 · 18890a8 · 18890a8
1 parent 5a8847b
commit 18890a8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 7 deletions.
diff --git a/CsCodeGenerator.Test/Generated/ComplexNumber.cs b/CsCodeGenerator.Test/Generated/ComplexNumber.cs
@@ -39,7 +39,7 @@ public ComplexNumber Add(ComplexNumber input)
             ComplexNumber result = new ComplexNumber
             {
                 Real = Real + input.Real,
-                Imaginary = Imaginary + input.Imaginary
+                Imaginary = Imaginary + input.Imaginary,
             };
             return result;
         }

diff --git a/CsCodeGenerator.Test/WriteTests.cs b/CsCodeGenerator.Test/WriteTests.cs
@@ -293,6 +293,8 @@ private void WriteComplexNumberFile()
             secondConstructor.BodyLines.Add("Imaginary = imaginary;");
             complexNumberClass.Constructors.Add(secondConstructor);
 
+            var indent = complexNumberClass.Indent;
+
             var fields = new List<Field>
             {
                 new (BuiltInDataType.Double, "PI") { SingleKeyWord = KeyWord.Const, DefaultValue = "3.14" },
@@ -330,9 +332,11 @@ private void WriteComplexNumberFile()
                     Parameters = new List<Parameter> { new Parameter("ComplexNumber", "input") },
                     BodyLines = new List<string>
                     {
-                        "ComplexNumber result = new ComplexNumber();",
-                        "result.Real = Real + input.Real;",
-                        "result.Imaginary = Imaginary + input.Imaginary;",
+                        "ComplexNumber result = new ComplexNumber",
+                        "{",
+                        $"{indent}Real = Real + input.Real,",
+                        $"{indent}Imaginary = Imaginary + input.Imaginary,",
+                        "};",
                         "return result;"
                     }
                 },
@@ -407,9 +411,11 @@ private string GetComplexNumberFileText()
                 "",
                 "        public ComplexNumber Add(ComplexNumber input)",
                 "        {",
-                "            ComplexNumber result = new ComplexNumber();",
-                "            result.Real = Real + input.Real;",
-                "            result.Imaginary = Imaginary + input.Imaginary;",
+                "            ComplexNumber result = new ComplexNumber",
+                "            {",
+                "                Real = Real + input.Real,",
+                "                Imaginary = Imaginary + input.Imaginary,",
+                "            };",
                 "            return result;",
                 "        }",
                 "",