diff options
Diffstat (limited to 'helpers/url_test.go')
-rw-r--r-- | helpers/url_test.go | 32 |
1 files changed, 30 insertions, 2 deletions
diff --git a/helpers/url_test.go b/helpers/url_test.go index e27e2bb02..c38750a63 100644 --- a/helpers/url_test.go +++ b/helpers/url_test.go @@ -1,9 +1,9 @@ package helpers import ( - "testing" - "github.com/stretchr/testify/assert" + "strings" + "testing" ) func TestUrlize(t *testing.T) { @@ -26,6 +26,34 @@ func TestUrlize(t *testing.T) { } } +func TestSanitizeUrl(t *testing.T) { + tests := []struct { + input string + expected string + }{ + {"http://foo.bar/", "http://foo.bar/"}, + {"http://foo.bar/zoo/", "http://foo.bar/zoo"}, // issue #931 + } + + for _, test := range tests { + o1 := SanitizeUrl(test.input) + o2 := SanitizeUrlKeepTrailingSlash(test.input) + + expected2 := test.expected + + if strings.HasSuffix(test.input, "/") && !strings.HasSuffix(expected2, "/") { + expected2 += "/" + } + + if o1 != test.expected { + t.Errorf("Expected %#v, got %#v\n", test.expected, o1) + } + if o2 != expected2 { + t.Errorf("Expected %#v, got %#v\n", expected2, o2) + } + } +} + func TestMakePermalink(t *testing.T) { type test struct { host, link, output string |