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