gedcomer_test.go

package gedcom_test

import (
	"testing"

	"github.com/elliotchance/gedcom/v39"
	"github.com/elliotchance/tf"
)

func TestGEDCOMLine(t *testing.T) {
	GEDCOMLine := tf.Function(t, gedcom.GEDCOMLine)
	nameNode := gedcom.NewNameNode("Joe /Bloggs/",
		gedcom.NewDateNode("3 Sep 1943"),
	)

	GEDCOMLine(nil, 0).Returns("")

	GEDCOMLine(nameNode, gedcom.NoIndent).Returns("NAME Joe /Bloggs/")

	GEDCOMLine(nameNode, 0).Returns("0 NAME Joe /Bloggs/")

	GEDCOMLine(nameNode, 5).Returns("5 NAME Joe /Bloggs/")
}

func TestGEDCOMString(t *testing.T) {
	GEDCOMString := tf.Function(t, gedcom.GEDCOMString)
	nameNode := gedcom.NewNameNode("Joe /Bloggs/",
		gedcom.NewDateNode("3 Sep 1943"),
	)

	GEDCOMString(nil, 0).Returns("")

	GEDCOMString(nameNode, gedcom.NoIndent).
		Returns("NAME Joe /Bloggs/\nDATE 3 Sep 1943\n")

	GEDCOMString(nameNode, 0).
		Returns("0 NAME Joe /Bloggs/\n1 DATE 3 Sep 1943\n")

	GEDCOMString(nameNode, 5).
		Returns("5 NAME Joe /Bloggs/\n6 DATE 3 Sep 1943\n")
}