1 files changed, 77 insertions, 0 deletions
diff --git a/src/test/java/org/zwobble/hobgoblin/compiler/output/CodeWriterTests.java b/src/test/java/org/zwobble/hobgoblin/compiler/output/CodeWriterTests.java
index decea99..9e3f900 100644
--- a/src/test/java/org/zwobble/hobgoblin/compiler/output/CodeWriterTests.java
+++ b/src/test/java/org/zwobble/hobgoblin/compiler/output/CodeWriterTests.java
@@ -134,4 +134,81 @@ public class CodeWriterTests {
         assertThat(stringWriter.toString(), equalTo(""));
         assertThat(file.isWriterOpened(), equalTo(false));
     }
+
+    @Test
+    public void strippingDoesNothingToFileWithoutHeader() {
+        var originalContents = """
+            Hello, world!
+            """;
+
+        var result = CodeWriter.stripFileContents(
+            originalContents,
+            new CodeWriter.CodeWriterConfig("//")
+        );
+
+        assertThat(result, equalTo(new CodeWriter.StripResult.Unchanged()));
+    }
+
+    @Test
+    public void whenFileHasNoCustomAreasThenFileIsDeleted() {
+        var originalContents = """
+            // Generated by hobgoblin.
+
+            Hello, world!
+            """;
+
+        var result = CodeWriter.stripFileContents(
+            originalContents,
+            new CodeWriter.CodeWriterConfig("//")
+        );
+
+        assertThat(result, equalTo(new CodeWriter.StripResult.Delete()));
+    }
+
+    @Test
+    public void whenCustomAreasAreAllEmptyThenFileIsDeleted() {
+        var originalContents = """
+            // Generated by hobgoblin.
+
+            Before
+            // Custom area start: body
+            // Custom area end: body
+            After
+            """;
+
+        var result = CodeWriter.stripFileContents(
+            originalContents,
+            new CodeWriter.CodeWriterConfig("//")
+        );
+
+        assertThat(result, equalTo(new CodeWriter.StripResult.Delete()));
+    }
+
+    @Test
+    public void nonEmptyCustomAreasArePreserved() {
+        var originalContents = """
+            // Generated by hobgoblin.
+
+            Before
+            // Custom area start: head
+            Custom head.
+            // Custom area end: head
+            // Custom area start: body
+            // Custom area end: body
+            After
+            """;
+
+        var result = CodeWriter.stripFileContents(
+            originalContents,
+            new CodeWriter.CodeWriterConfig("//")
+        );
+
+        assertThat(result, equalTo(new CodeWriter.StripResult.NewContent("""
+            // Generated by hobgoblin.
+
+            // Custom area start: head
+            Custom head.
+            // Custom area end: head
+            """)));
+    }
 }