1 files changed, 117 insertions, 0 deletions
diff --git a/frontmatter_test.go b/frontmatter_test.go
new file mode 100644
index 0000000..98d8164
--- /dev/null
+++ b/frontmatter_test.go
@@ -0,0 +1,117 @@
+// Copyright (c) 2023 Nicolas Paul All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package main
+
+import (
+	"strconv"
+	"testing"
+	"time"
+
+	"github.com/google/go-cmp/cmp"
+)
+
+func TestParseFrontMatter(t *testing.T) {
+	tests := []struct {
+		input []byte
+		want  FrontMatter
+		rest  string
+	}{
+		{
+			input: []byte(`---
+title: "Test"
+description: "Test"
+publication_time: 2020-01-01T00:00:00Z
+last_update_time: 2020-01-01T00:00:00Z
+keywords: ["test"]
+author: "Test"
+hide: false
+---
+# Test`),
+			want: FrontMatter{
+				Title:           "Test",
+				Description:     "Test",
+				PublicationTime: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
+				LastUpdateTime:  time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
+				Keywords:        []string{"test"},
+				Author:          "Test",
+				Hide:            false,
+			},
+			rest: "# Test",
+		},
+		{
+			input: []byte(`---
+title: "Test"
+description: "Test"
+publication_time: 2020-01-01T00:00:00Z
+keywords: ["test", "test2", "test3"]
+---
+# Test`),
+			want: FrontMatter{
+				Title:           "Test",
+				Description:     "Test",
+				PublicationTime: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
+				Keywords:        []string{"test", "test2", "test3"},
+			},
+			rest: "# Test",
+		},
+	}
+
+	for _, tt := range tests {
+		fm, rest, err := ParseFrontMatter(tt.input)
+
+		if err != nil {
+			t.Fatalf("ParseFrontMatter(%v) got error %v", tt.input, err)
+		}
+
+		if !cmp.Equal(fm, tt.want) {
+			t.Fatalf("ParseFrontMatter(%v) got %v want %v", tt.input, fm, tt.want)
+		}
+
+		if string(rest) != tt.rest {
+			t.Fatalf("ParseFrontMatter(%v) got %v want %v", tt.input, rest, tt.rest)
+		}
+	}
+}
+
+func FuzzParseFrontMatter(f *testing.F) {
+	// Do not fuzz *_time fields as time.Time is not supported by f.Fuzz().
+	f.Fuzz(func(t *testing.T, title string, description string, keyword1 string, keyword2 string, keyword3 string, author string, hide bool, text string) {
+		input := []byte(`---
+title: "` + title + `"
+description: "` + description + `"
+publication_time: 2020-01-01T00:00:00Z
+last_update_time: 2020-01-01T00:00:00Z
+keywords: ["` + keyword1 + `", "` + keyword2 + `", "` + keyword2 + `"]
+author: "` + author + `"
+hide: ` + strconv.FormatBool(hide) + `
+---
+` + text)
+
+		fm, rest, err := ParseFrontMatter(input)
+
+		if err != nil {
+			t.Fatalf("ParseFrontMatter(%v) got error %v", input, err)
+		}
+
+		if string(rest) != text {
+			t.Fatalf("ParseFrontMatter(%v) got %v want %v", input, rest, text)
+		}
+
+		result := FrontMatter{
+			Title:           title,
+			Description:     description,
+			Keywords:        []string{keyword1, keyword2, keyword3},
+			PublicationTime: time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
+			LastUpdateTime:  time.Date(2020, 1, 1, 0, 0, 0, 0, time.UTC),
+			Author:          author,
+			Hide:            hide,
+		}
+
+		if !cmp.Equal(fm, result) {
+			t.Fatalf("ParseFrontMatter(%v) got %v want %v", input, fm, result)
+		}
+
+	})
+}