markdown_test.go

package rescript

import (
	"bytes"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestMarkdownPage(t *testing.T) {
	assert := assert.New(t)

	var buf bytes.Buffer

	append := func(n *Node, s string) {
		head := n
		for node := n; node != nil; node = node.Next() {
			head = node
		}
		head.InsertAfter(NewNode(NewToken(s)))
	}

	node := NewNode(NewToken("foo"))
	append(node, " ")
	append(node, "bar")
	append(node, " ")
	append(node, "baz")
	append(node, "\n")
	append(node, "newline")

	node1 := NewNode(NewToken("second page"))

	m := Metadata{
		Title:   "My Title",
		PageIDs: []string{"page0", "page1"},
	}

	nodes := map[string]*Node{
		"page0": node,
		"page1": node1,
	}

	c := NewMarkdownComposer()
	err := c(&buf, m, nodes)
	assert.Nil(err)

	expected := "# My Title\n\n**Page 1**\n\nfoo bar baz\nnewline\n\n---\n\n**Page 2**\n\nsecond page\n"
	assert.Equal(expected, buf.String())
}

func TestMarkdownError(t *testing.T) {
	assert := assert.New(t)

	node := NewNode(NewToken("foo"))
	w := failWriter{}

	err := markdownPage(w, 2, node)
	assert.Error(err)
}