Add a few test cases to lib

1 files changed, 23 insertions, 0 deletions

diff --git a/lib/lib_test.go b/lib/lib_test.go
new file mode 100644
index 0000000..7b46bfb
--- /dev/null
+++ b/lib/lib_test.go
@@ -0,0 +1,23 @@
+// Copyright (c) 2019, Grégoire Duchêne <gduchene@awhk.org>
+//
+// Use of this source code is governed by the ISC license that can be
+// found in the LICENSE file.
+
+package lib
+
+import "testing"
+
+func TestGetDest(t *testing.T) {
+	cs := []struct{ srcPrefix, destPrefix, pkg, expected string }{
+		{"src.example.com/x/", "https://example.com/git/", "src.example.com/x/foo", "https://example.com/git/foo"},
+		{"src.example.com/x/", "https://example.com/git/", "src.example.com/x/foo/bar", "https://example.com/git/foo"},
+		{"src.example.com/x", "https://example.com/git", "src.example.com/x/foo", "https://example.com/git/foo"},
+		{"src.example.com/x", "https://example.com/git", "src.example.com/x/foo/bar", "https://example.com/git/foo"},
+	}
+	for _, c := range cs {
+		actual := GetDest(c.srcPrefix, c.destPrefix, c.pkg)
+		if actual != c.expected {
+			t.Errorf("expected %s, got %s", c.expected, actual)
+		}
+	}
+}