1 files changed, 37 insertions, 3 deletions
diff --git a/src/title.zig b/src/title.zig
index de88c67..62d7929 100644
--- a/src/title.zig
+++ b/src/title.zig
@@ -9,6 +9,7 @@ pub const Error = error{InvalidTitleContent} || paragraph.Error || Lexer.Error;
 
 pub fn parse(alloc: Allocator, l: *Lexer) Error!Element {
     var v = (try l.next(alloc)).?;
+    defer v.deinit();
     var el = try Element.init(alloc, .content, switch (v.content.items.len) {
         1 => "h1",
         2 => "h2",
@@ -18,12 +19,45 @@ pub fn parse(alloc: Allocator, l: *Lexer) Error!Element {
         6 => "h6",
         else => unreachable,
     });
+    errdefer el.deinit();
     try el.appendContent(paragraph.parseLine(alloc, l) catch |err| switch (err) {
         paragraph.Error.IllegalPlacement => return Error.InvalidTitleContent,
         else => return err,
     });
-    v = (try l.next(alloc)) orelse return el;
-    if (!v.kind.isDelimiter()) return Error.InvalidTitleContent;
-    v.deinit();
+    var next = (try l.next(alloc)) orelse return el;
+    defer next.deinit();
+    if (!next.kind.isDelimiter()) return Error.InvalidTitleContent;
     return el;
 }
+
+fn doTest(alloc: Allocator, t: []const u8, v: []const u8) !void {
+    var l = try Lexer.init(t);
+    var p = try parse(alloc, &l);
+    defer p.deinit();
+    const g = try p.render(alloc);
+    defer alloc.free(g);
+    std.testing.expect(std.mem.eql(u8, g, v)) catch |err| {
+        std.debug.print("{s}\n", .{g});
+        return err;
+    };
+}
+
+fn doTestError(alloc: Allocator, t: []const u8, err: Error) !void {
+    var l = try Lexer.init(t);
+    _ = parse(alloc, &l) catch |e| return std.testing.expect(err == e);
+    return std.testing.expect(false);
+}
+
+test "parse title" {
+    var arena = std.heap.DebugAllocator(.{}).init;
+    defer if (arena.deinit() == .leak) std.debug.print("leaking!\n", .{});
+    const alloc = arena.allocator();
+
+    try doTest(alloc, "# hey", "<h1>hey</h1>");
+    try doTest(alloc, "## hey", "<h2>hey</h2>");
+    try doTest(alloc, "### hey", "<h3>hey</h3>");
+
+    try doTest(alloc, "# hello *world*", "<h1>hello <b>world</b></h1>");
+
+    try doTestError(alloc, "# aa :::", Error.InvalidTitleContent);
+}