From 52361b8938acd93894b0f3ec3d5992e602929400 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sat, 24 Oct 2026 23:16:03 +0100
Subject: Add strip command

---
 .../hobgoblin/compiler/output/CodeWriterTests.java | 77 ++++++++++++++++++++++
 1 file changed, 77 insertions(+)

(limited to 'src/test/java')

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
+            """)));
+    }
 }
-- 
cgit v1.2.3