hugo/commands/server_test.go

package commands

import (
	"testing"

	"github.com/spf13/viper"
)

func TestFixURL(t *testing.T) {
	type data struct {
		TestName   string
		CLIBaseURL string
		CfgBaseURL string
		AppendPort bool
		Port       int
		Result     string
	}
	tests := []data{
		{"Basic http localhost", "", "http://foo.com", true, 1313, "http://localhost:1313/"},
		{"Basic https production, http localhost", "", "https://foo.com", true, 1313, "http://localhost:1313/"},
		{"Basic subdir", "", "http://foo.com/bar", true, 1313, "http://localhost:1313/bar/"},
		{"Basic production", "http://foo.com", "http://foo.com", false, 80, "http://foo.com/"},
		{"Production subdir", "http://foo.com/bar", "http://foo.com/bar", false, 80, "http://foo.com/bar/"},
		{"No http", "", "foo.com", true, 1313, "http://localhost:1313/"},
		{"Override configured port", "", "foo.com:2020", true, 1313, "http://localhost:1313/"},
		{"No http production", "foo.com", "foo.com", false, 80, "http://foo.com/"},
		{"No http production with port", "foo.com", "foo.com", true, 2020, "http://foo.com:2020/"},
	}

	for i, test := range tests {
		BaseURL = test.CLIBaseURL
		viper.Set("BaseURL", test.CfgBaseURL)
		serverAppend = test.AppendPort
		serverPort = test.Port
		result, err := fixURL(BaseURL)
		if err != nil {
			t.Errorf("Test #%d %s: unexpected error %s", i, test.TestName, err)
		}
		if result != test.Result {
			t.Errorf("Test #%d %s: expected %q, got %q", i, test.TestName, test.Result, result)
		}
	}
}