1 files changed, 28 insertions, 0 deletions
diff --git a/gazelle/pythonconfig/pythonconfig_test.go b/gazelle/pythonconfig/pythonconfig_test.go
new file mode 100644
index 0000000..1512eb9
--- /dev/null
+++ b/gazelle/pythonconfig/pythonconfig_test.go
@@ -0,0 +1,28 @@
+package pythonconfig
+
+import (
+	"testing"
+
+	"github.com/bazelbuild/rules_python/gazelle/pythonconfig"
+)
+
+func TestDistributionSanitizing(t *testing.T) {
+	tests := map[string]struct {
+		input string
+		want  string
+	}{
+		"upper case": {input: "DistWithUpperCase", want: "distwithuppercase"},
+		"dashes":     {input: "dist-with-dashes", want: "dist_with_dashes"},
+		"dots":       {input: "dist.with.dots", want: "dist_with_dots"},
+		"mixed":      {input: "To-be.sanitized", want: "to_be_sanitized"},
+	}
+
+	for name, tc := range tests {
+		t.Run(name, func(t *testing.T) {
+			got := pythonconfig.SanitizeDistribution(tc.input)
+			if tc.want != got {
+				t.Fatalf("expected %q, got %q", tc.want, got)
+			}
+		})
+	}
+}