Merge pull request #43 from dansiegel/codewriter-fix

Fix CodeWriter Dispose

src/CodeGenHelpers/Internals/CodeWriter.cs

@@ -111,7 +111,7 @@ private string GetIndentedValue(string value)
 
         public void Dispose()
         {
-            while (_indentLevel > 0)
+            if (_indentLevel > 0)
             {
                 _indentLevel--;
                 EnsureStringBuilder().AppendLine(GetIndentedValue("}"));

tests/CodeGenHelpers.Tests/SampleCode/MultipleMethodClass.cs

@@ -0,0 +1,22 @@
+//------------------------------------------------------------------------------
+// <auto-generated>
+//     This code was generated.
+//
+//     Changes to this file may cause incorrect behavior and will be lost if
+//     the code is regenerated.
+// </auto-generated>
+//------------------------------------------------------------------------------
+
+namespace CodeGenHelpers.SampleCode
+{
+    partial class MultipleMethodClass
+    {
+        public void Bar()
+        {
+        }
+
+        public void Foo()
+        {
+        }
+    }
+}

tests/CodeGenHelpers.Tests/Tests/CodeWriterTests.cs

@@ -29,4 +29,20 @@ public void GenerateCodeSubBlock()
         var r = sut.Render();
         Assert.Equal(expected, r);
     }
+
+    [Fact]
+    public void GenerateIfElse()
+    {
+        var writer = new CodeWriter(IndentStyle.Tabs);
+
+        writer.If("true")
+            .WithBody(w => w.AppendLine("CrashTheSystem();"))
+            .Else()
+            .WithBody(w => w.AppendLine("DontCrashTheSystem();"))
+            .EndIf();
+
+        const string expected = "if (true)\r\n{\r\n\tCrashTheSystem();\r\n}\r\nelse\r\n{\r\n\tDontCrashTheSystem();\r\n}\r\n";
+        var r = writer.Render();
+        Assert.Equal(expected, r);
+    }
 }

tests/CodeGenHelpers.Tests/Tests/MethodTests.cs

@@ -35,5 +35,20 @@ public void GenerateMethodWithPrimitiveParameter()
 
             MakeAssertion(builder);
         }
+
+        [Fact]
+        public void GenerateMultipleMethodsInClass()
+        {
+            var builder = CodeBuilder.Create(Namespace)
+                .AddClass("MultipleMethodClass")
+                .AddMethod("Foo")
+                .MakePublicMethod()
+                .Class
+                .AddMethod("Bar")
+                .MakePublicMethod()
+                .Class;
+
+            MakeAssertion(builder);
+        }
     }
 }