From 1ebed922b8c642b6ff4cec71820f24ceeb331c18 Mon Sep 17 00:00:00 2001
From: Ryo Nihei <nihei.dev@gmail.com>
Date: Sun, 29 May 2022 15:55:53 +0900
Subject: Support testable tree output in vartan-parse command

---
 spec/test/parser_test.go | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'spec/test/parser_test.go')

diff --git a/spec/test/parser_test.go b/spec/test/parser_test.go
index 6e77f6d..41b7189 100644
--- a/spec/test/parser_test.go
+++ b/spec/test/parser_test.go
@@ -7,6 +7,25 @@ import (
 	"testing"
 )
 
+func TestTree_Format(t *testing.T) {
+	expected := `(a
+    (b
+        (c))
+    (d)
+    (e))`
+	tree := NewTree("a",
+		NewTree("b",
+			NewTree("c"),
+		),
+		NewTree("d"),
+		NewTree("e"),
+	)
+	actual := string(tree.Format())
+	if actual != expected {
+		t.Fatalf("unexpected format:\n%v", actual)
+	}
+}
+
 func TestDiffTree(t *testing.T) {
 	tests := []struct {
 		t1        *Tree
-- 
cgit v1.2.3