test custom alphabet

publicid_test.go

@@ -5,6 +5,7 @@ package publicid
 
 import (
 	"fmt"
+	"strings"
 	"testing"
 )
 
@@ -124,3 +125,19 @@ func TestNewFailsAfterAttempts(t *testing.T) {
 		}
 	}
 }
+
+func TestNewWithCustomAlphabet(t *testing.T) {
+	customAlphabet := "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+	id, err := New(Alphabet(customAlphabet))
+	if err != nil {
+		t.Errorf("New(Alphabet(%q)) returned an error: %v", customAlphabet, err)
+	}
+	if len(id) != DefaultIDLength {
+		t.Errorf("New(Alphabet(%q)) returned id with length %d, want %d", customAlphabet, len(id), DefaultIDLength)
+	}
+	for _, char := range id {
+		if !strings.ContainsRune(customAlphabet, char) {
+			t.Errorf("New(Alphabet(%q)) returned id containing invalid character: %c", customAlphabet, char)
+		}
+	}
+}