hugo/hugolib/site_show_plan_test.go

package hugolib

import (
	"bytes"
	"testing"
	"github.com/spf13/hugo/target"
)

func checkShowPlanExpected(t *testing.T, expected, got string) {
	if got != expected {
		t.Errorf("ShowPlan expected:\n%q\ngot\n%q", expected, got)
	}
}

func TestDegenerateNoFiles(t *testing.T) {
	s := new(Site)
	out := new(bytes.Buffer)
	if err := s.ShowPlan(out); err != nil {
		t.Errorf("ShowPlan unexpectedly returned an error: %s", err)
	}
	expected := "No source files provided.\n"
	got := out.String()
	checkShowPlanExpected(t, expected, got)
}

func TestDegenerateNoTarget(t *testing.T) {
	s := new(Site)
	s.Files = append(s.Files, "foo/bar/file.md")
	out := new(bytes.Buffer)
	if err := s.ShowPlan(out); err != nil {
		t.Errorf("ShowPlan unexpectedly returned an error: %s", err)
	}

	expected := "foo/bar/file.md\n canonical => !no target specified!\n"
	checkShowPlanExpected(t, expected, out.String())
}

func TestFileTarget(t *testing.T) {
	s := &Site{Target: new(target.Filesystem)}
	s.Files = append(s.Files, "foo/bar/file.md")
	out := new(bytes.Buffer)
	s.ShowPlan(out)

	expected := "foo/bar/file.md\n canonical => foo/bar/file/index.html\n"
	checkShowPlanExpected(t, expected, out.String())
}

func TestFileTargetUgly(t *testing.T) {
	s := &Site{Target: &target.Filesystem{UglyUrls: true}}
	s.Files = append(s.Files, "foo/bar/file.md")
	out := new(bytes.Buffer)
	s.ShowPlan(out)

	expected := "foo/bar/file.md\n canonical => foo/bar/file.html\n"
	checkShowPlanExpected(t, expected, out.String())
}