From 287926d2788ba19baac1dc59bd5c81757a4b1c08 Mon Sep 17 00:00:00 2001
From: Kare Nuorteva <kare.nuorteva@me.com>
Date: Sun, 15 Sep 2024 20:30:44 +0300
Subject: [PATCH] test: add validate Go package test

---
 vanity_internal_test.go | 42 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 vanity_internal_test.go

diff --git a/vanity_internal_test.go b/vanity_internal_test.go
new file mode 100644
index 0000000..9de6eb2
--- /dev/null
+++ b/vanity_internal_test.go
@@ -0,0 +1,42 @@
+package vanity
+
+import "testing"
+
+func TestValidateGoPackage(t *testing.T) {
+	tests := []struct {
+		name    string
+		pkg     string
+		wantErr bool
+	}{
+		{
+			name:    "all allowed characters",
+			pkg:     "/foo/bar/123/foo_abc",
+			wantErr: false,
+		},
+		{
+			name:    "contains query",
+			pkg:     "/foobar?go-get=1",
+			wantErr: false,
+		},
+		{
+			name:    "contains dot",
+			pkg:     "/foo.bar",
+			wantErr: false,
+		},
+		{
+			name:    "contains minus",
+			pkg:     "/foo-bar",
+			wantErr: false,
+		},
+	}
+	for _, test := range tests {
+		test := test
+		t.Run(test.name, func(t *testing.T) {
+			t.Parallel()
+
+			if err := validateGoPkgPath(test.pkg); (err != nil) != test.wantErr {
+				t.Errorf("expecting error, none reported. error: %v", err)
+			}
+		})
+	}
+}