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) } } }